up

backend/database.py

@@ -0,0 +1,65 @@
+import sqlite3
+import os
+
+DB_PATH = os.path.join(os.path.dirname(__file__), "stock_data.db")
+
+SCHEMA_SQL = """
+CREATE TABLE IF NOT EXISTS stock_collections (
+  id INTEGER PRIMARY KEY AUTOINCREMENT,
+  name TEXT NOT NULL,
+  description TEXT,
+  user_id TEXT NOT NULL DEFAULT '',
+  last_accessed_at TEXT NOT NULL DEFAULT (datetime('now','localtime')),
+  created_at TEXT NOT NULL DEFAULT (datetime('now','localtime')),
+  updated_at TEXT NOT NULL DEFAULT (datetime('now','localtime'))
+);
+
+CREATE TABLE IF NOT EXISTS collection_stocks (
+  id INTEGER PRIMARY KEY AUTOINCREMENT,
+  collection_id INTEGER NOT NULL REFERENCES stock_collections(id) ON DELETE CASCADE,
+  stock_code TEXT NOT NULL,
+  stock_name TEXT NOT NULL,
+  added_price REAL,
+  added_at TEXT NOT NULL DEFAULT (datetime('now','localtime')),
+  UNIQUE(collection_id, stock_code)
+);
+
+CREATE TABLE IF NOT EXISTS share_links (
+  id INTEGER PRIMARY KEY AUTOINCREMENT,
+  collection_id INTEGER NOT NULL REFERENCES stock_collections(id) ON DELETE CASCADE,
+  short_code TEXT NOT NULL UNIQUE,
+  created_at TEXT NOT NULL DEFAULT (datetime('now','localtime')),
+  expires_at TEXT
+);
+
+CREATE TABLE IF NOT EXISTS cache (
+  key TEXT PRIMARY KEY,
+  value TEXT NOT NULL,
+  expires_at TEXT NOT NULL
+);
+"""
+
+
+def get_connection() -> sqlite3.Connection:
+    conn = sqlite3.connect(DB_PATH)
+    conn.row_factory = sqlite3.Row
+    conn.execute("PRAGMA foreign_keys = ON")
+    return conn
+
+
+def init_db():
+    conn = get_connection()
+    conn.executescript(SCHEMA_SQL)
+
+    # 清理过期缓存
+    from services.cache import clean_expired
+    clean_expired()
+
+    conn.commit()
+    conn.close()
+
+
+def dict_from_row(row: sqlite3.Row) -> dict:
+    if row is None:
+        return None
+    return dict(row)