Pārlūkot izejas kodu

video_generate_text

luojunhui 4 mēneši atpakaļ
vecāks
revīzija
4f3c49ae61
2 mainītis faili ar 12 papildinājumiem un 7 dzēšanām
  1. 10 0
      run_article_summary.py
  2. 2 7
      tasks/article_summary_task.py

+ 10 - 0
run_article_summary.py

@@ -0,0 +1,10 @@
+"""
+@author: luojunhui
+"""
+from tasks.article_summary_task import ArticleSummaryTask
+
+
+if __name__ == '__main__':
+    article_summary_task = ArticleSummaryTask()
+    article_summary_task.connect_db()
+    article_summary_task.deal()

+ 2 - 7
tasks/article_summary_task.py

@@ -2,6 +2,7 @@
 @author: luojunhui
 """
 from pymysql.cursors import DictCursor
+from tqdm import tqdm
 
 from applications.api import deep_seek_api
 from applications.db import DatabaseConnector
@@ -101,7 +102,7 @@ class ArticleSummaryTask(object):
         开始处理任务
         """
         task_list = self.get_task_list()
-        for task in task_list:
+        for task in tqdm(task_list):
             try:
                 self.process_each_task(task)
             except Exception as e:
@@ -109,12 +110,6 @@ class ArticleSummaryTask(object):
                 continue
 
 
-if __name__ == '__main__':
-    article_summary_task = ArticleSummaryTask()
-    article_summary_task.connect_db()
-    article_summary_task.deal()
-
-