alex/assets-bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

log: Improve processing logs

Browse Source
This commit is contained in:
Aleksey Lobanov
2021-02-26 18:18:06 +03:00
parent eb20e30d44
commit 5606251a83
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
process_responses.py
View File

@@ -89,17 +89,15 @@ def process_task_replies(session, channel, consume_count: int):
): ):
if data is None: if data is None:
return return
data = data.decode("utf-8") raw_reply = json.loads(data.decode("utf-8"))
raw_reply = json.loads(data)
task_id = raw_reply["task_id"] task_id = raw_reply["task_id"]
logging.info(f"New response: {raw_reply}")
db_task = session.query(AnalyzeTask).filter(AnalyzeTask.id == task_id).first() db_task = session.query(AnalyzeTask).filter(AnalyzeTask.id == task_id).first()
if db_task is None: if db_task is None:
logging.warning(f"task_id {task_id} not found in DB") logging.warning(f"task_id {task_id} not found in DB")
continue continue
if raw_reply["status"] == "error": if raw_reply["status"] == "error":
db_task.status = TaskStatus.Error db_task.status = TaskStatus.Error
logging.error(f"Error {raw_reply['message']} with task_id {task_id} not found in DB") logging.error(f"Error `{raw_reply['message']}` with task_id {task_id}.")
continue continue
response = AnalyzeResponse.from_data(raw_reply) response = AnalyzeResponse.from_data(raw_reply)
db_task.status = TaskStatus.Success db_task.status = TaskStatus.Success