浏览代码

fix: 修复decode_workflow模型字段默认值及环境配置

修正decode_workflow模型中字段的默认值设置,统一命名规范
将APP_ENV从local改为prod环境配置
在返回数据中添加type字段
max_liu 1 天之前
父节点
当前提交
90be8d1687
共有 3 个文件被更改,包括 9 次插入8 次删除
  1. 1 1
      docker-compose.yaml
  2. 4 3
      main.py
  3. 4 4
      models/decode_workflow.py

+ 1 - 1
docker-compose.yaml

@@ -13,5 +13,5 @@ services:
     env_file:
       - .env
     environment:
-      - APP_ENV=local
+      - APP_ENV=prod
     entrypoint: ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

+ 4 - 3
main.py

@@ -51,6 +51,8 @@ def decode_topic(param:TopicListParam):
     video_list = param.video_list
     logger.info(f"数据池数据 = {video_list}")
     data_list = []
+
+
     for video in video_list:
         video_id = video.video_id
         video_url = video.video_url
@@ -63,15 +65,14 @@ def decode_topic(param:TopicListParam):
             title=video_title,
             type=param.type,
             result=None,
-            task_status = 0
         ).save()
         data_list.append({
             "task_id": task_id,
             "video_id": video_id,
             "video_url": video_url,
             "title": video_title,
-            "task_status": 0
-            
+            "task_status": 0,
+            "type": param.type,
         })
     return {
         "code": 0,

+ 4 - 4
models/decode_workflow.py

@@ -10,13 +10,13 @@ class DecodeWorkflow(BaseModel):
     table_name:       Annotated[Optional[str], Field(description='表名', exclude=True)] = 'decode_workflow'
 
     task_id:          Annotated[str, Field(description='任务ID')]
-    video_id:         Annotated[Optional[str], Field(description='视频ID')]
+    video_id:         Annotated[Optional[str], Field(description='视频ID', default=None)]
     video_url:        Annotated[str, Field(description='视频地址')]
-    title:            Annotated[Optional[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='任务结果')]
-    create_at:        Annotated[Optional[int], Field(description='任务创建时间戳', default_factory=get_now_ts)]
-    error_reason:     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:评估任务
 
     def save(self):