增加新功能

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()
 
             # 数据库连接