修改说明
This commit is contained in:
33
src/main.py
33
src/main.py
@@ -251,6 +251,39 @@ class TSPAssistant:
|
||||
self.logger.error(f"获取活跃预警失败: {e}")
|
||||
return []
|
||||
|
||||
def create_alert(self, alert_type: str, title: str, description: str, level: str = "medium") -> Dict[str, Any]:
|
||||
"""创建预警"""
|
||||
try:
|
||||
from ..core.database import db_manager
|
||||
from ..core.models import Alert
|
||||
from datetime import datetime
|
||||
|
||||
with db_manager.get_session() as session:
|
||||
alert = Alert(
|
||||
rule_name=f"手动预警_{datetime.now().strftime('%Y%m%d_%H%M%S')}",
|
||||
alert_type=alert_type,
|
||||
level=level,
|
||||
message=f"{title}: {description}",
|
||||
is_active=True,
|
||||
created_at=datetime.now()
|
||||
)
|
||||
session.add(alert)
|
||||
session.commit()
|
||||
|
||||
self.logger.info(f"创建预警成功: {title}")
|
||||
return {
|
||||
"id": alert.id,
|
||||
"title": title,
|
||||
"description": description,
|
||||
"level": level,
|
||||
"alert_type": alert_type,
|
||||
"created_at": alert.created_at.isoformat()
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"创建预警异常: {e}")
|
||||
return {"error": f"创建异常: {str(e)}"}
|
||||
|
||||
def resolve_alert(self, alert_id: int) -> bool:
|
||||
"""解决预警"""
|
||||
try:
|
||||
|
||||
Reference in New Issue
Block a user