""" 配置模块：系统常量、RSS源、关键词库、评分维度、分类定义、AI配置加载 """ import os import json from pathlib import Path # ============================================================================= # 军事科技 RSS 订阅源 # 优先读取 config/rss_feeds.txt 文件，文件不存在则使用默认12个源 # ============================================================================= def load_rss_feeds() -> list: """从配置文件读取RSS源，格式每行：名称|URL，#开头为注释""" default_feeds = [ {"name": "Defense One", "url": "https://www.defenseone.com/rss/all/"}, {"name": "NEWUAS", "url": f"{RSS_PROXY_BASE}/feed/MP_WXS_3246130840.rss"}, {"name": "Seapower", "url": "https://seapowermagazine.org/feed/"}, {"name": "The War Zone", "url": "https://www.twz.com/feed"}, {"name": "国防科技要闻", "url": f"{RSS_PROXY_BASE}/feed/MP_WXS_3921202880.rss"}, {"name": "战略前沿技术", "url": f"{RSS_PROXY_BASE}/feed/MP_WXS_3271903347.rss"}, {"name": "无人机邦", "url": f"{RSS_PROXY_BASE}/feed/MP_WXS_3957361285.rss"}, {"name": "浮空飞行器", "url": f"{RSS_PROXY_BASE}/feed/MP_WXS_3291643707.rss"}, {"name": "海鹰资讯", "url": f"{RSS_PROXY_BASE}/feed/MP_WXS_3091381580.rss"}, {"name": "渊亭防务", "url": f"{RSS_PROXY_BASE}/feed/MP_WXS_3865628826.rss"}, {"name": "电波之矛", "url": f"{RSS_PROXY_BASE}/feed/MP_WXS_3004998095.rss"}, {"name": "龙牙的一座山", "url": f"{RSS_PROXY_BASE}/feed/MP_WXS_3517933009.rss"}, ] # RSS配置文件路径：项目根目录/config/rss_feeds.txt config_file = Path(__file__).resolve().parent.parent.parent / "config" / "rss_feeds.txt" if not config_file.exists(): return default_feeds feeds = [] with open(config_file, 'r', encoding='utf-8') as f: for line in f: line = line.strip() # 跳过空行和注释行 if not line or line.startswith('#'): continue # 按|分割，确保格式正确 parts = line.split('|', 1) if len(parts) == 2: name = parts[0].strip() url = parts[1].strip() if name and url: feeds.append({"name": name, "url": url}) # 如果读取到的源为空，返回默认源 return feeds if feeds else default_feeds # ============================================================================= # RSS 代理基础 URL（read_env 尚未定义，直接读文件/file和ennv） # ============================================================================= def _load_rss_proxy_base(): default = "https://werss.yynnice.top" env_val = os.environ.get("RSS_PROXY_BASE") if env_val: return env_val env_file = Path(__file__).resolve().parent.parent.parent / "config" / ".env" if env_file.exists(): with open(env_file, 'r', encoding='utf-8') as f: for line in f: line = line.strip() if line and not line.startswith('#') and line.startswith('RSS_PROXY_BASE='): _, _, val = line.partition('=') return val.strip().strip('"').strip("'") return default RSS_PROXY_BASE = _load_rss_proxy_base() RSS_FEEDS = load_rss_feeds() WECHAT_SOURCES = {'NEWUAS', '国防科技要闻', '战略前沿技术', '无人机邦', '浮空飞行器', '海鹰资讯', '渊亭防务', '电波之矛', '龙牙的一座山'} # ============================================================================= # 抓取配置 # ============================================================================= FETCH_TIMEOUT = 30 MAX_THREADS = 15 # 抓取并发数，100+源建议15-20 AI_CONCURRENCY = 8 # AI调用并发数，根据API限流调整 TIME_WINDOW_HOURS = 24 # ============================================================================= # 关键词库（44个） # ============================================================================= KEYWORDS = [ "首飞", "曝光", "美军", "解放军", "新型", "研发", "部署", "军演", "导弹", "战机", "舰艇", "雷达", "芯片", "突破", "测试", "服役", "列装", "武器", "装备", "军事", "武器系统", "冲突", "封锁", "威慑", "无人机", "AI", "人工智能", "战争", "以色列", "伊朗", "俄罗斯", "乌克兰", "航母", "潜艇", "坦克", "防空", "反导", "激光", "电磁", "卫星", "航天", "蜂群", "拦截", "打击", "攻击", "作战", "战略", "战术", "军工" ] # ============================================================================= # AI评分维度（4维度加权） # ============================================================================= SCORE_DIMENSIONS = [ {"name": "技术突破性", "weight": 0.35, "description": "是否有新技术、新武器、新架构曝光，1-10分"}, {"name": "战略价值", "weight": 0.30, "description": "对地缘政治、军事格局的影响程度，1-10分"}, {"name": "信息可信度", "weight": 0.20, "description": "来源权威性、数据可验证性，1-10分"}, {"name": "时效性", "weight": 0.15, "description": "事件的新鲜程度，1-10分"}, ] # ============================================================================= # 分类定义 # ============================================================================= CATEGORIES = { "today_must_read": {"name": "🔥 今日必看", "desc": "最重磅、最紧急的头条要闻"}, "equipment": {"name": "⚙️ 装备动态", "desc": "新型武器、技术突破、军备交付"}, "conflict": {"name": "💥 地区冲突", "desc": "热点战事、边境对峙、反恐行动"}, "strategy": {"name": "🎯 战略政策", "desc": "国防战略、军演部署、外交安全"}, } def _read_env_file(file_path): """从.env文件读取配置，返回 {key: value} 字典""" result = {} if file_path.exists(): with open(file_path, 'r', encoding='utf-8') as f: for line in f: line = line.strip() if line and not line.startswith('#') and '=' in line: key, _, value = line.partition('=') value = value.strip().strip('"').strip("'") result[key] = value return result def load_ai_config(): """加载AI配置（环境变量 > 项目config/.env > 硬编码默认值）""" api_key = None base_url = "https://api.openai.com/v1" model = "gpt-4o-mini" webzine_model = "deepseek-v3.2" local_env = Path(__file__).resolve().parent.parent.parent / "config" / ".env" values = _read_env_file(local_env) if values.get('OPENAI_API_KEY'): api_key = values['OPENAI_API_KEY'] if values.get('OPENAI_API_BASE'): base_url = values['OPENAI_API_BASE'] if values.get('OPENAI_MODEL'): model = values['OPENAI_MODEL'] if values.get('WEBZINE_MODEL'): webzine_model = values['WEBZINE_MODEL'] if os.environ.get('OPENAI_API_KEY'): api_key = os.environ['OPENAI_API_KEY'] if os.environ.get('OPENAI_API_BASE'): base_url = os.environ['OPENAI_API_BASE'] if os.environ.get('OPENAI_MODEL'): model = os.environ['OPENAI_MODEL'] if os.environ.get('WEBZINE_MODEL'): webzine_model = os.environ['WEBZINE_MODEL'] return api_key, base_url, model, webzine_model def load_push_config(): """加载推送配置（环境变量 > config.json > 默认值）""" config = { "enable_wechat_push": False, "enable_feishu_push": False, "wechat_key": "", "wechat_target": "", "feishu_webhook": "", } config_file = Path(__file__).resolve().parent.parent.parent / "config" / "config.json" values = _read_env_file(Path(__file__).resolve().parent.parent.parent / "config" / ".env") if config_file.exists(): try: with open(config_file, 'r', encoding='utf-8') as f: json_config = json.load(f) config["enable_wechat_push"] = json_config.get("enable_wechat_push", False) config["enable_feishu_push"] = json_config.get("enable_feishu_push", False) except Exception: pass if values.get("WECHAT_ACCOUNT"): config["wechat_key"] = values["WECHAT_ACCOUNT"] if values.get("WECHAT_TARGET"): config["wechat_target"] = values["WECHAT_TARGET"] if values.get("FEISHU_WEBHOOK"): config["feishu_webhook"] = values["FEISHU_WEBHOOK"] if os.environ.get("WECHAT_ACCOUNT"): config["wechat_key"] = os.environ["WECHAT_ACCOUNT"] if os.environ.get("WECHAT_TARGET"): config["wechat_target"] = os.environ["WECHAT_TARGET"] if os.environ.get("FEISHU_WEBHOOK"): config["feishu_webhook"] = os.environ["FEISHU_WEBHOOK"] return config