A simple interactive Python script to organize messy folders on the computer, supports macOS and linux but not windows. The mini-program has the following features:
- Create alphanumeric folders (A-Z / 0-9) based on the available set of letters in the folder specified in the path and move the folders / files to the newly created folders.
- Creation of folders specific to the type (Audio, Video …) and moving files in the appropriate type folders.
- Run directly from shell or from a Python interpreter.
- Interactive.
It is planned to add more features to the code and to possibly perform other optimizations / updates, here is my implementation in Python, I hope it suits you, suggestions for improvements / features to add / problems encountered when using the code are more than welcome. The program works fine, no bug has been found so far, however, you may want to test it before using it on a folder containing test files to get you used to it.
#!/usr/bin/env python
import os
import string
import shutil
import random
def thank_user():
"""Print a thank you message and end program."""
print(5 * ' ' + 10 * '=')
print('Thank you for using file organizer.')
print(5 * ' ' + 10 * '=')
exit()
def get_folder_contents(path):
"""Return folder contents."""
while not os.path.exists(path):
path = input('Invalid path, please enter a valid path or q to exit: ').rstrip()
if path == 'q':
thank_user()
os.chdir(path)
return (file for file in os.listdir(path) if file != '.DS_Store')
def welcome_user():
"""Print a welcome message."""
print(5 * ' ' + 10 * '=')
print('Welcome to file organizer.')
print(5 * ' ' + 10 * '=')
action = input('Do you wish to continue? y/n: ')
while action != 'y' and action != 'n':
print(f'Invalid command {action}')
action = input('Do you wish to continue? y/n: ')
if action == 'n':
thank_user()
if action == 'y':
pass
def create_alpha_num_folders(path):
"""Create alpha-numeric folders for first characters present in files of the given path ."""
folder_contents = get_folder_contents(path)
if not folder_contents:
print(f"0 files found in {str(path).split('/')(-1)}")
print(105 * '=')
thank_user()
chars_to_produce = set((item(0).upper() for item in folder_contents))
invalid_folders = (folder for folder in folder_contents if folder in string.ascii_uppercase
or folder in string.digits)
if invalid_folders:
print(f'Found the following {len(invalid_folders)} folders named from A - Z/ 1 - n:')
for invalid_folder in invalid_folders:
print(invalid_folder)
while invalid_folders:
action = input(f'Enter q to exit. Do you want to to rename the {len(invalid_folders)}'
f' invalid folder(s)? y/n/q: ')
if action == 'q' or action == 'n':
print('Creating A - Z / 1 - n folders unsuccessful.')
thank_user()
if action == 'y':
confirm = input(f'Are you sure, do you want to rename {len(invalid_folders)} invalid folders? y/n: ')
while not confirm == 'y' or confirm == 'n':
print(f'Invalid command {confirm}')
confirm = input(f'Are you sure, do you want to rename {len(invalid_folders)} '
f'invalid folders? y/n: ')
if confirm == 'y':
count = random.randint(10 ** 6, 10 ** 7)
for filename in invalid_folders:
new_name = filename + '_' + str(count)
os.rename(filename, new_name)
print(f'Changed {len(invalid_folders)} folder name(s) successful.')
break
if confirm == 'n':
thank_user()
else:
print(f'Invalid command {action}')
for char in sorted(chars_to_produce):
os.mkdir(char)
print(f'Creation of folder {char} successful.')
print(105 * '=')
def get_supported_formats():
"""Return supported formats."""
formats = {
'AUDIO': ('aif', 'cda', 'mid', 'midi', 'mp3', 'mpa', 'ogg', 'wav', 'wma', 'wpl'),
'COMPRESSED': ('7z', 'arj', 'deb', 'pkg', 'rar', 'rpm', 'tar.gz', 'z', 'zip'),
'DISC': ('bin', 'dmg', 'iso', 'toast', 'vcd'),
'DATA': ('csv', 'dat', 'db', 'dbf', 'log', 'mdb', 'sav', 'sql', 'tar', 'xml'),
'EXECUTABLE': ('apk', 'bat', 'bin', 'cgi', 'pl', 'com', 'exe', 'gadget', 'jar', 'wsf'),
'FONT': ('fnt', 'fon', 'otf', 'ttf'),
'IMAGE': ('ai', 'bmp', 'gif', 'ico', 'jpeg', 'jpg', 'png', 'ps', 'psd', 'svg', 'tif', 'tiff'),
'INTERNET': ('asp', 'aspx', 'cer', 'cfm', 'css', 'htm', 'html', 'js', 'jsp',
'part', 'php', 'rss', 'xhtml'),
'PRESENTATION': ('key', 'odp', 'pps', 'ppt', 'pptx'),
'PROGRAMMING': ('c', 'class', 'cpp', 'cs', 'h', 'java', 'sh', 'swift', 'vb', 'py'),
'SPREADSHEET': ('ods', 'xlr', 'xls', 'xlsx'),
'SYSTEM': ('bak', 'cab', 'cfg', 'cpl', 'cur', 'dll', 'dmp', 'drv', 'icns', 'ini',
'lnk', 'msi', 'sys', 'tmp'),
'VIDEO': ('3g2', '3gp', 'avi', 'flv', 'h264', 'm4v', 'mkv', 'mov', 'mp4', 'mpg', 'mpeg',
'rm', 'swf', 'vob', 'wmv', 'webm'),
'WORD_PROCESSOR': ('doc', 'docx', 'odt', 'pdf', 'rtf', 'tex', 'txt', 'wks', 'wps', 'wpd'),
'SUBTITLES': ('srt', 'sub', 'sbv')
}
return formats
def get_extension_type(extension):
"""Return extension type ex: mp3 --> AUDIO."""
formats = get_supported_formats()
possible_extensions = tuple(ext for ext, exts in formats.items() if extension.lower() in exts)
if not possible_extensions:
return ('UNCLASSIFIED',)
return possible_extensions
def get_file_extension(filename):
"""Return file extension"""
return filename.split('.')(-1).lower()
def get_folder_extensions(path):
"""Return all folder extensions."""
folder_contents = get_folder_contents(path)
return set((get_file_extension(filename) for filename in folder_contents))
def create_folders(path):
"""Create alpha-numeric folders or type specific folders in the given path."""
folder_contents = get_folder_contents(path)
print(f"Current folder: {str(path).split('/')(-1)}")
print(f'Folder contents ({len(folder_contents)}) items:')
print(105 * '=')
if not folder_contents:
print(f"0 files found in {str(path).split('/')(-1)}")
print(105 * '=')
thank_user()
for number, filename in enumerate(sorted(folder_contents), 1):
if os.path.isdir(filename):
print(f'{number}. Folder: {filename}')
if os.path.isfile(filename):
print(f'{number}. File: {filename}')
print(105 * '=')
action1 = input(f'Enter q to exit. Do you want to group the {len(folder_contents)}'
f' items above Alpha-numerically (A-Z / 0-9) or by type? a/t/q: ')
while action1 != 'a' and action1 != 't' and action1 != 'q':
print(f'Invalid command {action1}')
action1 = input(f'Enter q to exit. Do you want to group the {len(folder_contents)}'
f' items above Alpha-numerically (A-Z / 0-9) or by type? a/t/q: ')
if action1 == 'a':
action2 = input(f"Enter q to exit. Are you sure, do you want to create Alpha-numeric (A-Z / 0-9) folders in "
f"{str(path).split('/')(-1)}? y/n/q: ")
while action2 != 'y' and action2 != 'n' and action2 != 'q':
print(f'Invalid command {action2}')
action2 = input(
f"Enter q to exit. Are you sure, do you want to create Alpha-numeric (A-Z / 0-9) folders in "
f"{str(path).split('/')(-1)}? y/n/q: ")
if action2 == 'n' or action2 == 'q':
print('Creation of new Alpha-numeric (A-Z / 0-9) folders unsuccessful.')
thank_user()
if action2 == 'y':
create_alpha_num_folders(path)
return 'alpha'
if action1 == 't':
folder_extensions = get_folder_extensions(path)
new_folders = set((get_extension_type(extension) for extension in folder_extensions))
new_folder_names = (folder_name for folder_names in new_folders for folder_name in folder_names)
print('New type folders to create: ')
for number, name in enumerate(sorted(new_folder_names), 1):
print(number, name)
action2 = input(f"Enter q to exit. Are you sure do you want to create the {len(new_folder_names)} folders"
f" above in {str(path).split('/')(-1)}? y/n/q: ")
while action2 != 'y' and action2 != 'n' and action2 != 'q':
print(f'Invalid command {action2}')
action2 = input(f"Enter q to exit. Are you sure do you want to create the {len(new_folder_names)} folders"
f" above in {str(path).split('/')(-1)}? y/n/q: ")
if action2 == 'y':
for folder in sorted(new_folder_names):
try:
os.mkdir(folder)
print(f'Creation of {folder} folder successful.')
except FileExistsError:
print(f'Filename {folder} already exists.')
print(f'Creation of {folder} folder unsuccessful.')
return 'type'
if action2 == 'n' or action2 == 'q':
print(f'Creation of {len(new_folder_names)} folders unsuccessful.')
thank_user()
if action1 == 'q':
print('Creation of new folders unsuccessful.')
thank_user()
def organize_files(path):
"""Move files and folders into new organized relevant folders."""
folder_types = create_folders(path)
if folder_types == 'alpha':
new_folder_contents = get_folder_contents(path)
files_to_move = (folder for folder in new_folder_contents if folder not in string.ascii_uppercase
and folder not in string.digits)
new_alpha_folders = (folder for folder in new_folder_contents if folder in string.ascii_uppercase
or folder in string.digits)
print(f"Current folder: {str(path).split('/')(-1)}")
print(f'Folder contents to be moved ({len(files_to_move)}) items:')
print(105 * '=')
for number, filename in enumerate(sorted(files_to_move), 1):
if os.path.isdir(filename):
print(f'{number}. Folder: {filename}')
if os.path.isfile(filename):
print(f'{number}. File: {filename}')
print(105 * '=')
action = input(f'Enter q to exit. Are you sure, do you want to organize the {len(files_to_move)} files above'
f' to Alpha-numeric (A-Z / 0-9) folders? y/n/q: ')
while action != 'q' and action != 'y' and action != 'n':
print(f'Invalid command {action}')
action = input(
f'Enter q to exit. Are you sure, do you want to organize the {len(files_to_move)} files above'
f' to Alpha-numeric (A-Z / 0-9) folders? y/n/q: ')
if action == 'q' or action == 'n':
print(f'Organization of {len(files_to_move)} Alpha-numerically (A-Z / 0-9) unsuccessful.')
thank_user()
if action == 'y':
for filename in files_to_move:
for folder_name in new_alpha_folders:
if filename.startswith(folder_name.lower()) or filename.startswith(folder_name):
shutil.move(filename, folder_name)
print(f'Moving successful ... {filename} to {folder_name}')
if folder_types == 'type':
new_folder_contents = get_folder_contents(path)
files_to_move = (folder for folder in new_folder_contents if folder not in get_supported_formats())
new_type_folders = (folder for folder in new_folder_contents if folder in get_supported_formats())
print(f"Current folder: {str(path).split('/')(-1)}")
print(f'Folder contents to be moved ({len(files_to_move)}) items:')
print(105 * '=')
for number, filename in enumerate(sorted(files_to_move, key=get_file_extension), 1):
if os.path.isdir(filename):
print(f'{number}. Folder: {filename}')
if os.path.isfile(filename):
print(f'{number}. File: {filename}')
print(105 * '=')
action = input(f'Enter q to exit. Are you sure, do you want to organize the {len(files_to_move)} files above'
f' into type specific folders? y/n/q: ')
while action != 'y' and action != 'n' and action != 'q':
print(f'Invalid command {action}')
action = input(
f'Enter q to exit. Are you sure, do you want to organize the {len(files_to_move)} files above'
f' into type specific folders? y/n/q: ')
if action == 'n' or action == 'q':
print(f'Organization of {len(files_to_move)} by type unsuccessful.')
thank_user()
if action == 'y':
for filename in files_to_move:
file_type = get_extension_type(get_file_extension(filename))
for folder_name in new_type_folders:
if file_type(0) == folder_name:
shutil.move(filename, folder_name)
print(f'Moving successful ... {filename} to {folder_name}')
thank_user()
if __name__ == '__main__':
welcome_user()
path_to_folder = input('Please enter a valid path or q to exit: ').rstrip()
if path_to_folder == 'q':
thank_user()
organize_files(path_to_folder)
