added a save feature using json

Author: CodeCraftPlugin

.gitignore

@@ -1,3 +1,4 @@
 .venv/
 .vscode/
-build_windows/
+build_windows/
+__pycache__/

auto_download_cleaner.py

@@ -1,30 +1,37 @@
 import os
 import shutil
 import variable as var
+import save as sv
 import tray_icon as ti
 import time as time
-def move_the_file_to_right_place(main_directory, move_folder, filename):
+def move_file(main_directory, move_folder, filename):
     if not os.path.exists(move_folder):
         os.makedirs(move_folder)
     shutil.move(os.path.join(main_directory, filename), os.path.join(move_folder, filename))
-def check_file_extension(file_ext, dir, mov_dir, filename):
+def check_file_extension(file_ext, mov_dir, filename):
     if filename.endswith(file_ext):
         time.sleep(1)
-        move_the_file_to_right_place(main_directory=dir, move_folder=mov_dir, filename=filename)
+        move_file(main_directory=sv.get_default_download_folder(), move_folder=mov_dir, filename=filename)
 def main_stuff():
+    loaded_vars = sv.load_variables_from_json()
+    print(loaded_vars)
+    print(sv.get_default_download_folder())
+    print(loaded_vars['DOCX'])
     print('Program started.')
     while var.run:
-        for filename in os.listdir(var.directory):
-            check_file_extension(".pdf", dir=var.directory, mov_dir=var.PDF, filename=filename)
-            check_file_extension(".docx", dir=var.directory, mov_dir=var.DOCX, filename=filename)
-            check_file_extension('.zip', dir=var.directory, mov_dir=var.ZIP, filename=filename)
-            check_file_extension(tuple(var.Image), dir=var.directory, mov_dir=var.IMAGE, filename=filename)
-            check_file_extension('.py', dir=var.directory, mov_dir=var.PYTHON, filename=filename)
-            check_file_extension(tuple(var.Video), dir=var.directory, mov_dir=var.VIDEO, filename=filename)
-            check_file_extension(tuple(var.audio_formats), dir=var.directory, mov_dir=var.MUSIC, filename=filename)
-            check_file_extension('.torrent', dir=var.directory, mov_dir=var.TORRENT, filename=filename)
-
-
-if __name__ == '__main__':
+        for filename in os.listdir(sv.get_default_download_folder()):
+            try:
+                check_file_extension(".pdf",  mov_dir=loaded_vars[var.PDF], filename=filename)
+                check_file_extension(".docx",  mov_dir=loaded_vars[var.DOCX], filename=filename)
+                check_file_extension('.zip',  mov_dir=loaded_vars[var.ZIP], filename=filename)
+                check_file_extension(var.Image,  mov_dir=loaded_vars[var.IMAGE], filename=filename)
+                check_file_extension('.py',  mov_dir=loaded_vars[var.Python], filename=filename)
+                check_file_extension(var.Video,  mov_dir=loaded_vars[var.VIDEO], filename=filename)
+                check_file_extension(var.audio_formats,  mov_dir=loaded_vars[var.MUSIC], filename=filename)
+                check_file_extension('.torrent',  mov_dir=loaded_vars[var.Torrent], filename=filename)
+            except PermissionError:
+                continue
+if __name__ == "__main__":
+    sv.main()
     ti.start()
-    main_stuff()
+    main_stuff() 

config.json

@@ -0,0 +1,10 @@
+{
+    "PDF": "C:\\Users\\CodeCraft\\Documents\\PDF",
+    "DOCX": "C:\\Users\\CodeCraft\\Documents\\DOCX",
+    "ZIP": "D:\\Zip",
+    "Image": "C:\\Users\\CodeCraft\\Pictures\\Photos",
+    "Python": "D:\\Python",
+    "Video": "C:\\Users\\CodeCraft\\Videos",
+    "Music": "C:\\Users\\CodeCraft\\Music",
+    "Torrent": "C:\\Users\\CodeCraft\\Downloads\\Torrents"
+}

save.py

