diff --git a/app.py b/app.py index ea2d963..1285b56 100644 --- a/app.py +++ b/app.py @@ -16,6 +16,8 @@ SQUID_PASSWD_FILE = 'config/squid_passwd' def init_db(): with sqlite3.connect(DATABASE) as conn: cursor = conn.cursor() + + # 创建admin_users表 cursor.execute(''' CREATE TABLE IF NOT EXISTS admin_users ( id INTEGER PRIMARY KEY AUTOINCREMENT, @@ -23,6 +25,8 @@ def init_db(): password TEXT NOT NULL ) ''') + + # 创建squid_users表 cursor.execute(''' CREATE TABLE IF NOT EXISTS squid_users ( id INTEGER PRIMARY KEY AUTOINCREMENT, @@ -31,6 +35,8 @@ def init_db(): is_active INTEGER DEFAULT 1 ) ''') + + # 创建settings表 cursor.execute(''' CREATE TABLE IF NOT EXISTS settings ( id INTEGER PRIMARY KEY DEFAULT 1, @@ -40,21 +46,23 @@ def init_db(): ) ''') - # 检查是否有管理员用户 + # 检查并初始化管理员用户 cursor.execute("SELECT COUNT(*) FROM admin_users") if cursor.fetchone()[0] == 0: cursor.execute( "INSERT INTO admin_users (username, password) VALUES (?, ?)", ('admin', generate_password_hash('admin123')) + ) - # 检查是否有设置 - cursor.execute("SELECT COUNT(*) FROM settings") - if cursor.fetchone()[0] == 0: - cursor.execute( - "INSERT INTO settings (proxy_address, proxy_port) VALUES (?, ?)", - ('proxy.example.com', '3128')) + # 检查并初始化设置 + cursor.execute("SELECT COUNT(*) FROM settings") + if cursor.fetchone()[0] == 0: + cursor.execute( + "INSERT INTO settings (proxy_address, proxy_port) VALUES (?, ?)", + ('proxy.example.com', '3128') + ) - conn.commit() + conn.commit() # 数据库连接