Create ArrangeFiles.py

Simple scripts to arrange different files into respective folders

ArrangeFiles.py

@@ -0,0 +1,100 @@
+#!/usr/bin/python3
+import os
+import sys
+from pathlib import Path, PurePath
+import shutil
+
+FILE_EXTENSIONS = {
+    'Text Files': ['.txt', '.csv', '.json', '.xml', '.html'],
+    'Programming Files': ['.py', '.java', '.cpp', '.js', '.php'],
+    'Document Files': ['.docx', '.xlsx', '.pptx', '.pdf', '.odt'],
+    'Image Files': ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.svg'],
+    'Audio Files': ['.mp3', '.wav', '.aac', '.flac', '.m4a'],
+    'Video Files': ['.mp4', '.avi', '.mkv', '.mov', '.wmv'],
+    'Archive Files': ['.zip', '.rar', '.tar', '.gz', '.7z'],
+
+}
+
+
+# define method to create dummy files using the above
+
+def populate_folder_with_dummy_files():
+    path = Path("/home/eliasderby/Desktop/test")
+    for file_ext in FILE_EXTENSIONS.values():
+        for ext in file_ext:
+            file_name = path / f"test1{ext}"
+            file_name.touch()
+
+
+def arranger(path):
+    purified_path = PurePath(path)
+    transformed_path = Path(purified_path)
+
+    if not transformed_path.exists() or not transformed_path.is_dir():
+        print(f"Path '{path}' does not exist. or is not a directory")
+        return
+
+    counter = 0
+    # check if folders have been created , if not , create them
+    for file in transformed_path.iterdir():
+
+        if file.is_dir():
+            counter += 1
+            if file.name in FILE_EXTENSIONS:
+                continue
+            else:
+                directory_name = transformed_path / file.name
+
+                if not directory_name.exists():
+                    try:
+                        directory_name.mkdir()
+                    except FileExistsError:
+                        print(f"Folder '{directory_name}' already exists.")
+                else:
+                    print(f"Folder '{directory_name}' already exists.")
+
+        elif file.is_file():
+            counter += 1
+            suffix = file.suffix
+
+            # if path.is_file():
+            for folder_name, corresponding_file_extensions in FILE_EXTENSIONS.items():
+                if suffix in corresponding_file_extensions:
+                    folder = transformed_path / folder_name
+                    if not folder.exists():
+                        try:
+                            folder.mkdir()
+                        except FileExistsError:
+                            print(f"Folder '{folder}' already exists.")
+                    try:
+                        directory_name = transformed_path / file.name
+                        if directory_name.exists():
+                            shutil.copy2(str(file), str(folder))
+
+                            file.unlink()
+                        else:
+                            shutil.move(str(file), str(folder))
+
+                    except shutil.Error as error:
+                        print(f"Error moving file '{file}' to '{folder}': {error}")
+
+    if counter == 0: print("folder doesn't contain any files")
+
+
+def main():
+
+    # populate_folder_with_dummy_files()
+
+    if len(sys.argv) < 2:
+        # use "./"
+        print("pass in a path")
+        sys.exit(1)
+
+    folder_path = sys.argv[1:]
+
+    for elem in folder_path:
+        arranger(elem)
+
+
+if __name__ == '__main__':
+    main()