Files
bettafish-company/MindSpider/DeepSentimentCrawling/platform_crawler.py
T
2025-12-02 14:01:39 +08:00

532 lines
23 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
DeepSentimentCrawling模块 - 平台爬虫管理器
负责配置和调用MediaCrawler进行多平台爬取
"""
import os
import sys
import subprocess
import tempfile
from datetime import datetime
from pathlib import Path
from typing import List, Dict, Optional
import json
from loguru import logger
# 添加项目根目录到路径
project_root = Path(__file__).parent.parent
sys.path.append(str(project_root))
try:
import config
except ImportError:
raise ImportError("无法导入config.py配置文件")
class PlatformCrawler:
"""平台爬虫管理器"""
def __init__(self):
"""初始化平台爬虫管理器"""
self.mediacrawler_path = Path(__file__).parent / "MediaCrawler"
self.supported_platforms = ['xhs', 'dy', 'ks', 'bili', 'wb', 'tieba', 'zhihu']
self.crawl_stats = {}
# 确保MediaCrawler目录存在
if not self.mediacrawler_path.exists():
raise FileNotFoundError(f"MediaCrawler目录不存在: {self.mediacrawler_path}")
logger.info(f"初始化平台爬虫管理器,MediaCrawler路径: {self.mediacrawler_path}")
def configure_mediacrawler_db(self):
"""配置MediaCrawler使用我们的数据库(MySQL或PostgreSQL"""
try:
# 判断数据库类型
db_dialect = (config.settings.DB_DIALECT or "mysql").lower()
is_postgresql = db_dialect in ("postgresql", "postgres")
# 修改MediaCrawler的数据库配置
db_config_path = self.mediacrawler_path / "config" / "db_config.py"
# 读取原始配置
with open(db_config_path, 'r', encoding='utf-8') as f:
content = f.read()
# PostgreSQL配置值:如果使用PostgreSQL则使用MindSpider配置,否则使用默认值或环境变量
pg_password = config.settings.DB_PASSWORD if is_postgresql else "bettafish"
pg_user = config.settings.DB_USER if is_postgresql else "bettafish"
pg_host = config.settings.DB_HOST if is_postgresql else "127.0.0.1"
pg_port = config.settings.DB_PORT if is_postgresql else 5432
pg_db_name = config.settings.DB_NAME if is_postgresql else "bettafish"
# 替换数据库配置 - 使用MindSpider的数据库配置
new_config = f'''# 声明:本代码仅供学习和研究目的使用。使用者应遵守以下原则:
# 1. 不得用于任何商业用途。
# 2. 使用时应遵守目标平台的使用条款和robots.txt规则。
# 3. 不得进行大规模爬取或对平台造成运营干扰。
# 4. 应合理控制请求频率,避免给目标平台带来不必要的负担。
# 5. 不得用于任何非法或不当的用途。
#
# 详细许可条款请参阅项目根目录下的LICENSE文件。
# 使用本代码即表示您同意遵守上述原则和LICENSE中的所有条款。
import os
# mysql config - 使用MindSpider的数据库配置
MYSQL_DB_PWD = "{config.settings.DB_PASSWORD}"
MYSQL_DB_USER = "{config.settings.DB_USER}"
MYSQL_DB_HOST = "{config.settings.DB_HOST}"
MYSQL_DB_PORT = {config.settings.DB_PORT}
MYSQL_DB_NAME = "{config.settings.DB_NAME}"
mysql_db_config = {{
"user": MYSQL_DB_USER,
"password": MYSQL_DB_PWD,
"host": MYSQL_DB_HOST,
"port": MYSQL_DB_PORT,
"db_name": MYSQL_DB_NAME,
}}
# redis config
REDIS_DB_HOST = "127.0.0.1" # your redis host
REDIS_DB_PWD = os.getenv("REDIS_DB_PWD", "123456") # your redis password
REDIS_DB_PORT = os.getenv("REDIS_DB_PORT", 6379) # your redis port
REDIS_DB_NUM = os.getenv("REDIS_DB_NUM", 0) # your redis db num
# cache type
CACHE_TYPE_REDIS = "redis"
CACHE_TYPE_MEMORY = "memory"
# sqlite config
SQLITE_DB_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "database", "sqlite_tables.db")
sqlite_db_config = {{
"db_path": SQLITE_DB_PATH
}}
# postgresql config - 使用MindSpider的数据库配置(如果DB_DIALECT是postgresql)或环境变量
POSTGRESQL_DB_PWD = os.getenv("POSTGRESQL_DB_PWD", "{pg_password}")
POSTGRESQL_DB_USER = os.getenv("POSTGRESQL_DB_USER", "{pg_user}")
POSTGRESQL_DB_HOST = os.getenv("POSTGRESQL_DB_HOST", "{pg_host}")
POSTGRESQL_DB_PORT = os.getenv("POSTGRESQL_DB_PORT", "{pg_port}")
POSTGRESQL_DB_NAME = os.getenv("POSTGRESQL_DB_NAME", "{pg_db_name}")
postgresql_db_config = {{
"user": POSTGRESQL_DB_USER,
"password": POSTGRESQL_DB_PWD,
"host": POSTGRESQL_DB_HOST,
"port": POSTGRESQL_DB_PORT,
"db_name": POSTGRESQL_DB_NAME,
}}
'''
# 写入新配置
with open(db_config_path, 'w', encoding='utf-8') as f:
f.write(new_config)
db_type = "PostgreSQL" if is_postgresql else "MySQL"
logger.info(f"已配置MediaCrawler使用MindSpider {db_type}数据库")
return True
except Exception as e:
logger.exception(f"配置MediaCrawler数据库失败: {e}")
return False
def create_base_config(self, platform: str, keywords: List[str],
crawler_type: str = "search", max_notes: int = 50) -> bool:
"""
创建MediaCrawler的基础配置
Args:
platform: 平台名称
keywords: 关键词列表
crawler_type: 爬取类型
max_notes: 最大爬取数量
Returns:
是否配置成功
"""
try:
# 判断数据库类型,确定 SAVE_DATA_OPTION
db_dialect = (config.settings.DB_DIALECT or "mysql").lower()
is_postgresql = db_dialect in ("postgresql", "postgres")
save_data_option = "postgresql" if is_postgresql else "db"
base_config_path = self.mediacrawler_path / "config" / "base_config.py"
# 将关键词列表转换为逗号分隔的字符串
keywords_str = ",".join(keywords)
# 读取原始配置文件
with open(base_config_path, 'r', encoding='utf-8') as f:
content = f.read()
# 修改关键配置项
lines = content.split('\n')
new_lines = []
for line in lines:
if line.startswith('PLATFORM = '):
new_lines.append(f'PLATFORM = "{platform}" # 平台,xhs | dy | ks | bili | wb | tieba | zhihu')
elif line.startswith('KEYWORDS = '):
new_lines.append(f'KEYWORDS = "{keywords_str}" # 关键词搜索配置,以英文逗号分隔')
elif line.startswith('CRAWLER_TYPE = '):
new_lines.append(f'CRAWLER_TYPE = "{crawler_type}" # 爬取类型,search(关键词搜索) | detail(帖子详情)| creator(创作者主页数据)')
elif line.startswith('SAVE_DATA_OPTION = '):
new_lines.append(f'SAVE_DATA_OPTION = "{save_data_option}" # csv or db or json or sqlite or postgresql')
elif line.startswith('CRAWLER_MAX_NOTES_COUNT = '):
new_lines.append(f'CRAWLER_MAX_NOTES_COUNT = {max_notes}')
elif line.startswith('ENABLE_GET_COMMENTS = '):
new_lines.append('ENABLE_GET_COMMENTS = True')
elif line.startswith('CRAWLER_MAX_COMMENTS_COUNT_SINGLENOTES = '):
new_lines.append('CRAWLER_MAX_COMMENTS_COUNT_SINGLENOTES = 20')
elif line.startswith('HEADLESS = '):
new_lines.append('HEADLESS = True') # 使用无头模式
else:
new_lines.append(line)
# 写入新配置
with open(base_config_path, 'w', encoding='utf-8') as f:
f.write('\n'.join(new_lines))
logger.info(f"已配置 {platform} 平台,爬取类型: {crawler_type},关键词数量: {len(keywords)},最大爬取数量: {max_notes},保存数据方式: {save_data_option}")
return True
except Exception as e:
logger.exception(f"创建基础配置失败: {e}")
return False
def run_crawler(self, platform: str, keywords: List[str],
login_type: str = "qrcode", max_notes: int = 50) -> Dict:
"""
运行爬虫
Args:
platform: 平台名称
keywords: 关键词列表
login_type: 登录方式
max_notes: 最大爬取数量
Returns:
爬取结果统计
"""
if platform not in self.supported_platforms:
raise ValueError(f"不支持的平台: {platform}")
if not keywords:
raise ValueError("关键词列表不能为空")
start_message = f"\n开始爬取平台: {platform}"
start_message += f"\n关键词: {keywords[:5]}{'...' if len(keywords) > 5 else ''} (共{len(keywords)}个)"
logger.info(start_message)
start_time = datetime.now()
try:
# 配置数据库
if not self.configure_mediacrawler_db():
return {"success": False, "error": "数据库配置失败"}
# 创建基础配置
if not self.create_base_config(platform, keywords, "search", max_notes):
return {"success": False, "error": "基础配置创建失败"}
# 判断数据库类型,确定 save_data_option
db_dialect = (config.settings.DB_DIALECT or "mysql").lower()
is_postgresql = db_dialect in ("postgresql", "postgres")
save_data_option = "postgresql" if is_postgresql else "db"
# 构建命令
cmd = [
sys.executable, "main.py",
"--platform", platform,
"--lt", login_type,
"--type", "search",
"--save_data_option", save_data_option
]
logger.info(f"执行命令: {' '.join(cmd)}")
# 切换到MediaCrawler目录并执行,捕获输出
result = subprocess.run(
cmd,
cwd=self.mediacrawler_path,
timeout=3600, # 60分钟超时
capture_output=True,
text=True,
encoding='utf-8',
errors='replace'
)
end_time = datetime.now()
duration = (end_time - start_time).total_seconds()
# 解析输出,提取实际保存的数据量
output_lines = result.stdout.split('\n') if result.stdout else []
error_lines = result.stderr.split('\n') if result.stderr else []
# 合并所有输出行用于解析
all_lines = output_lines + error_lines
# 解析统计信息
parsed_stats = self._parse_crawl_output(all_lines, error_lines)
# 创建统计信息
crawl_stats = {
"platform": platform,
"keywords_count": len(keywords),
"duration_seconds": duration,
"start_time": start_time.isoformat(),
"end_time": end_time.isoformat(),
"return_code": result.returncode,
"success": result.returncode == 0,
"notes_count": parsed_stats.get("notes_count", 0),
"comments_count": parsed_stats.get("comments_count", 0),
"errors_count": parsed_stats.get("errors_count", 0),
"output_preview": '\n'.join(output_lines[-20:]) if output_lines else "", # 最后20行输出
"error_preview": '\n'.join(error_lines[-20:]) if error_lines else "" # 最后20行错误
}
# 保存统计信息
self.crawl_stats[platform] = crawl_stats
if result.returncode == 0:
logger.info(f"{platform} 爬取完成,耗时: {duration:.1f}")
logger.info(f" 保存内容: {crawl_stats['notes_count']} 条,评论: {crawl_stats['comments_count']}")
if crawl_stats['notes_count'] == 0 and crawl_stats['comments_count'] == 0:
logger.warning(f"⚠️ {platform} 爬取成功但未保存任何数据,请检查数据库连接和保存逻辑")
# 输出部分日志用于调试
if crawl_stats['error_preview']:
logger.warning(f" 错误信息: {crawl_stats['error_preview'][:500]}")
else:
logger.error(f"{platform} 爬取失败,返回码: {result.returncode}")
if error_lines:
logger.error(f" 错误信息: {crawl_stats['error_preview'][:500]}")
return crawl_stats
except subprocess.TimeoutExpired:
logger.exception(f"{platform} 爬取超时")
return {"success": False, "error": "爬取超时", "platform": platform}
except Exception as e:
logger.exception(f"{platform} 爬取异常: {e}")
return {"success": False, "error": str(e), "platform": platform}
def _parse_crawl_output(self, output_lines: List[str], error_lines: List[str]) -> Dict:
"""解析爬取输出,提取统计信息"""
import re
stats = {
"notes_count": 0,
"comments_count": 0,
"errors_count": 0,
"login_required": False
}
# 合并所有行用于解析
all_lines = output_lines + error_lines
# 解析输出行,查找各种可能的数据保存信息
for line in all_lines:
line_lower = line.lower()
# 查找保存的内容数量(多种可能的格式)
# 例如:"保存了 10 条笔记"、"成功保存 5 条内容"、"inserted 3 records"等
if any(keyword in line_lower for keyword in ["条笔记", "条内容", "条视频", "条帖子", "条回答"]):
try:
# 提取数字,优先取第一个数字
numbers = re.findall(r'\d+', line)
if numbers:
# 如果找到多个数字,取最大的(通常是总数)
potential_count = max([int(n) for n in numbers])
if potential_count > stats["notes_count"]:
stats["notes_count"] = potential_count
except:
pass
# 查找保存的评论数量
if "条评论" in line_lower:
try:
numbers = re.findall(r'\d+', line)
if numbers:
potential_count = max([int(n) for n in numbers])
if potential_count > stats["comments_count"]:
stats["comments_count"] = potential_count
except:
pass
# 查找数据库相关错误
if any(keyword in line_lower for keyword in ["数据库", "database", "connection", "连接失败", "保存失败"]):
if "error" in line_lower or "失败" in line_lower or "异常" in line_lower:
stats["errors_count"] += 1
# 查找登录相关
if any(keyword in line_lower for keyword in ["登录", "扫码", "login", "需要登录"]):
stats["login_required"] = True
# 如果没有找到明确的保存数量,尝试从数据库操作日志中提取
if stats["notes_count"] == 0 and stats["comments_count"] == 0:
# 查找可能的数据库插入信息
for line in all_lines:
# 查找类似 "insert into" 或 "保存到" 的信息
if "insert" in line_lower or "保存到" in line_lower:
try:
numbers = re.findall(r'\d+', line)
if numbers:
# 尝试提取可能的记录数
pass # 这里可以进一步解析
except:
pass
return stats
def run_multi_platform_crawl_by_keywords(self, keywords: List[str], platforms: List[str],
login_type: str = "qrcode", max_notes_per_keyword: int = 50) -> Dict:
"""
基于关键词的多平台爬取 - 每个关键词在所有平台上都进行爬取
Args:
keywords: 关键词列表
platforms: 平台列表
login_type: 登录方式
max_notes_per_keyword: 每个关键词在每个平台的最大爬取数量
Returns:
总体爬取统计
"""
start_message = f"\n🚀 开始全平台关键词爬取"
start_message += f"\n 关键词数量: {len(keywords)}"
start_message += f"\n 平台数量: {len(platforms)}"
start_message += f"\n 登录方式: {login_type}"
start_message += f"\n 每个关键词在每个平台的最大爬取数量: {max_notes_per_keyword}"
start_message += f"\n 总爬取任务: {len(keywords)} × {len(platforms)} = {len(keywords) * len(platforms)}"
logger.info(start_message)
total_stats = {
"total_keywords": len(keywords),
"total_platforms": len(platforms),
"total_tasks": len(keywords) * len(platforms),
"successful_tasks": 0,
"failed_tasks": 0,
"total_notes": 0,
"total_comments": 0,
"keyword_results": {},
"platform_summary": {}
}
# 初始化平台统计
for platform in platforms:
total_stats["platform_summary"][platform] = {
"successful_keywords": 0,
"failed_keywords": 0,
"total_notes": 0,
"total_comments": 0
}
# 对每个平台一次性爬取所有关键词
for platform in platforms:
logger.info(f"\n📝 在 {platform} 平台爬取所有关键词")
logger.info(f" 关键词: {', '.join(keywords[:5])}{'...' if len(keywords) > 5 else ''}")
try:
# 一次性传递所有关键词给平台
result = self.run_crawler(platform, keywords, login_type, max_notes_per_keyword)
if result.get("success"):
total_stats["successful_tasks"] += len(keywords)
total_stats["platform_summary"][platform]["successful_keywords"] = len(keywords)
notes_count = result.get("notes_count", 0)
comments_count = result.get("comments_count", 0)
total_stats["total_notes"] += notes_count
total_stats["total_comments"] += comments_count
total_stats["platform_summary"][platform]["total_notes"] = notes_count
total_stats["platform_summary"][platform]["total_comments"] = comments_count
# 为每个关键词记录结果
for keyword in keywords:
if keyword not in total_stats["keyword_results"]:
total_stats["keyword_results"][keyword] = {}
total_stats["keyword_results"][keyword][platform] = result
logger.info(f" ✅ 成功: {notes_count} 条内容, {comments_count} 条评论")
else:
total_stats["failed_tasks"] += len(keywords)
total_stats["platform_summary"][platform]["failed_keywords"] = len(keywords)
# 为每个关键词记录失败结果
for keyword in keywords:
if keyword not in total_stats["keyword_results"]:
total_stats["keyword_results"][keyword] = {}
total_stats["keyword_results"][keyword][platform] = result
logger.error(f" ❌ 失败: {result.get('error', '未知错误')}")
except Exception as e:
total_stats["failed_tasks"] += len(keywords)
total_stats["platform_summary"][platform]["failed_keywords"] = len(keywords)
error_result = {"success": False, "error": str(e)}
# 为每个关键词记录异常结果
for keyword in keywords:
if keyword not in total_stats["keyword_results"]:
total_stats["keyword_results"][keyword] = {}
total_stats["keyword_results"][keyword][platform] = error_result
logger.error(f" ❌ 异常: {e}")
# 打印详细统计
finish_message = f"\n📊 全平台关键词爬取完成!"
finish_message += f"\n 总任务: {total_stats['total_tasks']}"
finish_message += f"\n 成功: {total_stats['successful_tasks']}"
finish_message += f"\n 失败: {total_stats['failed_tasks']}"
finish_message += f"\n 成功率: {total_stats['successful_tasks']/total_stats['total_tasks']*100:.1f}%"
finish_message += f"\n 总内容: {total_stats['total_notes']}"
finish_message += f"\n 总评论: {total_stats['total_comments']}"
logger.info(finish_message)
platform_summary_message = f"\n 各平台统计:"
for platform, stats in total_stats["platform_summary"].items():
success_rate = stats["successful_keywords"] / len(keywords) * 100 if keywords else 0
platform_summary_message += f"\n {platform}: {stats['successful_keywords']}/{len(keywords)} 关键词成功 ({success_rate:.1f}%), "
platform_summary_message += f"{stats['total_notes']} 条内容"
logger.info(platform_summary_message)
return total_stats
def get_crawl_statistics(self) -> Dict:
"""获取爬取统计信息"""
return {
"platforms_crawled": list(self.crawl_stats.keys()),
"total_platforms": len(self.crawl_stats),
"detailed_stats": self.crawl_stats
}
def save_crawl_log(self, log_path: str = None):
"""保存爬取日志"""
if not log_path:
log_path = f"crawl_log_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
try:
with open(log_path, 'w', encoding='utf-8') as f:
json.dump(self.crawl_stats, f, ensure_ascii=False, indent=2)
logger.info(f"爬取日志已保存到: {log_path}")
except Exception as e:
logger.exception(f"保存爬取日志失败: {e}")
if __name__ == "__main__":
# 测试平台爬虫管理器
crawler = PlatformCrawler()
# 测试配置
test_keywords = ["科技", "AI", "编程"]
result = crawler.run_crawler("xhs", test_keywords, max_notes=5)
logger.info(f"测试结果: {result}")
logger.info("平台爬虫管理器测试完成!")