@@ -0,0 +1,70 @@
+import os
+import json
+import getpass
+import variable
+def get_default_download_folder():
+
+    # try to automatically detect the download folder
+    default_download_folder = None
+
+    # Platform-specific logic to detect the download folder
+    if os.name == "posix":  # Linux/Mac
+        default_download_folder = os.path.expanduser("~/Downloads")
+    elif os.name == "nt":  # Windows
+        default_download_folder = os.path.join(os.path.expanduser("~"), "Downloads")
+    else:
+        print("Unsupported operating system.")
+        exit(1)
+
+    return default_download_folder
+
+def create_default_directories():
+    # If username is "codecraft", use the hardcoded download folder
+    username = getpass.getuser()
+    if username=="CodeCraft":
+        default_dirs = default_dirs = {
+        "PDF": os.path.join(os.path.expanduser("~"),"Documents", "PDF"),
+        "DOCX": os.path.join(os.path.expanduser("~"),"Documents", "DOCX"),
+        "ZIP": variable.ZIP_DIR,
+        "Image": os.path.join(os.path.expanduser("~"),"Pictures", "Photos"),
+        "Python": variable.PYTHON_DIR,
+        "Video": os.path.join(os.path.expanduser("~"),"Videos"),
+        "Music": os.path.join(os.path.expanduser("~"), "Music"),
+        "Torrent": os.path.join(get_default_download_folder(), "Torrents")
+        }
+        return default_dirs
+    # Define default directories for different file types
+    default_dirs = {
+        "PDF": os.path.join(os.path.expanduser("~"),"Documents", "PDF"),
+        "DOCX": os.path.join(os.path.expanduser("~"),"Documents", "DOCX"),
+        "ZIP": os.path.join(get_default_download_folder(), "ZIP"),
+        "Image": os.path.join(os.path.expanduser("~"),"Pictures", "Photos"),
+        "Python": os.path.join(os.path.expanduser("~"),"Code", "Python"),
+        "Video": os.path.join(os.path.expanduser("~"),"Videos"),
+        "Music": os.path.join(os.path.expanduser("~"), "Music"),
+        "Torrent": os.path.join(get_default_download_folder(), "Torrents")
+    }
+
+    return default_dirs
+
+def save_variables_to_json(vars_dict):
+    with open("config.json", "w") as json_file:
+        json.dump(vars_dict, json_file, indent=4)
+def load_variables_from_json():
+    dirname = os.path.dirname(__file__)
+    filename = os.path.join(dirname, 'config.json')
+    try:
+        with open(filename, "r") as json_file:
+            loaded_vars = json.load(json_file)
+    except FileNotFoundError:
+        # If the config file doesn't exist, return an empty dictionary
+        return {}
+    return loaded_vars
+def main():
+    username = getpass.getuser()
+    # Create default directories for different file types
+    default_directories = create_default_directories()
+
+    # Save the variables to a JSON file
+    if (os.path.exists("config.json"))==False:
+        save_variables_to_json(default_directories)     

setup.py

@@ -3,12 +3,12 @@
 
 # Dependencies are automatically detected, but it might need
 # fine tuning.
-build_options = {'packages': [], 'excludes': [],'build_exe': 'build_windows','include_files':['icon.ico']}
+build_options = {'packages': ['os','getpass'], 'excludes': [],'build_exe': 'build_windows','include_files':['icon.ico']}
 
-base = 'Win32GUI' if sys.platform=='win32' else None
+base = "Console"
 
 executables = [
-    Executable('auto_download_cleaner.py', base=base, target_name = 'Auto-Download-Cleanup',icon='icon.ico')
+    Executable('auto_downloader_clearner.py', base=base, target_name = 'Auto-Download-Cleanup',icon='icon.ico')
 ]
 
 setup(name='Auto-Download-Cleanup',

tray_icon.py

@@ -10,6 +10,6 @@ def on_click(icon, item):
         icon.stop()
         vars.run = False
 
-icon = pystray.Icon("name", image, "title", menu=pystray.Menu(pystray.MenuItem("Quit", on_click)))
+icon = pystray.Icon("name", image, "Auto Download Cleanup", menu=pystray.Menu(pystray.MenuItem("Quit", on_click)))
 def start():
     icon.run_detached()

variable.py

@@ -1,14 +1,14 @@
 run = True
-
-directory = 'C:\\Users\\CodeCraft\\Downloads'
-PDF = 'C:\\Users\\CodeCraft\\Documents\\PDF'
-DOCX = 'C:\\Users\\CodeCraft\\Documents\\DOC'
-TORRENT = 'C:\\Users\\CodeCraft\\Downloads\\TORRENT'
-ZIP = 'D:\\Zip'
-Image = ['.png', '.jpg', '.jpeg', '.gif', '.tiff', '.psd', '.raw', '.bmp', '.heif', '.indd', '.svg', '.ai', '.eps', '.pdf', '.ico', '.webp']
-IMAGE = 'C:\\Users\\CodeCraft\\Pictures\\Photos'
-PYTHON = 'D:\\Python'
-VIDEO = 'C:\\Users\\CodeCraft\\Videos'
-MUSIC = 'C:\\Users\\CodeCraft\\Music'
-Video = ['.mp4', '.mov', '.wmv', '.avi', '.avchd', '.flv', '.f4v', '.swf', '.mkv', '.webm', '.mng', '.gifv', '.mpg', '.mp2', '.mpeg', '.mpe', '.mpv', '.ogg', '.m4p', '.m4v', '.wmv', '.mov', '.qt', '.flv', '.swf', '.avchd']
+ZIP_DIR = 'D:\\Zip'
+Image = ('.png', '.jpg', '.jpeg', '.gif', '.tiff', '.psd', '.raw', '.bmp', '.heif', '.indd', '.svg', '.ai', '.eps', '.ico', '.webp')
+PYTHON_DIR = 'D:\\Python'
+Video = ('.mp4', '.mov', '.wmv', '.avi', '.avchd', '.flv', '.f4v', '.swf', '.mkv', '.webm', '.mng', '.gifv', '.mpg', '.mp2', '.mpeg', '.mpe', '.mpv', '.ogg', '.m4p', '.m4v', '.wmv', '.mov', '.qt', '.flv', '.swf', '.avchd')
 audio_formats = ('.mp3', '.wav', '.aiff', '.flac', '.aac', '.ogg', '.wma', '.m4a', '.ape', '.alac', '.dsd')
+PDF = "PDF"
+DOCX = "DOCX"
+ZIP = "ZIP"
+IMAGE = "Image"
+Python = "Python"
+VIDEO = "Video"
+MUSIC = "Music"
+Torrent = "Torrent"