Update models.py

models.py

@@ -1,45 +1,45 @@
-import pymysql
-from config import Config
-
-class Database:
-    def __init__(self):
-        self.conn = pymysql.connect(
-            host=Config.MYSQL_CONFIG['host'],
-            port=Config.MYSQL_CONFIG['port'],
-            user=Config.MYSQL_CONFIG['user'],
-            password=Config.MYSQL_CONFIG['password'],
-            charset='utf8mb4',
-            cursorclass=pymysql.cursors.DictCursor
-        )
-        self.init_db()
-    
-    def init_db(self):
-        with self.conn.cursor() as cursor:
-            # 创建数据库
-            cursor.execute(f"CREATE DATABASE IF NOT EXISTS {Config.MYSQL_CONFIG['database']}")
-            cursor.execute(f"USE {Config.MYSQL_CONFIG['database']}")
-            
-            # 创建文件表
-            cursor.execute("""
-                CREATE TABLE IF NOT EXISTS files (
-                    id BIGINT PRIMARY KEY AUTO_INCREMENT,
-                    original_name VARCHAR(255) NOT NULL,
-                    stored_name VARCHAR(255) NOT NULL,
-                    file_path VARCHAR(500) NOT NULL,
-                    file_type VARCHAR(50),
-                    file_size BIGINT,
-                    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-                    UNIQUE KEY idx_path (file_path)
-                ) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci
-            """)
-        self.conn.commit()
-    
-    def search_files(self, keyword):
-        with self.conn.cursor() as cursor:
-            cursor.execute("""
-                SELECT * FROM files 
-                WHERE original_name LIKE %s 
-                OR file_path LIKE %s 
-                ORDER BY created_at DESC
-            """, (f"%{keyword}%", f"%{keyword}%"))
+import pymysql
+from config import Config
+
+class Database:
+    def __init__(self):
+        self.conn = pymysql.connect(
+            host=Config.MYSQL_CONFIG['host'],
+            port=Config.MYSQL_CONFIG['port'],
+            user=Config.MYSQL_CONFIG['user'],
+            password=Config.MYSQL_CONFIG['password'],
+            charset='utf8mb4',
+            cursorclass=pymysql.cursors.DictCursor
+        )
+        self.init_db()
+    
+    def init_db(self):
+        with self.conn.cursor() as cursor:
+            # 创建数据库
+            cursor.execute(f"CREATE DATABASE IF NOT EXISTS {Config.MYSQL_CONFIG['database']}")
+            cursor.execute(f"USE {Config.MYSQL_CONFIG['database']}")
+            
+            # 创建文件表
+            cursor.execute("""
+                CREATE TABLE IF NOT EXISTS files (
+                    id BIGINT PRIMARY KEY AUTO_INCREMENT,
+                    original_name VARCHAR(255) NOT NULL,
+                    stored_name VARCHAR(255) NOT NULL,
+                    file_path VARCHAR(500) NOT NULL,
+                    file_type VARCHAR(50),
+                    file_size BIGINT,
+                    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                    UNIQUE KEY idx_path (file_path)
+                ) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci
+            """)
+        self.conn.commit()
+    
+    def search_files(self, keyword):
+        with self.conn.cursor() as cursor:
+            cursor.execute("""
+                SELECT * FROM files 
+                WHERE original_name LIKE %s 
+                OR file_path LIKE %s 
+                ORDER BY created_at DESC
+            """, (f"%{keyword}%", f"%{keyword}%"))
             return cursor.fetchall()