first commit

models/connect_db.py

@@ -0,0 +1,88 @@
+import toml
+import pymysql
+from pymysql.cursors import DictCursor
+from pymysql.connections import Connection
+from models.logger import setup_logger
+
+logger = setup_logger()
+
+def connect_db() -> Connection | None:
+    """连接数据库，如果没有则初始化数据库和表"""
+    try:
+        config = toml.load("../config.toml")
+        db_host = config["db_host"]
+        db_port = config["db_port"]
+        db_user = config["db_user"]
+        db_password = config["db_password"]
+        db_database = config["db_database"]
+    except Exception as e:
+        logger.error(f"读取数据库配置错误{e}")
+        return None
+
+    connection = pymysql.connect(
+        host=db_host,
+        port=int(db_port),
+        user=db_user,
+        password=db_password,
+        database=db_database,
+        charset='utf8mb4',
+        cursorclass=DictCursor
+    )
+
+    try:
+        with connection.cursor() as cursor:
+            # 创建 emoji 表
+            cursor.execute('''
+                        CREATE TABLE IF NOT EXISTS emoji (
+                            id INT AUTO_INCREMENT PRIMARY KEY,
+                            uuid VARCHAR(255) NOT NULL,
+                            description VARCHAR(255),
+                            url VARCHAR(255) NOT NULL,
+                            hash VARCHAR(255) NOT NULL
+                        )
+                ''')
+
+            # 创建 approved 表
+            cursor.execute('''
+                           CREATE TABLE IF NOT EXISTS approved
+                           (
+                               id          INT AUTO_INCREMENT PRIMARY KEY,
+                               uuid        VARCHAR(255) NOT NULL,
+                               approved    BOOL,
+                               approver    VARCHAR(255)
+                           )
+                           ''')
+
+            # 创建 token 表
+            cursor.execute('''
+                           CREATE TABLE IF NOT EXISTS token
+                           (
+                               id INT AUTO_INCREMENT PRIMARY KEY,
+                               token VARCHAR(255) NOT NULL ,
+                               userid INT NOT NULL ,
+                               expires TIMESTAMP NOT NULL
+                           )
+                           ''')
+
+            # 创建 user 表
+            cursor.execute('''
+                           CREATE TABLE IF NOT EXISTS user
+                           (
+                               id INT AUTO_INCREMENT PRIMARY KEY,
+                               username VARCHAR(255) NOT NULL,
+                               passwd VARCHAR(255) NOT NULL,
+                               email VARCHAR(255),
+                               lastlogin TIMESTAMP,
+                               submission_count INT DEFAULT 0
+                           )
+                           ''')
+
+        return connection
+    except Exception as e:
+        logger.error(f"数据库连接或表创建失败: {e}")
+        return None
+
+
+if __name__ == "__main__":
+    conn = connect_db()
+    conn.cursor()