Merge pull request #18 from TrueBankai416/GUI

Add support for more file extensions

Author: TrueBankai416

email_utils.py

@@ -77,6 +77,10 @@ def load_email_config_from_file(config_file="media_manager_config.json"):
                 config = json.load(f)
                 email_config = config.get("email", {})
 
+                # Check if email is enabled
+                if not email_config.get("enabled", False):
+                    return None
+                
                 return EmailConfig(
                     sender_name=email_config.get("sender_name", ""),
                     sender_email=email_config.get("sender_email", ""),

example_config.json

@@ -5,7 +5,12 @@
         "D:\\Media\\Movies"
     ],
     "output_directory": "C:\\Users\\YourName\\Documents\\MediaLists",
+    "file_extensions": {
+        "media": [".mp4", ".mkv", ".avi"],
+        "additional": [".mov", ".wmv", ".flv", ".webm"]
+    },
     "email": {
+        "enabled": true,
         "sender_name": "Media Manager",
         "sender_email": "[email protected]",
         "receiver_email": "[email protected]",

generate_media_list.py

@@ -2,20 +2,28 @@
 import os
 import argparse
 
-def generate_media_list(directories, output_file):
+def generate_media_list(directories, output_file, file_extensions=None):
     """
     Generates a list of media files from the specified directories and writes them to the output file.
 
     Args:
     directories (list of str): Directories to search for media files.
     output_file (str): Path to the output file where the list will be saved.
+    file_extensions (list of str, optional): File extensions to search for. 
+                                           Defaults to ['.mp4', '.mkv', '.avi']
     """
+    if file_extensions is None:
+        file_extensions = ['.mp4', '.mkv', '.avi']
+    
+    # Convert to tuple for endswith() method
+    file_extensions = tuple(file_extensions)
+    
     media_files = []
     for directory in directories:
         for root, dirs, files in os.walk(directory, topdown=True):
             dirs[:] = [d for d in dirs if not d.startswith('.')]  # Ignore hidden directories
             for file in files:
-                if file.endswith(('.mp4', '.mkv', '.avi')):
+                if file.lower().endswith(file_extensions):
                     media_files.append(os.path.join(root, file))
 
     with open(output_file, 'w', encoding='utf-8') as f: