File size: 2,484 Bytes
c6393bd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 | """
migrate.py — Migrate accounts from accounts.json to SQLite
"""
import json
import time
from pathlib import Path
from database import get_db
DATA_DIR = Path(__file__).parent / "data"
ACCOUNTS_FILE = DATA_DIR / "accounts.json"
def migrate():
if not ACCOUNTS_FILE.exists():
print("No accounts.json found, skipping migration")
return 0
db = get_db()
# Check if already migrated
existing = db.fetchone("SELECT COUNT(*) as c FROM accounts")
if existing["c"] > 0:
print(f"Database already has {existing['c']} accounts, skipping migration")
return existing["c"]
raw = json.loads(ACCOUNTS_FILE.read_text(encoding="utf-8"))
count = 0
for item in raw:
db.execute(
"""INSERT OR IGNORE INTO accounts
(id, name, token, visitorid, enabled, email, password,
mail_provider, mail_api_key, created_at, quota_remain)
VALUES (?,?,?,?,?,?,?,?,?,?,?)""",
(
item.get("id", ""),
item.get("name", ""),
item.get("token", ""),
item.get("visitorid", ""),
1 if item.get("enabled", True) else 0,
item.get("email", ""),
item.get("password", ""),
item.get("mail_provider", ""),
item.get("mail_api_key", ""),
item.get("created_at", time.time()),
100, # default quota
)
)
count += 1
# Also migrate .env config values if they exist
import os
from dotenv import load_dotenv
load_dotenv()
env_mappings = {
"RITA_UPSTREAM": os.getenv("RITA_UPSTREAM", ""),
"RITA_ORIGIN": os.getenv("RITA_ORIGIN", ""),
"AUTH_TOKEN": os.getenv("AUTH_TOKEN", ""),
"DISABLE_SSL_VERIFY": os.getenv("DISABLE_SSL_VERIFY", ""),
"HEALTH_CHECK_INTERVAL": os.getenv("HEALTH_CHECK_INTERVAL", ""),
"AUTO_REGISTER_ENABLED": os.getenv("AUTO_REGISTER_ENABLED", ""),
"YESCAPTCHA_KEY": os.getenv("YESCAPTCHA_KEY", ""),
"GPTMAIL_API_KEY": os.getenv("GPTMAIL_API_KEY", ""),
"YYDSMAIL_API_KEY": os.getenv("YYDSMAIL_API_KEY", ""),
}
for key, val in env_mappings.items():
if val: # Only override if .env has a non-empty value
db.set_config(key, val)
print(f"Migrated {count} accounts from accounts.json to SQLite")
return count
if __name__ == "__main__":
migrate()
|