Merge pull request #134 from nabroleonx/master

Add folder organizer script

scripts/Folder-organizer/README.md

@@ -0,0 +1,23 @@
+# Automated file organizer
+
+If you have that messed up folder structure with tons of different files all jumbled together, this is just the right script you need. It works for both linux and Windows.
+It will sort the same file types into same folders.
+
+It has two options:-
+
+    1). Organize all files in the directory.
+    2). Organize files of certain extensions in the directory.
+
+### Usage:
+
+It's really simple.
+
+    1). Run the code.
+    2). Enter the path you want to organize.
+    3). Choose the file extenstions you want to organize(i.e. ".exe", ".py", ".mp4", ".mp3"...).
+    4). Enter folder name.
+    5). There is no step 5. Enjoy your Organized folder and try not to make a mess for the future.
+
+Possible Improvements:
+
+> It can be further improved to organize files with certain keyword in their filename using regex. Even select only specific files not just using extensions.

scripts/Folder-organizer/main.py

@@ -0,0 +1,123 @@
+import os
+import shutil
+from sys import platform
+from threading import Thread
+
+organise_folder = download_path = None
+
+category = {"Audios": [".aif", ".cda", ".mid.mp3", ".mpa", ".ogg", ".wav", ".wma", ".wpl", ".midi"],
+            "Compressed": [".7z", ".arj", ".deb", ".pkg", ".rar", ".rpm", ".tar", ".z", ".zip", ".gz"],
+            "Documents": [".bin", ".dmg", ".iso", ".toast", ".vcd", ".csv", ".dat", ".db", ".log", ".mdb", ".sav",
+                          ".sql", ".tar", ".xml", ".dbf", ".email", ".eml", ".emlx", ".msg", ".oft", ".ost", ".pst",
+                          ".vcf", ".asp", ".cer", ".cfm", ".cgi", ".css", ".htm", ".js", ".jsp", ".part", ".php", ".py",
+                          ".rss", ".xhtml", ".fnt", ".fon", ".otf", ".ttf", ".doc", ".odt", ".pdf", ".rtf", ".tex",
+                          ".txt", ".wpd", ".key", ".odp", ".pps", ".ppt", ".pptx", ".c", ".cgi", ".class", ".cpp",
+                          ".cs", ".h", ".java", ".php", ".py", ".sh", ".swift", ".vb", ".ods", ".xls", ".xlsm", ".xlsx",
+                          ".docx", ".aspx", ".html"],
+            "Images": [".ai", ".bmp", ".gif", ".ico", ".jpeg", ".png", ".ps", ".psd", ".svg", ".tif", ".jpg", ".tiff"],
+            "Videos": [".3g2", ".3gp", ".avi", ".flv", ".h264", ".m4v", ".mkv", ".mov", ".mp4", ".mpg.rm", ".swf",
+                       ".vob", ".wmv", ".mpeg", ".webm"],
+            "Setups": [".apk", ".bat", ".bin", ".cgi", ".com", ".exe", ".gadget", ".jar", ".msi", ".py", ".wsf"],
+            "Systemfiles": [".bak", ".cab", ".cfg", ".cpl", ".cur", ".dll", ".dmp", ".drv", ".icns", ".ico", ".ini",
+                            ".lnk", ".msi",
+                            ".sys", ".tmp"]}
+
+
+def movers(source, destination):
+    if not os.path.exists(destination):
+        os.makedirs(destination)
+    try:
+        shutil.move(source, destination)
+    except OSError as error:
+        print(str(source) + " <= File is open. Error => ", error)
+
+
+def our_cat_dir(main_path, filepath):
+    for cat in category:
+        if filepath == os.path.join(main_path, cat):
+            return True
+    return False
+
+
+def our_ext_dir(main_path, filepath):
+    for cat in category:
+        for ext in category[cat]:
+            if filepath == os.path.join(main_path, ext):
+                return True
+    return False
+
+
+def org_by_cat(path):
+    for file in os.listdir(path):
+        file_path = os.path.join(path, file)
+        if os.path.isdir(file_path):
+            if not our_cat_dir(path, file_path):
+                Thread(target=org_by_cat, args=[file_path]).start()
+                continue
+        else:
+            file_name, file_extension = os.path.splitext(file_path)
+            for cat in category:
+                cat_folder = os.path.join(path, cat)
+                if file_extension.lower() in category[cat]:
+                    movers(file_path, cat_folder)
+
+
+def org_by_ext(path):
+    for file in os.listdir(path):
+        if os.path.isdir(file):
+            if not our_ext_dir(path, file):
+                Thread(target=org_by_ext, args=[file]).start()
+                continue
+        else:
+            file_path = os.path.join(path, file)
+            file_name, file_extension = os.path.splitext(file_path)
+            for cat in category:
+                if file_extension.lower() in category[cat]:
+                    ext_folder = os.path.join(path, file_extension)
+                    movers(file_path, ext_folder)
+
+
+def main():
+    global organise_folder, download_path
+    if platform == "linux" or platform == "linux2":
+        download_path = "/home/" + os.environ.get('USERNAME') + "/Downloads"
+    elif platform == "win32":
+        download_path = "C:\\Users\\" + os.environ.get('USERNAME') + "\\Downloads"
+    while True:
+        print("Press Q to exit anytime.")
+        path = input('''\tOrganise Custom Directory? Enter Path (default : Downloads)\n>>>''')
+        if path == "q":
+            break
+        elif path == "":
+            organise_folder = download_path
+        else:
+            organise_folder = path
+        if not os.path.exists(organise_folder):
+            print("\nInvalid Path.")
+            continue
+        else:
+            if os.path.isdir(organise_folder):
+                break
+            else:
+                print("\nEntered Path not a directory.")
+                continue
+    while True:
+        organise_type = input("\nIn what way do you want to organize.\n\t"
+                              + "1) Organize files by category.\n\t"
+                              + "2) Organize files by extension.\n>>>")
+        if organise_type == "q":
+            break
+        elif organise_type == "1":
+            print("Working")
+            org_by_cat(organise_folder)
+            print("Done")
+            break
+        elif organise_type == "2":
+            print("Working")
+            org_by_ext(organise_folder)
+            print("Done")
+            break
+
+
+if __name__ == "__main__":
+    main()