generate_media_list.py

#!/usr/bin/env python3
import os
import argparse

def check_for_updates_if_enabled():
    """Check for updates in a non-intrusive way"""
    try:
        from update_checker import UpdateChecker
        from version import __version__
        
        checker = UpdateChecker()
        update_info = checker.check_for_updates()
        
        if not 'error' in update_info and update_info.get('update_available', False):
            print(f"📢 Media Manager update available: v{update_info['current_version']} → v{update_info['latest_version']}")
            print(f"   View release: {update_info['release_url']}")
            print()
    except ImportError:
        # Update checker not available, skip silently
        pass
    except Exception:
        # Any error checking updates should be silent to not interfere with main operation
        pass

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.lower().endswith(file_extensions):
                    media_files.append(os.path.join(root, file))
    
    with open(output_file, 'w', encoding='utf-8') as f:
        for file in media_files:
            f.write(f"{file}\n")

def main():
    parser = argparse.ArgumentParser(description="Generate a list of media files.")
    parser.add_argument('-d', '--directories', nargs='+', help='Directories to search for media files.', required=True)
    parser.add_argument('-o', '--output', help='Output file to write the list of media files.', required=True)
    parser.add_argument('--no-update-check', action='store_true', help='Skip checking for updates')
    args = parser.parse_args()

    # Check for updates unless disabled
    if not args.no_update_check:
        check_for_updates_if_enabled()

    generate_media_list(args.directories, args.output)

if __name__ == "__main__":
    main()