#!/usr/bin/env python3
"""
测试数据清理脚本
"""
import sys
import os
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent.parent / "test-data-manager"))

from manager import TestDataManager
from config import TestDataManagerConfig
import logging

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


def main():
    config = TestDataManagerConfig()
    manager = TestDataManager(config)
    
    try:
        logger.info("开始清理测试数据...")
        manager.clean_test_data()
        
        logger.info("测试数据清理完成！")
        
        return 0
    except Exception as e:
        logger.error(f"清理测试数据失败: {e}")
        return 1
    finally:
        manager.close()


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