add_user.py

#!/usr/bin/env python3

import sys
import os

# Add current directory to path so we can import app
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from app import app, db, User
from werkzeug.security import generate_password_hash

def create_test_user():
    with app.app_context():
        try:
            # Ensure tables exist
            db.create_all()
            
            # Check if user exists
            user = User.query.filter_by(username='testuser').first()
            if user:
                print('User already exists, updating password...')
                user.password_hash = generate_password_hash('testpass')
                db.session.commit()
            else:
                # Create user
                print('Creating new user...')
                user = User(username='testuser', email='test@example.com')
                user.password_hash = generate_password_hash('testpass')
                db.session.add(user)
                db.session.commit()
                print('Test user created successfully')
            
            # Verify the user
            user = User.query.filter_by(username='testuser').first()
            if user and user.check_password('testpass'):
                print('✓ User verification successful')
                return True
            else:
                print('✗ User verification failed')
                return False
                
        except Exception as e:
            print(f'Error: {e}')
            return False

if __name__ == '__main__':
    success = create_test_user()
    sys.exit(0 if success else 1)