mirror of
https://github.com/NanmiCoder/MediaCrawler.git
synced 2026-07-24 05:10:30 +08:00
9f4f8bf768
- 用户 ID 转为匿名 creator_hash，昵称中间脱敏，IP/头像/主页/签名/性别不再采集 - 覆盖 xhs/weibo/bilibili/douyin/kuaishou/tieba/zhihu 7 个平台 - 删除 7 张 creator 档案 ORM 表，15 张内容/评论表新增 creator_hash 列 - B 站禁用粉丝/关注/联系人列表抓取 - 新增 tools/user_hash.py 与 4 个平台的 mock+SQLite 端到端测试 测试: pytest tests/test_no_user_info.py tests/test_weibo_no_user_info.py tests/test_douyin_no_user_info.py tests/test_kuaishou_no_user_info.py (21 passed)
115 lines
5.0 KiB
Python
115 lines
5.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
# Copyright (c) 2025 relakkes@gmail.com
|
||
#
|
||
# This file is part of MediaCrawler project.
|
||
# Repository: https://github.com/NanmiCoder/MediaCrawler/blob/main/store/kuaishou/__init__.py
|
||
# GitHub: https://github.com/NanmiCoder
|
||
# Licensed under NON-COMMERCIAL LEARNING LICENSE 1.1
|
||
#
|
||
|
||
# 声明：本代码仅供学习和研究目的使用。使用者应遵守以下原则：
|
||
# 1. 不得用于任何商业用途。
|
||
# 2. 使用时应遵守目标平台的使用条款和robots.txt规则。
|
||
# 3. 不得进行大规模爬取或对平台造成运营干扰。
|
||
# 4. 应合理控制请求频率，避免给目标平台带来不必要的负担。
|
||
# 5. 不得用于任何非法或不当的用途。
|
||
#
|
||
# 详细许可条款请参阅项目根目录下的LICENSE文件。
|
||
# 使用本代码即表示您同意遵守上述原则和LICENSE中的所有条款。
|
||
|
||
|
||
# -*- coding: utf-8 -*-
|
||
# @Author : relakkes@gmail.com
|
||
# @Time : 2024/1/14 20:03
|
||
# @Desc :
|
||
from typing import List
|
||
|
||
import config
|
||
from var import source_keyword_var
|
||
from tools.user_hash import anonymize_user_id, mask_nickname
|
||
|
||
from ._store_impl import *
|
||
|
||
|
||
class KuaishouStoreFactory:
|
||
STORES = {
|
||
"csv": KuaishouCsvStoreImplement,
|
||
"db": KuaishouDbStoreImplement,
|
||
"postgres": KuaishouDbStoreImplement,
|
||
"json": KuaishouJsonStoreImplement,
|
||
"jsonl": KuaishouJsonlStoreImplement,
|
||
"sqlite": KuaishouSqliteStoreImplement,
|
||
"mongodb": KuaishouMongoStoreImplement,
|
||
"excel": KuaishouExcelStoreImplement,
|
||
}
|
||
|
||
@staticmethod
|
||
def create_store() -> AbstractStore:
|
||
store_class = KuaishouStoreFactory.STORES.get(config.SAVE_DATA_OPTION)
|
||
if not store_class:
|
||
raise ValueError(
|
||
"[KuaishouStoreFactory.create_store] Invalid save option only supported csv or db or json or sqlite or mongodb or excel ...")
|
||
return store_class()
|
||
|
||
|
||
async def update_kuaishou_video(video_item: Dict):
|
||
photo_info: Dict = video_item.get("photo", {})
|
||
video_id = photo_info.get("id")
|
||
if not video_id:
|
||
return
|
||
user_info = video_item.get("author", {})
|
||
save_content_item = {
|
||
"video_id": video_id,
|
||
"video_type": str(video_item.get("type")),
|
||
"title": photo_info.get("caption", "")[:500],
|
||
"desc": photo_info.get("caption", "")[:500],
|
||
"create_time": photo_info.get("timestamp"),
|
||
"creator_hash": anonymize_user_id(user_info.get("id")), # 创作者匿名哈希(不存原始 user_id)
|
||
"nickname": mask_nickname(user_info.get("name")), # 用户昵称(已脱敏)
|
||
"liked_count": str(photo_info.get("realLikeCount")),
|
||
"viewd_count": str(photo_info.get("viewCount")),
|
||
"last_modify_ts": utils.get_current_timestamp(),
|
||
"video_url": f"https://www.kuaishou.com/short-video/{video_id}",
|
||
"video_cover_url": photo_info.get("coverUrl", ""),
|
||
"video_play_url": photo_info.get("photoUrl", ""),
|
||
"source_keyword": source_keyword_var.get(),
|
||
}
|
||
utils.logger.info(
|
||
f"[store.kuaishou.update_kuaishou_video] Kuaishou video id:{video_id}, title:{save_content_item.get('title')}")
|
||
await KuaishouStoreFactory.create_store().store_content(content_item=save_content_item)
|
||
|
||
|
||
async def batch_update_ks_video_comments(video_id: str, comments: List[Dict]):
|
||
utils.logger.info(f"[store.kuaishou.batch_update_ks_video_comments] video_id:{video_id}, comments:{comments}")
|
||
if not comments:
|
||
return
|
||
for comment_item in comments:
|
||
await update_ks_video_comment(video_id, comment_item)
|
||
|
||
|
||
async def update_ks_video_comment(video_id: str, comment_item: Dict):
|
||
# V2 API uses snake_case field names and comment_id is int type
|
||
# Old GraphQL API used camelCase field names
|
||
# Support both formats for backward compatibility
|
||
comment_id = comment_item.get("comment_id") or comment_item.get("commentId")
|
||
save_comment_item = {
|
||
"comment_id": str(comment_id) if comment_id else None, # Convert to string for storage
|
||
"create_time": comment_item.get("timestamp"),
|
||
"video_id": video_id,
|
||
"content": comment_item.get("content"),
|
||
# 创作者匿名哈希(不存原始 user_id)：V2: author_id, Old: authorId
|
||
"creator_hash": anonymize_user_id(comment_item.get("author_id") or comment_item.get("authorId")),
|
||
# 用户昵称(已脱敏)：V2: author_name, Old: authorName
|
||
"nickname": mask_nickname(comment_item.get("author_name") or comment_item.get("authorName")),
|
||
# V2: commentCount, Old: subCommentCount
|
||
"sub_comment_count": str(comment_item.get("commentCount") or comment_item.get("subCommentCount", 0)),
|
||
"last_modify_ts": utils.get_current_timestamp(),
|
||
}
|
||
utils.logger.info(
|
||
f"[store.kuaishou.update_ks_video_comment] Kuaishou video comment: {comment_id}, content: {save_comment_item.get('content')}")
|
||
await KuaishouStoreFactory.create_store().store_comment(comment_item=save_comment_item)
|
||
|
||
async def save_creator(user_id: str, creator: Dict):
|
||
# 教学版：创作者个人资料(昵称/性别/头像/签名/IP/粉丝数等)不再落库，防骚扰。
|
||
return
|