Procházet zdrojové kódy

refactor(解码流程): 拆分result字段为topic_result和script_result

将decode_workflow表中的result字段拆分为更具体的topic_result和script_result
更新相关代码以适配字段变更
max_liu před 6 dny
rodič
revize
1ff576157b
3 změnil soubory, kde provedl 6 přidání a 4 odebrání
  1. 2 2
      decode_task/topicTask.py
  2. 2 1
      main.py
  3. 2 1
      models/decode_workflow.py

+ 2 - 2
decode_task/topicTask.py

@@ -19,7 +19,7 @@ def update_topic_result_by_id(param):
         logger.info(f"task_id = {param.task_id} , 任务不存在")
         return None
     # 取旧值用于返回
-    result, status, error_reason, video_url, title = tasks['result'], tasks['task_status'], tasks['error_reason'], tasks['video_url'], tasks['title']
+    status, video_id, video_url, title = tasks['task_status'], tasks['video_id'], tasks['video_url'], tasks['title']
     # 动态更新可传字段
     fields = []
     values = []
@@ -39,7 +39,7 @@ def update_topic_result_by_id(param):
         sql = f"UPDATE decode_workflow SET {', '.join(fields)} WHERE task_id = %s"
         values.append(param.task_id)
         mysql.execute(sql, tuple(values))
-    return result, status, error_reason, video_url, title
+    return video_id, status, video_url, title
 
 
 def get_topic_result_by_id(task_id:str):

+ 2 - 1
main.py

@@ -74,7 +74,8 @@ def decode_topic(param:TopicListParam):
             video_url=video_url,
             title=video_title,
             type=param.type,
-            result=None,
+            topic_result=None,
+            script_result=None,
         ).save()
         data_list.append({
             "task_id": task_id,

+ 2 - 1
models/decode_workflow.py

@@ -14,7 +14,8 @@ class DecodeWorkflow(BaseModel):
     video_url:        Annotated[str, Field(description='视频地址')]
     title:            Annotated[Optional[str], Field(description='视频标题', default=None)]
     task_status:      Annotated[Optional[int], Field(description='任务状态', default=0)]  # 0:待执行  /   1:执行中  /  2:执行成功  3:执行失败
-    result:           Annotated[Optional[str], Field(description='任务结果')]
+    topic_result:     Annotated[Optional[str], Field(description='选题结果')]
+    script_result:    Annotated[Optional[str], Field(description='脚本结果')]
     created_at:       Annotated[Optional[int], Field(description='任务创建时间戳', default_factory=get_now_ts)]
     error_reason:     Annotated[Optional[str], Field(description='错误原因', default=None)]
     type:             Annotated[Optional[int], Field(description='任务类型', default=1)]  # 0:解码任务  /  1:评估任务