Skip to content

Commit c09f9fe

Browse files
author
ccgo
committed
fix: 增加db.close(),解决抓取命令执行完不退出的问题
1 parent 4f5f83d commit c09f9fe

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

db.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@ async def init_db(create_db: bool = False) -> None:
1818
_create_db=create_db
1919
)
2020

21+
async def close() -> None:
22+
await Tortoise.close_connections()
2123

2224
async def init():
2325
await init_db(create_db=True)

main.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,9 @@ async def main():
5151
crawler_type=args.type
5252
)
5353
await crawler.start()
54+
55+
if config.SAVE_DATA_OPTION == "db":
56+
await db.close()
5457

5558

5659
if __name__ == '__main__':

0 commit comments

Comments
 (0)