125 lines
4.2 KiB
Python
125 lines
4.2 KiB
Python
import sqlite3
|
|
import os
|
|
import secrets
|
|
|
|
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,
|
|
bandit_camp_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',
|
|
'ALTER TABLE farm_settings ADD COLUMN bandit_camp_enabled INTEGER NOT NULL DEFAULT 0',
|
|
]:
|
|
try:
|
|
c.execute(_col)
|
|
except Exception:
|
|
pass # column already exists
|
|
|
|
# Users — website admin accounts
|
|
c.execute('''
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
username TEXT NOT NULL UNIQUE,
|
|
password_hash TEXT NOT NULL,
|
|
created_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
)
|
|
''')
|
|
|
|
# Clans — groups owned by a user, identified by a unique clan_key
|
|
c.execute('''
|
|
CREATE TABLE IF NOT EXISTS clans (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
owner_id INTEGER NOT NULL REFERENCES users(id),
|
|
name TEXT NOT NULL,
|
|
clan_key TEXT NOT NULL UNIQUE,
|
|
created_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
)
|
|
''')
|
|
|
|
# Clan members — links Grepolis player_ids to a clan
|
|
c.execute('''
|
|
CREATE TABLE IF NOT EXISTS clan_members (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
clan_id INTEGER NOT NULL REFERENCES clans(id),
|
|
player_id TEXT NOT NULL,
|
|
player_name TEXT,
|
|
joined_at TEXT NOT NULL DEFAULT (datetime('now')),
|
|
UNIQUE(clan_id, player_id)
|
|
)
|
|
''')
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
|
|
def generate_clan_key():
|
|
"""Generate a short, unique, human-readable clan key."""
|
|
return secrets.token_urlsafe(8).upper()[:10]
|