#!/usr/bin/env python3
from app import app
from models import User, db
import sys

with app.app_context():
    if len(sys.argv) < 3:
        print("Usage: python reset_password.py <email> <new_password>")
        print("\nExisting users:")
        users = User.query.all()
        for user in users:
            print(f"  - {user.email} ({user.username}) - Role: {user.role}")
        sys.exit(1)
    
    email = sys.argv[1]
    new_password = sys.argv[2]
    
    user = User.query.filter_by(email=email).first()
    if user:
        user.set_password(new_password)
        db.session.commit()
        print(f"Password updated successfully for {user.email}")
        print(f"You can now login with:")
        print(f"  Email: {user.email}")
        print(f"  Password: {new_password}")
    else:
        print(f"User with email '{email}' not found")
        print("\nExisting users:")
        users = User.query.all()
        for user in users:
            print(f"  - {user.email} ({user.username})")