新版本重构了ai模型提供商逻辑,新增MCUNC,阿里云百炼ai提供商,支持语言可配置,优化日志输出格式
This commit is contained in:
42
model/sql_tools.py
Normal file
42
model/sql_tools.py
Normal file
@@ -0,0 +1,42 @@
|
||||
import sqlite3
|
||||
from model.logger import setup_logger
|
||||
|
||||
|
||||
|
||||
logger = setup_logger()
|
||||
def init_db():
|
||||
try:
|
||||
with sqlite3.connect("uuid.db") as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('''
|
||||
CREATE TABLE IF NOT EXISTS groups
|
||||
(
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
group_openid TEXT UNIQUE NOT NULL,
|
||||
uuid TEXT NOT NULL
|
||||
)
|
||||
''')
|
||||
conn.commit()
|
||||
except sqlite3.Error as e:
|
||||
logger.error(f"数据库初始化失败: {e}")
|
||||
|
||||
|
||||
def get_uuid(group_openid):
|
||||
try:
|
||||
with sqlite3.connect("uuid.db") as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT uuid FROM groups WHERE group_openid = ?", (group_openid,))
|
||||
result = cursor.fetchone()
|
||||
return result[0] if result else ""
|
||||
except sqlite3.Error as e:
|
||||
logger.error(f"数据库查询错误: {e}")
|
||||
return ""
|
||||
|
||||
def add_uuid(group_openid, uuid):
|
||||
try:
|
||||
with sqlite3.connect("uuid.db") as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("INSERT INTO groups (group_openid, uuid) VALUES (?, ?)",
|
||||
(group_openid, uuid))
|
||||
except sqlite3.Error as e:
|
||||
logger.error(f"数据库插入错误: {e}")
|
||||
Reference in New Issue
Block a user