fix: 增加db.close()，解决抓取命令执行完不退出的问题

haodaohong · Feb 21, 2024 · c09f9fe · c09f9fe
1 parent 4f5f83d
commit c09f9fe
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/db.py b/db.py
@@ -18,6 +18,8 @@ async def init_db(create_db: bool = False) -> None:
         _create_db=create_db
     )
 
+async def close() -> None:
+    await Tortoise.close_connections()
 
 async def init():
     await init_db(create_db=True)

diff --git a/main.py b/main.py
@@ -51,6 +51,9 @@ async def main():
         crawler_type=args.type
     )
     await crawler.start()
+
+    if config.SAVE_DATA_OPTION == "db":
+        await db.close()
 
 
 if __name__ == '__main__':