| #!/usr/bin/env python3 | |
| """ | |
| One-time script to set is_admin flag for existing admin user(s) | |
| Run this after adding the is_admin column to the database | |
| """ | |
| import sys | |
| import os | |
| # Add parent directory to path so we can import app | |
| sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | |
| from app import db | |
| from models import User | |
| def main(): | |
| print("Setting admin flag for existing admin users...") | |
| # Find the admin user | |
| admin_user = User.query.filter_by(username='admin').first() | |
| if admin_user: | |
| admin_user.is_admin = True | |
| db.session.commit() | |
| print(f"Updated user {admin_user.username} (ID: {admin_user.id}) - admin flag set to True") | |
| else: | |
| print("Warning: No 'admin' user found in the database") | |
| print("Done!") | |
| if __name__ == "__main__": | |
| main() | |