jiazhizhong/MediaCrawler
1
0
Fork 0
mirror of https://github.com/NanmiCoder/MediaCrawler.git synced 2026-05-30 06:27:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 更新各平台store SQL文件以支持SQLite\n\n- 更新 store/kuaishou/kuaishou_store_sql.py: 快手平台SQLite适配\n- 更新 store/tieba/tieba_store_sql.py: 贴吧平台SQLite适配\n- 更新 store/weibo/weibo_store_sql.py: 微博平台SQLite适配\n- 更新 store/xhs/xhs_store_sql.py: 小红书平台SQLite适配\n- 更新 store/zhihu/zhihu_store_sql.py: 知乎平台SQLite适配

Browse Source
This commit is contained in:
买定不离手
2025-07-14 03:36:20 +08:00
parent 3a2959d86c
commit fb938f38aa
5 changed files with 57 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
store/zhihu/zhihu_store_sql.py
View File

@@ -10,9 +10,10 @@
# -*- coding: utf-8 -*-
from typing import Dict, List
from typing import Dict, List, Union
from db import AsyncMysqlDB
from async_db import AsyncMysqlDB
from async_sqlite_db import AsyncSqliteDB
from var import media_crawler_db_var
@@ -25,7 +26,7 @@ async def query_content_by_content_id(content_id: str) -> Dict:
Returns:
"""
async_db_conn: AsyncMysqlDB = media_crawler_db_var.get()
async_db_conn: Union[AsyncMysqlDB, AsyncSqliteDB] = media_crawler_db_var.get()
sql: str = f"select * from zhihu_content where content_id = '{content_id}'"
rows: List[Dict] = await async_db_conn.query(sql)
if len(rows) > 0:
@@ -42,7 +43,7 @@ async def add_new_content(content_item: Dict) -> int:
Returns:
"""
async_db_conn: AsyncMysqlDB = media_crawler_db_var.get()
async_db_conn: Union[AsyncMysqlDB, AsyncSqliteDB] = media_crawler_db_var.get()
last_row_id: int = await async_db_conn.item_to_table("zhihu_content", content_item)
return last_row_id
@@ -57,7 +58,7 @@ async def update_content_by_content_id(content_id: str, content_item: Dict) -> i
Returns:
"""
async_db_conn: AsyncMysqlDB = media_crawler_db_var.get()
async_db_conn: Union[AsyncMysqlDB, AsyncSqliteDB] = media_crawler_db_var.get()
effect_row: int = await async_db_conn.update_table("zhihu_content", content_item, "content_id", content_id)
return effect_row
@@ -72,7 +73,7 @@ async def query_comment_by_comment_id(comment_id: str) -> Dict:
Returns:
"""
async_db_conn: AsyncMysqlDB = media_crawler_db_var.get()
async_db_conn: Union[AsyncMysqlDB, AsyncSqliteDB] = media_crawler_db_var.get()
sql: str = f"select * from zhihu_comment where comment_id = '{comment_id}'"
rows: List[Dict] = await async_db_conn.query(sql)
if len(rows) > 0:
@@ -89,7 +90,7 @@ async def add_new_comment(comment_item: Dict) -> int:
Returns:
"""
async_db_conn: AsyncMysqlDB = media_crawler_db_var.get()
async_db_conn: Union[AsyncMysqlDB, AsyncSqliteDB] = media_crawler_db_var.get()
last_row_id: int = await async_db_conn.item_to_table("zhihu_comment", comment_item)
return last_row_id
@@ -104,7 +105,7 @@ async def update_comment_by_comment_id(comment_id: str, comment_item: Dict) -> i
Returns:
"""
async_db_conn: AsyncMysqlDB = media_crawler_db_var.get()
async_db_conn: Union[AsyncMysqlDB, AsyncSqliteDB] = media_crawler_db_var.get()
effect_row: int = await async_db_conn.update_table("zhihu_comment", comment_item, "comment_id", comment_id)
return effect_row
@@ -118,7 +119,7 @@ async def query_creator_by_user_id(user_id: str) -> Dict:
Returns:
"""
async_db_conn: AsyncMysqlDB = media_crawler_db_var.get()
async_db_conn: Union[AsyncMysqlDB, AsyncSqliteDB] = media_crawler_db_var.get()
sql: str = f"select * from zhihu_creator where user_id = '{user_id}'"
rows: List[Dict] = await async_db_conn.query(sql)
if len(rows) > 0:
@@ -135,7 +136,7 @@ async def add_new_creator(creator_item: Dict) -> int:
Returns:
"""
async_db_conn: AsyncMysqlDB = media_crawler_db_var.get()
async_db_conn: Union[AsyncMysqlDB, AsyncSqliteDB] = media_crawler_db_var.get()
last_row_id: int = await async_db_conn.item_to_table("zhihu_creator", creator_item)
return last_row_id
@@ -150,6 +151,6 @@ async def update_creator_by_user_id(user_id: str, creator_item: Dict) -> int:
Returns:
"""
async_db_conn: AsyncMysqlDB = media_crawler_db_var.get()
async_db_conn: Union[AsyncMysqlDB, AsyncSqliteDB] = media_crawler_db_var.get()
effect_row: int = await async_db_conn.update_table("zhihu_creator", creator_item, "user_id", user_id)
return effect_row