# Author: OMKAR PATHAK

# This script helps to organize files

import os

from pathlib import Path

DIRECTORIES = {

"HTML": [".html5", ".html", ".htm", ".xhtml"],

"IMAGES": [".jpeg", ".jpg", ".tiff", ".gif", ".bmp", ".png", ".bpg", "svg",

".heif", ".psd"],

"VIDEOS": [".avi", ".flv", ".wmv", ".mov", ".mp4", ".webm", ".vob", ".mng",

".qt", ".mpg", ".mpeg", ".3gp", ".mkv"],

"DOCUMENTS": [".oxps", ".epub", ".pages", ".docx", ".doc", ".fdf", ".ods",

".odt", ".pwi", ".xsn", ".xps", ".dotx", ".docm", ".dox",

".rvg", ".rtf", ".rtfd", ".wpd", ".xls", ".xlsx", ".ppt",

"pptx"],

"ARCHIVES": [".a", ".ar", ".cpio", ".iso", ".tar", ".gz", ".rz", ".7z",

".dmg", ".rar", ".xar", ".zip"],

"AUDIO": [".aac", ".aa", ".aac", ".dvf", ".m4a", ".m4b", ".m4p", ".mp3",

".msv", "ogg", "oga", ".raw", ".vox", ".wav", ".wma"],

"PLAINTEXT": [".txt", ".in", ".out"],

"PDF": [".pdf"],

"PYTHON": [".py"],

"XML": [".xml"],

"EXE": [".exe"],

"SHELL": [".sh"]

}

FILE_FORMATS = {file_format: directory

for directory, file_formats in DIRECTORIES.items()

for file_format in file_formats}

def organize_junk():

for entry in os.scandir():

if entry.is_dir():

continue

file_path = Path(entry.name)

file_format = file_path.suffix.lower()

if file_format in FILE_FORMATS:

directory_path = Path(FILE_FORMATS[file_format])

directory_path.mkdir(exist_ok=True)

file_path.rename(directory_path.joinpath(file_path))

try:

os.mkdir("OTHER-FILES")

except:

pass

for dir in os.scandir():

try:

if dir.is_dir():

os.rmdir(dir)

else:

os.rename(os.getcwd() + '/' + str(Path(dir)), os.getcwd() + '/OTHER-FILES/' + str(Path(dir)))

except:

pass

if __name__ == "__main__":

organize_junk()