1.0.0

server/db.py

@@ -0,0 +1,30 @@
+import sqlite3
+from pathlib import Path
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+DB_FILE = os.getenv("DB_FILE")
+SQL_FILE = os.getenv("SQL_FILE")
+
+# 初始化数据库
+def init_db():
+    conn = sqlite3.connect(DB_FILE, timeout=30, check_same_thread=False)
+    conn.execute("PRAGMA journal_mode=WAL")
+    cur = conn.cursor()
+    sql = Path(SQL_FILE).read_text("utf-8")
+    cur.executescript(sql)
+    conn.commit()
+    conn.close()
+    print(f"数据库 {DB_FILE} 创建完成")
+
+# 获取连接
+def get_conn():
+    conn = sqlite3.connect(DB_FILE, timeout=30, check_same_thread=False)
+    conn.row_factory = sqlite3.Row
+    conn.execute("PRAGMA foreign_keys=ON")
+    return conn
+
+if __name__ == "__main__":
+    init_db()