#!/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.reset_test_data()
        
        status = manager.get_status()
        logger.info(f"测试数据重置完成！当前状态：{status}")
        
        return 0
    except Exception as e:
        logger.error(f"重置测试数据失败: {e}")
        return 1
    finally:
        manager.close()


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