87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
"""task_logs 表读写模板。
|
|
|
|
通用业务日志仓储：记录每次任务执行的 task_type / target / input / 状态 / 结果摘要。
|
|
不假设任何特定业务（发布、对账、审核等都可复用）。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, List, Optional, Tuple
|
|
|
|
from db.connection import get_conn, init_db
|
|
|
|
|
|
def save_task_log(
|
|
task_type: str,
|
|
target_id: Optional[str] = None,
|
|
input_id: Optional[str] = None,
|
|
input_title: Optional[str] = None,
|
|
status: str = "success",
|
|
error_msg: Optional[str] = None,
|
|
result_summary: Optional[str] = None,
|
|
) -> int:
|
|
init_db()
|
|
conn = get_conn()
|
|
try:
|
|
cur = conn.cursor()
|
|
cur.execute(
|
|
"""
|
|
INSERT INTO task_logs
|
|
(task_type, target_id, input_id, input_title, status, error_msg, result_summary)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?)
|
|
""",
|
|
(task_type, target_id, input_id, input_title or "", status, error_msg, result_summary),
|
|
)
|
|
new_id = int(cur.lastrowid)
|
|
conn.commit()
|
|
finally:
|
|
conn.close()
|
|
return new_id
|
|
|
|
|
|
def list_task_logs(
|
|
limit: int,
|
|
status: Optional[str] = None,
|
|
task_type: Optional[str] = None,
|
|
target_id: Optional[str] = None,
|
|
) -> List[Tuple[Any, ...]]:
|
|
init_db()
|
|
conn = get_conn()
|
|
try:
|
|
cur = conn.cursor()
|
|
sql = (
|
|
"SELECT id, task_type, target_id, input_id, input_title, status, error_msg, result_summary, "
|
|
"created_at, updated_at FROM task_logs WHERE 1=1 "
|
|
)
|
|
params: List[Any] = []
|
|
if status:
|
|
sql += "AND status = ? "
|
|
params.append(status)
|
|
if task_type:
|
|
sql += "AND task_type = ? "
|
|
params.append(task_type)
|
|
if target_id:
|
|
sql += "AND target_id = ? "
|
|
params.append(target_id)
|
|
sql += "ORDER BY created_at DESC, id DESC LIMIT ?"
|
|
params.append(int(limit))
|
|
cur.execute(sql, tuple(params))
|
|
return list(cur.fetchall())
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
def get_task_log_by_id(log_id: int) -> Optional[Tuple[Any, ...]]:
|
|
init_db()
|
|
conn = get_conn()
|
|
try:
|
|
cur = conn.cursor()
|
|
cur.execute(
|
|
"SELECT id, task_type, target_id, input_id, input_title, status, error_msg, result_summary, "
|
|
"created_at, updated_at FROM task_logs WHERE id = ?",
|
|
(int(log_id),),
|
|
)
|
|
return cur.fetchone()
|
|
finally:
|
|
conn.close()
|