张翔
40 lines
872 B
Python
Executable File
40 lines
872 B
Python
Executable File
#!/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())
|