from cross_validation import DataValidator
import sys

def main():
    """主函数"""
    import argparse
    
    parser = argparse.ArgumentParser(
        description='从xtask导出数据中抽取牌面进行EHS验证',
        formatter_class=argparse.RawDescriptionHelpFormatter
    )
    
    parser.add_argument('--river-samples', type=int, default=10, help='River样本数 (默认: 10)')
    parser.add_argument('--turn-samples', type=int, default=5, help='Turn样本数 (默认: 5)')
    parser.add_argument('--flop-samples', type=int, default=3, help='Flop样本数 (默认: 3)')
    parser.add_argument('--data-path', type=str, default='ehs_data', help='数据路径 (默认: ehs_data)')
    
    args = parser.parse_args()
    

    validator = DataValidator(data_path=args.data_path)
    results = validator.run_full_validation(
        river_samples=args.river_samples,
        turn_samples=args.turn_samples,
        flop_samples=args.flop_samples
    )
    
    return 0 if results['overall_success'] else 1


if __name__ == '__main__':
    sys.exit(main())