Organise.py

from __future__ import print_function

import os
import shutil
import sys

EXT_VIDEO_LIST = ["FLV", "WMV", "MOV", "MP4", "MPEG", "3GP", "MKV", "AVI"]
EXT_IMAGE_LIST = ["JPG", "JPEG", "GIF", "PNG", "SVG"]
EXT_DOCUMENT_LIST = [
    "DOC",
    "DOCX",
    "PPT",
    "PPTX",
    "PAGES",
    "PDF",
    "ODT",
    "ODP",
    "XLSX",
    "XLS",
    "ODS",
    "TXT",
    "IN",
    "OUT",
    "MD",
]
EXT_MUSIC_LIST = ["MP3", "WAV", "WMA", "MKA", "AAC", "MID", "RA", "RAM", "RM", "OGG"]
EXT_CODE_LIST = ["CPP", "RB", "PY", "HTML", "CSS", "JS"]
EXT_EXECUTABLE_LIST = ["LNK", "DEB", "EXE", "SH", "BUNDLE"]
EXT_COMPRESSED_LIST = [
    "RAR",
    "JAR",
    "ZIP",
    "TAR",
    "MAR",
    "ISO",
    "LZ",
    "7ZIP",
    "TGZ",
    "GZ",
    "BZ2",
]

# Taking the location of the Folder to Arrange
try:
    destLocation = str(sys.argv[1])
except IndexError:
    destLocation = str(input("Enter the Path of directory: "))


# When we make a folder that already exist then WindowsError happen
# Changing directory may give WindowsError
def ChangeDirectory(dir):
    try:
        os.chdir(dir)
    except WindowsError:
        print("Error! Cannot change the Directory")
        print("Enter a valid directory!")
        ChangeDirectory(str(input("Enter the Path of directory: ")))


ChangeDirectory(destLocation)


def Organize(dirs, name):
    try:
        os.mkdir(name)
        print("{} Folder Created".format(name))
    except WindowsError:
        print("{} Folder Exist".format(name))

    src = "{}\\{}".format(destLocation, dirs)
    dest = "{}\{}".format(destLocation, name)

    os.chdir(dest)
    shutil.move(src, "{}\\{}".format(dest, dirs))

    print(os.getcwd())
    os.chdir(destLocation)


TYPES_LIST = [
    "Video",
    "Images",
    "Documents",
    "Music",
    "Codes",
    "Executables",
    "Compressed",
]
for dirs in os.listdir(os.getcwd()):
    if 1:
        for name, extensions_list in zip(
            TYPES_LIST,
            [
                EXT_VIDEO_LIST,
                EXT_IMAGE_LIST,
                EXT_DOCUMENT_LIST,
                EXT_MUSIC_LIST,
                EXT_CODE_LIST,
                EXT_EXECUTABLE_LIST,
                EXT_COMPRESSED_LIST,
            ],
        ):
            if dirs.split(".")[-1].upper() in extensions_list:
                Organize(dirs, name)
    else:
        if dirs not in TYPES_LIST:
            Organize(dirs, "Folders")

print("Done Arranging Files and Folder in your specified directory")