reset_database.py

"""
Reset Database Helper Script

Use this to delete the database and start fresh if you had import issues
or want to re-import all transactions with different settings.

Usage:
    python reset_database.py
    
Or on Windows:
    reset_database.bat

WARNING: This will permanently delete all imported transactions!
"""
import os
import sys


def main() -> None:
    """
    Main function to reset the transaction database
    
    Prompts the user for confirmation before deleting transactions.db
    """
    db_file = "transactions.db"
    
    print("=" * 70)
    print("Reset Transaction Database")
    print("=" * 70)
    print()
    print("⚠️  WARNING: This will DELETE all imported transactions!")
    print()
    
    if not os.path.exists(db_file):
        print(f"✓ Database file '{db_file}' does not exist.")
        print("  No action needed. Just run transaction_manager_gui.py to start fresh.")
        print()
        return
    
    # Get file size
    size_bytes = os.path.getsize(db_file)
    size_kb = size_bytes / 1024
    
    print(f"Database file found: {db_file}")
    print(f"Size: {size_kb:.2f} KB")
    print()
    print("This will:")
    print("  1. Delete the existing database")
    print("  2. You'll need to re-import all your CSV files")
    print("  3. Categories will be recreated automatically")
    print()
    
    response = input("Are you sure you want to DELETE the database? (yes/no): ").strip().lower()
    
    if response == 'yes':
        try:
            os.remove(db_file)
            print()
            print("✓ Database deleted successfully!")
            print()
            print("Next steps:")
            print("  1. Run: python transaction_manager_gui.py")
            print("  2. Database will be auto-created")
            print("  3. Import your CSV files with correct hash settings")
            print()
        except Exception as e:
            print()
            print(f"✗ Error deleting database: {e}")
            print()
            sys.exit(1)
    else:
        print()
        print("Cancelled. No changes made.")
        print()


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        print("\n\nCancelled by user.")
        sys.exit(0)