Bladeren bron

add_script_config 添加user_id

liyuyi@c-top.com.cn 3 jaren geleden
bovenliggende
commit
0d28b8ddfb
1 gewijzigde bestanden met toevoegingen van 7 en 3 verwijderingen
  1. 7 3
      main.py

+ 7 - 3
main.py

@@ -1,4 +1,6 @@
 import hashlib
+import os
+import sys
 import traceback
 import uuid
 from concurrent.futures import ThreadPoolExecutor
@@ -7,8 +9,7 @@ from datetime import timedelta
 from io import BytesIO
 from typing import Optional, List
 from urllib.parse import quote
-import os
-import sys
+
 import pandas as pd
 import uvicorn
 import yaml
@@ -177,12 +178,14 @@ class QueryWordTaskInfoLst(BaseModel):
 class AddScriptConfig(BaseModel):
     query_word_lst: List[str] = Field(..., description="关键词组")
     operator: str = Field(..., description="操作者")
+    user_id: str = Field(..., description="user_id")
 
     class Config:
         schema_extra = {
             "example": {
                 "query_word_lst": ["红包", "淘特"],
-                "operator": "龙猫"
+                "operator": "龙猫",
+                "user_id": "234d46d1873f4dac85b2a2f9ad541e18"
             }
         }
 
@@ -358,6 +361,7 @@ def add_script_config(item: AddScriptConfig):
         config_df['config_id'] = config_id
         config_df['operator'] = item.operator
         config_df['operate_type'] = 1
+        config_df['user_id'] = item.user_id
 
         # 新增配置记录插入到 ctop_ai_script_query_word_config
         config_df.to_sql(name="ctop_ai_script_query_word_config",