"""
企业信用报告检测测试脚本
调用 python/credit_report_detector.py 进行测试
"""

import sys
import os

# 添加python目录到路径
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'python'))

from credit_report_detector import CreditReportDetector

if __name__ == '__main__':
    import sys
    
    detector = CreditReportDetector()
    
    # 如果提供了命令行参数，使用参数中的文件路径
    if len(sys.argv) > 1:
        pdf_path = sys.argv[1]
        result = detector.is_credit_report(pdf_path)
        
        if result['is_credit_report']:
            print(f"是企业信用报告")
            print(f"匹配关键词: {result['matched_keyword']}")
        else:
            print(f"不是企业信用报告")
            if result['error']:
                print(f"错误: {result['error']}")
    else:
        # 没有参数时，运行测试示例
        print("测试模式 - 检测示例文件")
        print("=" * 50)
        
        # 测试企业信用报告
        pdf_path = 'uploads/credit_reports/test_report.pdf'
        if os.path.exists(pdf_path):
            result = detector.is_credit_report(pdf_path)
            print(f"\n文件: {pdf_path}")
            print(f"是否为企业信用报告: {result['is_credit_report']}")
            print(f"匹配关键词: {result['matched_keyword']}")
            print(f"错误信息: {result['error']}")
        
        # 测试营业执照
        pdf_path = 'uploads/business_licenses/test_license.pdf'
        if os.path.exists(pdf_path):
            result = detector.is_credit_report(pdf_path)
            print(f"\n文件: {pdf_path}")
            print(f"是否为企业信用报告: {result['is_credit_report']}")
            print(f"匹配关键词: {result['matched_keyword']}")
            print(f"错误信息: {result['error']}")
        
        # 测试temp目录中的文件
        pdf_path = 'temp/莆田市城厢区拓宵贸易有限公司报告.pdf'
        if os.path.exists(pdf_path):
            result = detector.is_credit_report(pdf_path)
            print(f"\n文件: {pdf_path}")
            print(f"是否为企业信用报告: {result['is_credit_report']}")
            print(f"匹配关键词: {result['matched_keyword']}")
            print(f"错误信息: {result['error']}")