84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
import sqlite3
|
|
import os
|
|
|
|
DB_PATH = os.path.join(os.path.dirname(__file__), 'grepo.db')
|
|
|
|
|
|
def get_db():
|
|
conn = sqlite3.connect(DB_PATH)
|
|
conn.row_factory = sqlite3.Row
|
|
return conn
|
|
|
|
|
|
def init_db():
|
|
conn = get_db()
|
|
c = conn.cursor()
|
|
|
|
# Commands queue — sent from dashboard, consumed by Tampermonkey
|
|
c.execute('''
|
|
CREATE TABLE IF NOT EXISTS commands (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
town_id TEXT NOT NULL,
|
|
town_name TEXT,
|
|
type TEXT NOT NULL, -- 'build' | 'recruit'
|
|
payload TEXT NOT NULL, -- JSON string
|
|
status TEXT NOT NULL DEFAULT 'pending', -- pending | executing | done | failed
|
|
result_msg TEXT,
|
|
created_at TEXT NOT NULL DEFAULT (datetime('now')),
|
|
updated_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
)
|
|
''')
|
|
|
|
# Town state — pushed by Tampermonkey every poll cycle
|
|
c.execute('''
|
|
CREATE TABLE IF NOT EXISTS town_state (
|
|
town_id TEXT PRIMARY KEY,
|
|
town_name TEXT,
|
|
player TEXT,
|
|
player_id TEXT,
|
|
alliance_id TEXT,
|
|
world_id TEXT,
|
|
x REAL,
|
|
y REAL,
|
|
sea INTEGER,
|
|
data TEXT NOT NULL, -- full JSON snapshot
|
|
updated_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
)
|
|
''')
|
|
|
|
# Key-value store — generic flags (e.g. captcha_active)
|
|
c.execute('''
|
|
CREATE TABLE IF NOT EXISTS kv_store (
|
|
key TEXT PRIMARY KEY,
|
|
value TEXT,
|
|
updated_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
)
|
|
''')
|
|
|
|
# Farm settings — per-player auto-farm configuration
|
|
c.execute('''
|
|
CREATE TABLE IF NOT EXISTS farm_settings (
|
|
player_id TEXT PRIMARY KEY,
|
|
enabled INTEGER NOT NULL DEFAULT 0,
|
|
loot_option INTEGER NOT NULL DEFAULT 1, -- 1=5min, 2=20min, 3=90min, 4=4h
|
|
updated_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
)
|
|
''')
|
|
|
|
# Migration: add new columns if upgrading an existing database
|
|
for _col in [
|
|
'ALTER TABLE town_state ADD COLUMN player_id TEXT',
|
|
'ALTER TABLE town_state ADD COLUMN alliance_id TEXT',
|
|
'ALTER TABLE town_state ADD COLUMN x REAL',
|
|
'ALTER TABLE town_state ADD COLUMN y REAL',
|
|
'ALTER TABLE town_state ADD COLUMN sea INTEGER',
|
|
'ALTER TABLE commands ADD COLUMN player_id TEXT',
|
|
]:
|
|
try:
|
|
c.execute(_col)
|
|
except Exception:
|
|
pass # column already exists
|
|
|
|
conn.commit()
|
|
conn.close()
|