21 lines
755 B
Python
21 lines
755 B
Python
import sqlite3
|
|
|
|
DB_PATH = "/home/matrixhasyou/legal-swarm-core/services/domovoy_bot/database/domovoy.db"
|
|
|
|
def migrate():
|
|
print(f"[*] Adding has_ai_access column to users...")
|
|
conn = sqlite3.connect(DB_PATH)
|
|
cursor = conn.cursor()
|
|
try:
|
|
cursor.execute("ALTER TABLE users ADD COLUMN has_ai_access BOOLEAN DEFAULT FALSE;")
|
|
print("[+] Column has_ai_access added.")
|
|
# Даем доступ админу (тебе) сразу
|
|
cursor.execute("UPDATE users SET has_ai_access = TRUE WHERE is_admin = TRUE;")
|
|
print("[+] AI access granted to all admins.")
|
|
except Exception as e:
|
|
print(f"[!] Error or already exists: {e}")
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
if __name__ == "__main__":
|
|
migrate()
|