Spaces:
Paused
Paused
| import sqlite3 | |
| conn = sqlite3.connect('crowdata.db') | |
| cursor = conn.cursor() | |
| # Add MFA columns | |
| for col, col_type in [ | |
| ('mfa_enabled', 'BOOLEAN DEFAULT FALSE'), | |
| ('mfa_secret', 'VARCHAR(255)'), | |
| ('mfa_backup_codes', 'VARCHAR(500)'), | |
| ('mfa_verified_at', 'DATETIME') | |
| ]: | |
| try: | |
| cursor.execute(f'ALTER TABLE users ADD COLUMN {col} {col_type}') | |
| print(f'Added column: {col}') | |
| except Exception as e: | |
| if 'duplicate' in str(e).lower() or 'already exists' in str(e).lower(): | |
| print(f'Column already exists: {col}') | |
| else: | |
| print(f'Error adding {col}: {e}') | |
| conn.commit() | |
| conn.close() | |
| print('Migration done') |