add sqlite

This commit is contained in:
朱潮 2025-10-28 20:54:15 +08:00
parent 5b27079318
commit 06544ce321
4 changed files with 15 additions and 15 deletions

View File

@ -71,7 +71,7 @@ class EnhancedSurveyHandler(BaseHTTPRequestHandler):
# 创建学员记录 # 创建学员记录
student_id = str(uuid.uuid4()) student_id = str(uuid.uuid4())
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute(''' cursor.execute('''
@ -115,7 +115,7 @@ class EnhancedSurveyHandler(BaseHTTPRequestHandler):
return return
# 保存答题结果 - JSON格式存储包含用户信息 # 保存答题结果 - JSON格式存储包含用户信息
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
conn.row_factory = sqlite3.Row # 设置行工厂以支持字典访问 conn.row_factory = sqlite3.Row # 设置行工厂以支持字典访问
cursor = conn.cursor() cursor = conn.cursor()
@ -380,7 +380,7 @@ class EnhancedSurveyHandler(BaseHTTPRequestHandler):
"""根据会话ID获取筛选后的题目""" """根据会话ID获取筛选后的题目"""
try: try:
# 从数据库获取会话信息,包括选择的标签和题目配置 # 从数据库获取会话信息,包括选择的标签和题目配置
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
conn.row_factory = sqlite3.Row conn.row_factory = sqlite3.Row
cursor = conn.cursor() cursor = conn.cursor()
@ -575,7 +575,7 @@ class EnhancedSurveyHandler(BaseHTTPRequestHandler):
"""处理答题页面""" """处理答题页面"""
session_id = path.split('/')[-1] session_id = path.split('/')[-1]
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
conn.row_factory = sqlite3.Row conn.row_factory = sqlite3.Row
cursor = conn.cursor() cursor = conn.cursor()

View File

@ -73,7 +73,7 @@ class ReportGenerator:
def generate_analysis_text(self, session_data): def generate_analysis_text(self, session_data):
"""生成答题情况分析文本""" """生成答题情况分析文本"""
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
conn.row_factory = sqlite3.Row conn.row_factory = sqlite3.Row
cursor = conn.cursor() cursor = conn.cursor()
@ -302,7 +302,7 @@ class ReportGenerator:
def save_report_to_db(self, session_id, report_data, analysis_data): def save_report_to_db(self, session_id, report_data, analysis_data):
"""保存报告到数据库""" """保存报告到数据库"""
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
cursor = conn.cursor() cursor = conn.cursor()
# 创建报告表(如果不存在) # 创建报告表(如果不存在)
@ -338,7 +338,7 @@ class ReportGenerator:
def save_analysis_data_for_regeneration(self, session_id, analysis_data): def save_analysis_data_for_regeneration(self, session_id, analysis_data):
"""保存分析数据以便重新生成报告""" """保存分析数据以便重新生成报告"""
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
cursor = conn.cursor() cursor = conn.cursor()
# 创建临时分析数据表(如果不存在) # 创建临时分析数据表(如果不存在)
@ -382,7 +382,7 @@ class EnhancedSurveySystem:
def init_database(self): def init_database(self):
"""初始化数据库""" """初始化数据库"""
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
cursor = conn.cursor() cursor = conn.cursor()
# 确保必要的表存在 # 确保必要的表存在
@ -402,7 +402,7 @@ class EnhancedSurveySystem:
def get_reports_list(self, page=1, page_size=10): def get_reports_list(self, page=1, page_size=10):
"""获取报告列表""" """获取报告列表"""
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
conn.row_factory = sqlite3.Row conn.row_factory = sqlite3.Row
cursor = conn.cursor() cursor = conn.cursor()
@ -435,7 +435,7 @@ class EnhancedSurveySystem:
def get_report_by_id(self, report_id): def get_report_by_id(self, report_id):
"""根据ID获取报告""" """根据ID获取报告"""
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
conn.row_factory = sqlite3.Row conn.row_factory = sqlite3.Row
cursor = conn.cursor() cursor = conn.cursor()
@ -454,7 +454,7 @@ class EnhancedSurveySystem:
def get_sessions_can_regenerate(self): def get_sessions_can_regenerate(self):
"""获取可以重新生成的会话列表""" """获取可以重新生成的会话列表"""
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
conn.row_factory = sqlite3.Row conn.row_factory = sqlite3.Row
cursor = conn.cursor() cursor = conn.cursor()
@ -480,7 +480,7 @@ class EnhancedSurveySystem:
"""重新生成报告(从保存的分析数据)""" """重新生成报告(从保存的分析数据)"""
try: try:
# 获取保存的分析数据 # 获取保存的分析数据
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute(''' cursor.execute('''
@ -544,7 +544,7 @@ class EnhancedSurveySystem:
def cleanup_temp_analysis_data(self, session_id): def cleanup_temp_analysis_data(self, session_id):
"""清理临时分析数据""" """清理临时分析数据"""
conn = sqlite3.connect('survey.db') conn = sqlite3.connect('data/survey.db')
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute('DELETE FROM temp_analysis_data WHERE session_id = ?', (session_id,)) cursor.execute('DELETE FROM temp_analysis_data WHERE session_id = ?', (session_id,))
conn.commit() conn.commit()

View File

@ -8,7 +8,7 @@ from datetime import datetime
def init_database(): def init_database():
"""初始化SQLite数据库""" """初始化SQLite数据库"""
db_path = '/Users/moshui/Documents/survey/survey.db' db_path = '/Users/moshui/Documents/survey/data/survey.db'
# 如果数据库已存在,先删除 # 如果数据库已存在,先删除
if os.path.exists(db_path): if os.path.exists(db_path):

View File

@ -15,7 +15,7 @@ def get_east8_time_string():
return datetime.now(east8_tz).strftime('%Y-%m-%d %H:%M:%S') return datetime.now(east8_tz).strftime('%Y-%m-%d %H:%M:%S')
class SurveyAPI: class SurveyAPI:
def __init__(self, db_path='survey.db'): def __init__(self, db_path='data/survey.db'):
self.db_path = db_path self.db_path = db_path
def get_connection(self): def get_connection(self):