injectlib/MediaCrawler
1
0
Fork 0
mirror of https://github.com/NanmiCoder/MediaCrawler.git synced 2025-11-25 11:29:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(store): 将async for循环替换为async with语句来修复zhihu数据库会话管理

Browse Source
This commit is contained in:
persist-1
2025-09-08 00:29:04 +08:00
parent a38058856f
commit 40de0e47e5
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
store/zhihu/_store_impl.py
View File

@@ -93,9 +93,9 @@ class ZhihuDbStoreImplement(AbstractStore):
Args: Args:
content_item: content item dict content_item: content item dict
""" """
note_id = content_item.get("note_id") content_id = content_item.get("content_id")
async for session in get_session(): async with get_session() as session:
stmt = select(ZhihuContent).where(ZhihuContent.content_id == note_id) stmt = select(ZhihuContent).where(ZhihuContent.content_id == content_id)
result = await session.execute(stmt) result = await session.execute(stmt)
existing_content = result.scalars().first() existing_content = result.scalars().first()
if existing_content: if existing_content:
@@ -113,7 +113,7 @@ class ZhihuDbStoreImplement(AbstractStore):
comment_item: comment item dict comment_item: comment item dict
""" """
comment_id = comment_item.get("comment_id") comment_id = comment_item.get("comment_id")
async for session in get_session(): async with get_session() as session:
stmt = select(ZhihuComment).where(ZhihuComment.comment_id == comment_id) stmt = select(ZhihuComment).where(ZhihuComment.comment_id == comment_id)
result = await session.execute(stmt) result = await session.execute(stmt)
existing_comment = result.scalars().first() existing_comment = result.scalars().first()
@@ -132,7 +132,7 @@ class ZhihuDbStoreImplement(AbstractStore):
creator: creator dict creator: creator dict
""" """
user_id = creator.get("user_id") user_id = creator.get("user_id")
async for session in get_session(): async with get_session() as session:
stmt = select(ZhihuCreator).where(ZhihuCreator.user_id == user_id) stmt = select(ZhihuCreator).where(ZhihuCreator.user_id == user_id)
result = await session.execute(stmt) result = await session.execute(stmt)
existing_creator = result.scalars().first() existing_creator = result.scalars().first()