#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
海外中国投资项目分析简报 - 警报级别检查脚本
检查简报内容是否触发警报
"""

import argparse
from datetime import datetime
from pathlib import Path


RED_ALERT_TERMS = [
    '伤亡', '死亡', '袭击', '政变', '内战', '撤离', '征收',
    '武装', '恐怖', '绑架', '3 人以上', '多人死亡'
]

ORANGE_ALERT_TERMS = [
    '抗议', '罢工', '停工', '制裁', '调查', '反对',
    '大规模', '百人', '重要官员', '负面表态'
]


def check_alerts(input_file: str, output_dir: str) -> list:
    """检查文件中的警报"""
    content = Path(input_file).read_text(encoding='utf-8')
    alerts = []
    
    for term in RED_ALERT_TERMS:
        if term in content:
            alerts.append({
                'level': 'red',
                'term': term,
                'time': datetime.now().isoformat()
            })
            break
    
    for term in ORANGE_ALERT_TERMS:
        if term in content:
            alerts.append({
                'level': 'orange',
                'term': term,
                'time': datetime.now().isoformat()
            })
            break
    
    return alerts


def main():
    parser = argparse.ArgumentParser(description='警报检查脚本')
    parser.add_argument('--input', required=True, help='输入文件')
    parser.add_argument('--output', required=True, help='输出目录')
    
    args = parser.parse_args()
    
    alerts = check_alerts(args.input, args.output)
    
    if alerts:
        print(f"发现 {len(alerts)} 条警报:")
        for alert in alerts:
            print(f"  - {alert['level']}: {alert['term']}")
    else:
        print("未发现警报")


if __name__ == '__main__':
    main()