feat 🐛:add

scheduler/scheduler_module.py

@@ -8,7 +8,7 @@ import httpx
 from apscheduler.schedulers.asyncio import AsyncIOScheduler
 
 from util.config import DataLoader
-from util.database import get_latest_update_time
+from util.database import get_latest_update_time, insert_many_messages, update_sent_status
 
 logger = logging.getLogger(__name__) # 获取当前模块的logger
 logger.setLevel(logging.INFO)
@@ -56,9 +56,14 @@ async def poll_external_api():
             logger.info("接口返回错误，请检查响应。")
             return
         for report in reports_list:
+            id = report["id"]
             file_id = report["file_id"]
             download_dir_file_id = download_dir_all+"/"+file_id
             file_path = download_file(download_report_url,download_dir_file_id)
+            if file_path is not None:
+                messages_to_insert =[(id,file_id, current_time, 0)]
+                error_message = insert_many_messages(messages_to_insert)
+                if error_message is None:
                     async with httpx.AsyncClient() as client:
                         # 1，获取需要推送的文档ID 2，根据ID获取文档，3，文档写成文件，4，发送文件，5，删除文件
                         files = []
@@ -83,6 +88,8 @@ async def poll_external_api():
                             response.raise_for_status()
                             logger.info("文件发送成功！")
                         # 都发送成功,则删除文件，回填到数据库
+                        update_sent_status(id)
+                        os.remove(file_path)
     except httpx.HTTPStatusError as e:
         logger.info(f"调用外部接口 {push_message_url} 时发生 HTTP 状态码错误: {e.response.status_code} - {e.response.text}")
     except httpx.RequestError as e:
@@ -124,9 +131,9 @@ def download_file(url, download_dir_all):
     if not os.path.exists(download_dir_all):
         os.makedirs(download_dir_all)
     print(f"开始从 {url} 下载文件...")
+    filename = None
     try:
         # 使用流式方式（stream=True）下载
-        filename = ""
         with httpx.stream("GET", url, follow_redirects=True) as response:
             # 检查响应状态码，如果不是2xx则抛出异常
             response.raise_for_status()

util/database.py

@@ -76,6 +76,7 @@ def insert_many_messages(messages_to_insert):
     messages_to_insert 是一个包含 (id, update_time, sent) 元组的列表。
     例如: [('id1', '2023-10-27 10:00:00', 0), ('id2', '2023-10-27 10:05:00', 0)]
     """
+    try:
         conn = sqlite3.connect(DB_PATH)
         cursor = conn.cursor()
 
@@ -84,10 +85,13 @@ def insert_many_messages(messages_to_insert):
 
         conn.commit()
         conn.close()
+    except Exception as e:
+        return e
     print(f"已批量插入 {len(messages_to_insert)} 条数据到数据库。")
+    return None
 
 
-def update_sent_status(ids_to_update):
+def update_sent_status(id):
     """
     根据给定的 ID 列表，将 sent 字段更新为 1 (已发送)。
     ids_to_update 是一个包含 ID 的列表或元组。
@@ -97,10 +101,9 @@ def update_sent_status(ids_to_update):
     cursor = conn.cursor()
 
     # 构建 SQL 语句，使用 IN 子句批量更新
-    placeholders = ', '.join('?' for _ in ids_to_update)
+    sql_query = f"UPDATE messages SET sent = 1 WHERE id = ?"
-    sql_query = f"UPDATE messages SET sent = 1 WHERE id IN ({placeholders})"
 
-    cursor.execute(sql_query, ids_to_update)
+    cursor.execute(sql_query, id)
 
     conn.commit()
     conn.close()