#!/usr/bin/env python3
"""
Cleanup script for Flask applications
Removes __pycache__ directories, .pyc files, and database files
"""

import os
import shutil
import argparse

def cleanup(directory, verbose=False):
    """Clean up cache files and database files"""
    cleaned_dirs = 0
    cleaned_files = 0
    
    # Files to clean
    file_patterns = ['.pyc', '.pyo', '.~', '.swp', '.swo']
    db_patterns = ['.db', '.sqlite', '.sqlite3', '-journal']
    
    # Directories to clean
    dir_patterns = ['__pycache__', '.pytest_cache', '.coverage', 'htmlcov']
    
    # Clean main directory
    for root, dirs, files in os.walk(directory):
        # Clean directories
        for dir_name in list(dirs):
            if dir_name in dir_patterns:
                dir_path = os.path.join(root, dir_name)
                if verbose:
                    print(f"Removing directory: {dir_path}")
                shutil.rmtree(dir_path)
                cleaned_dirs += 1
                dirs.remove(dir_name)
        
        # Clean files
        for file in files:
            if any(file.endswith(pattern) for pattern in file_patterns + db_patterns):
                file_path = os.path.join(root, file)
                if verbose:
                    print(f"Removing file: {file_path}")
                os.remove(file_path)
                cleaned_files += 1
    
    # Clean instance directory
    instance_dir = os.path.join(directory, 'instance')
    if os.path.exists(instance_dir):
        for file in os.listdir(instance_dir):
            if any(file.endswith(pattern) for pattern in db_patterns):
                file_path = os.path.join(instance_dir, file)
                if verbose:
                    print(f"Removing database file: {file_path}")
                os.remove(file_path)
                cleaned_files += 1
    
    print(f"Cleanup completed! Removed {cleaned_dirs} directories and {cleaned_files} files.")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Clean up Flask application cache and database files")
    parser.add_argument("-v", "--verbose", action="store_true", help="Show detailed output")
    parser.add_argument("-d", "--directory", default=".", help="Directory to clean (default: current directory)")
    
    args = parser.parse_args()
    cleanup(args.directory, args.verbose)