mirror of
https://github.com/hastagAB/Awesome-Python-Scripts.git
synced 2024-10-06 04:49:27 +00:00
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
#!/usr/bin/python3
|
|
|
|
from shutil import move
|
|
from os import path
|
|
import os
|
|
|
|
|
|
folder_ex = {
|
|
'Programming Files': set(['ipynb', 'py', 'java', 'cs', 'js', 'vsix', 'jar', 'cc', 'ccc', 'html', 'xml', 'kt']),
|
|
'Compressed': set(['zip', 'rar', 'arj', 'gz', 'sit', 'sitx', 'sea', 'ace', 'bz2', '7z']),
|
|
'Applications': set(['exe', 'msi', 'deb', 'rpm']),
|
|
'Pictures': set(['jpeg', 'jpg', 'png', 'gif', 'tiff', 'raw', 'webp', 'jfif', 'ico', 'psd', 'svg', 'ai']),
|
|
'Videos': set(['mp4', 'webm', 'mkv', 'MPG', 'MP2', 'MPEG', 'MPE', 'MPV', 'OGG', 'M4P', 'M4V', 'WMV', 'MOV', 'QT', 'FLV', 'SWF', 'AVCHD', 'avi', 'mpg', 'mpe', 'mpeg', 'asf', 'wmv', 'mov', 'qt', 'rm']),
|
|
'Documents': set(['txt', 'pdf', 'doc', 'xlsx', 'pdf', 'ppt', 'pps', 'docx', 'pptx']),
|
|
'Music': set(['mp3', 'wav', 'wma', 'mpa', 'ram', 'ra', 'aac', 'aif', 'm4a', 'tsa']),
|
|
'Torrents': set(['torrent']),
|
|
'Other': set([])
|
|
}
|
|
|
|
|
|
def create_folders():
|
|
'''Creates the required folders to organize files ('Pictures', 'Videos'..).
|
|
'''
|
|
for root in folder_ex:
|
|
try:
|
|
os.mkdir(root)
|
|
print(f'{root:20} Created ✔')
|
|
except OSError:
|
|
print(f'{root:20} Already Exists')
|
|
|
|
|
|
def get_folder(ext):
|
|
'''Returns the Folder that corresponds to the given extension.
|
|
|
|
Args:
|
|
ext (String): The extension of the file.
|
|
|
|
Returns:
|
|
String: The name of the Folder that holds the ext.
|
|
'''
|
|
for f, ex in folder_ex.items():
|
|
if ext in ex:
|
|
return f
|
|
return 'Other'
|
|
|
|
|
|
def start():
|
|
'''Organize files on the current directory, each to the corresponding folder.
|
|
'''
|
|
for filename in os.listdir():
|
|
# Check it's not filemover.py, a hidden file or a directory
|
|
if filename != __file__ and filename[0] != '.' and '.' in filename:
|
|
ext = os.path.basename(filename).split('.')[-1]
|
|
folder = get_folder(ext)
|
|
if not os.path.isfile(os.path.join(folder, filename)):
|
|
move(filename, folder)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
create_folders()
|
|
start() |