"""
测试数据管理工具
"""
import os
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent))

from test_data_manager.manager import TestDataManager
from test_data_manager.config import TestDataManagerConfig


def main():
    config = TestDataManagerConfig()
    manager = TestDataManager(config)
    
    if len(sys.argv) < 2:
        print("用法: python main.py {generate|clean|reset|status}")
        sys.exit(1)
    
    command = sys.argv[1]
    
    if command == "generate":
        manager.generate_test_data()
    elif command == "clean":
        manager.clean_test_data()
    elif command == "reset":
        manager.reset_test_data()
    elif command == "status":
        manager.get_status()
    else:
        print(f"未知命令: {command}")
        sys.exit(1)


if __name__ == "__main__":
    main()