Просмотр исходного кода

初始化agent增加数据表是否为空判断

zhangqian 1 год назад
Родитель
Сommit
788572444e
1 измененных файлов с 3 добавлено и 0 удалено
  1. 3 0
      app/task/fetch_agent.py

+ 3 - 0
app/task/fetch_agent.py

@@ -106,6 +106,9 @@ def update_ids_in_local(data: List[Tuple]):
 def initialize_agents():
     db = SessionLocal()
     try:
+        count = db.query(AgentModel).count()
+        if count > 0:
+            return
         initial_agents = [
             ('80ee430a-e396-48c4-a12c-7c7cdf5eda51', 1, '报告生成', 'BISHENG', 'report'),
             ('basic_excel_merge', 2, '报表合并', 'BASIC', 'excelMerge'),