#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
使用小红书 MCP 爬取北京展览信息
"""

import json
import requests

# 小红书 cookies
XHS_COOKIES = "acw_tc=0a0bb41a17726804792802066ef22266fb6b0216da2a3e9f089735f8353b77; abRequestId=b5346cbb-6db3-5645-accc-df5d27fd9362; webBuild=5.13.1; xsecappid=xhs-pc-web; a1=19cbbfd8f46tok3grdu3mmi72tpiihd7co02rf9oa30000122754; webId=68ac71cf3f14eb4a280b442b71aad7e5; gid=yjSDDifj0fSfyjSDDifYilij4K9lTqkFf7q68l063WJ9UJq833xfWF888yJJW248dDqfjJ0Y; web_session=040069b8dcb7aa9bcf6957bd9d3b4b7b5c866c; id_token=VjEAAPwuVKkxM5M3tgTrefheWsAsAIisJtFuRYQM3EFnhkneE3Zag62PzVeUMmRAgCOznnJXrYICwToncTBIL4u7bKSd7M8QMVA0TgyLKo+Oknjg00IY1MQziJXeiutd2NTcUd+B; unread={%22ub%22:%2269a8d5d3000000002801e4e2%22%2C%22ue%22:%2269a85762000000002603e4ef%22%2C%22uc%22:30}; loadts=1772680707856"

def search_xiaohongshu(keyword):
    """搜索小红书笔记"""
    
    # 使用 mcporter 调用小红书 MCP
    import subprocess
    
    cmd = [
        "mcporter", "call",
        f'xiaohongshu.search_feeds(keyword: "{keyword}")'
    ]
    
    try:
        result = subprocess.run(cmd, capture_output=True, text=True, timeout=60)
        return result.stdout, result.stderr
    except subprocess.TimeoutExpired:
        return None, "Timeout"
    except Exception as e:
        return None, str(e)


if __name__ == "__main__":
    print("搜索小红书：北京展览 2026 年 3 月")
    print("=" * 50)
    
    stdout, stderr = search_xiaohongshu("北京展览 2026 年 3 月")
    
    if stdout:
        print("✅ 成功获取数据")
        print(f"\n数据长度：{len(stdout)} 字符")
        
        # 保存结果
        with open("data/xiaohongshu_result.json", "w", encoding="utf-8") as f:
            f.write(stdout)
        
        print("\n结果已保存到：data/xiaohongshu_result.json")
    else:
        print(f"❌ 失败：{stderr}")