run.py

import os
import shutil
from codesp import simulate_star_trajectory





def main():
    # test()
    simulate_star_trajectory(semi_major_axis=1.0, eccentricity=0.1, period=365.25, start_date='2023/1/1', end_date='2024/1/1', step_days=10)

    # # 定义初始路径和目标路径
    # source_dir = '/workspace/input/'
    # target_dir = '/workspace/output/'

    # # 确保源路径存在
    # if not os.path.exists(source_dir):
    #     print(f"错误：源路径 {source_dir} 不存在。")
    # else:
    #     # 创建目标路径（如果不存在）
    #     os.makedirs(target_dir, exist_ok=True)

    #     # 获取源路径下的所有文件（不递归子目录）
    #     for file_name in os.listdir(source_dir):
    #         source_file = os.path.join(source_dir, file_name)
    #         target_file = os.path.join(target_dir, file_name)

    #         # 只拷贝文件，忽略子目录
    #         if os.path.isfile(source_file):
    #             shutil.copy2(source_file, target_file)  # copy2 保留元数据（如修改时间）
    #             print(f"已拷贝: {source_file} -> {target_file}")

    #     print("拷贝完成。")

if __name__ == '__main__':
    main()