优化项目结构,格式化代码

This commit is contained in:
雅诺狐
2025-08-13 09:49:39 +08:00
parent 26039b003c
commit fe97d897ef
24 changed files with 57 additions and 77 deletions

View File

@@ -10,7 +10,7 @@ from typing import Dict, List, Any, Union, Type, Optional
from contextlib import contextmanager
from sqlalchemy.orm import Session
from sqlalchemy.exc import SQLAlchemyError, DisconnectionError, OperationalError
from sqlalchemy import desc, asc, func, and_, or_
from sqlalchemy import desc, asc, func, and_
from src.common.logger import get_logger
from src.common.database.sqlalchemy_models import (
Base, get_db_session, Messages, ActionRecords, PersonInfo, ChatStreams,
@@ -61,7 +61,7 @@ def get_db_session():
time.sleep(retry_delay * (attempt + 1))
else:
raise
except Exception as e:
except Exception:
if session:
session.rollback()
raise

View File

@@ -68,7 +68,7 @@ class SessionProxy:
# commit后不要清除session只是刷新状态
pass # 保持session活跃
return result
except Exception as e:
except Exception:
try:
if session and hasattr(session, 'rollback'):
session.rollback()
@@ -555,7 +555,7 @@ def get_db_session():
session = SessionLocal()
yield session
session.commit()
except Exception as e:
except Exception:
if session:
session.rollback()
raise

View File

@@ -1,5 +1,4 @@
import json
import asyncio
from datetime import datetime
from typing import Optional, List, Dict, Any
@@ -93,13 +92,13 @@ class ScheduleManager:
existing_schedule = session.query(Schedule).filter(Schedule.date == today_str).first()
if existing_schedule:
# 更新现有日程 - 通过setattr或直接赋值
setattr(existing_schedule, 'schedule_data', json.dumps(schedule_data))
setattr(existing_schedule, 'updated_at', datetime.now())
existing_schedule.schedule_data = json.dumps(schedule_data)
existing_schedule.updated_at = datetime.now()
else:
# 创建新日程
new_schedule = Schedule()
setattr(new_schedule, 'date', today_str)
setattr(new_schedule, 'schedule_data', json.dumps(schedule_data))
new_schedule.date = today_str
new_schedule.schedule_data = json.dumps(schedule_data)
session.add(new_schedule)
# 美化输出