Initialize database tables
(self)
| 606 | print("Database migration check completed.") |
| 607 | |
| 608 | async def init_db(self): |
| 609 | """Initialize database tables""" |
| 610 | async with self._connect(write=True) as db: |
| 611 | await db.execute("PRAGMA journal_mode = WAL") |
| 612 | await db.execute("PRAGMA synchronous = NORMAL") |
| 613 | # Tokens table (Flow2API版本) |
| 614 | await db.execute(""" |
| 615 | CREATE TABLE IF NOT EXISTS tokens ( |
| 616 | id INTEGER PRIMARY KEY AUTOINCREMENT, |
| 617 | st TEXT UNIQUE NOT NULL, |
| 618 | at TEXT, |
| 619 | at_expires TIMESTAMP, |
| 620 | email TEXT NOT NULL, |
| 621 | name TEXT, |
| 622 | remark TEXT, |
| 623 | is_active BOOLEAN DEFAULT 1, |
| 624 | created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP, |
| 625 | last_used_at TIMESTAMP, |
| 626 | use_count INTEGER DEFAULT 0, |
| 627 | credits INTEGER DEFAULT 0, |
| 628 | user_paygate_tier TEXT, |
| 629 | current_project_id TEXT, |
| 630 | current_project_name TEXT, |
| 631 | image_enabled BOOLEAN DEFAULT 1, |
| 632 | video_enabled BOOLEAN DEFAULT 1, |
| 633 | image_concurrency INTEGER DEFAULT -1, |
| 634 | video_concurrency INTEGER DEFAULT -1, |
| 635 | captcha_proxy_url TEXT, |
| 636 | extension_route_key TEXT, |
| 637 | protocol_mode TEXT DEFAULT 'session', |
| 638 | google_cookies TEXT DEFAULT '', |
| 639 | login_account TEXT DEFAULT '', |
| 640 | login_password TEXT DEFAULT '', |
| 641 | proxy_url TEXT DEFAULT '', |
| 642 | auto_refresh_enabled BOOLEAN DEFAULT 1, |
| 643 | refresh_interval_minutes INTEGER DEFAULT 120, |
| 644 | last_st_refresh_at TIMESTAMP, |
| 645 | last_st_refresh_result TEXT DEFAULT '', |
| 646 | ban_reason TEXT, |
| 647 | banned_at TIMESTAMP |
| 648 | ) |
| 649 | """) |
| 650 | |
| 651 | # Projects table (新增) |
| 652 | await db.execute(""" |
| 653 | CREATE TABLE IF NOT EXISTS projects ( |
| 654 | id INTEGER PRIMARY KEY AUTOINCREMENT, |
| 655 | project_id TEXT UNIQUE NOT NULL, |
| 656 | token_id INTEGER NOT NULL, |
| 657 | project_name TEXT NOT NULL, |
| 658 | tool_name TEXT DEFAULT 'PINHOLE', |
| 659 | is_active BOOLEAN DEFAULT 1, |
| 660 | created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP, |
| 661 | FOREIGN KEY (token_id) REFERENCES tokens(id) |
| 662 | ) |
| 663 | """) |
| 664 | |
| 665 | # Token stats table |