import ast
import importlib.metadata

def extract_imports_from_file(file_path):
    """提取Python文件中的所有庫導(dǎo)入"""
    with open(file_path, "r", encoding='utf-8') as file:  # 指定文件編碼為UTF-8
        tree = ast.parse(file.read(), filename=file_path)

    imports = set()
    for node in ast.walk(tree):
        if isinstance(node, ast.Import):
            for alias in node.names:
                imports.add(alias.name.split('.')[0])
        elif isinstance(node, ast.ImportFrom):
            # 忽略相對(duì)導(dǎo)入的情況
            if node.module:
                imports.add(node.module.split('.')[0])

    return imports

def get_package_versions(imports):
    """獲取導(dǎo)入庫的版本信息"""
    versions = {}
    for package_name in imports:
        try:
            version = importlib.metadata.version(package_name)  # 使用importlib.metadata替代pkg_resources
            versions[package_name] = version
        except importlib.metadata.PackageNotFoundError:
            print(f"Warning: Could not find version for {package_name}")
            versions[package_name] = "Not Found"
        except Exception as e:
            print(f"Error: Unexpected error while getting version for {package_name}: {e}")
            versions[package_name] = "Error"
    return versions

def save_versions_to_file(versions, output_file='imported_packages_versions.txt'):
    """將庫的版本信息保存到文件"""
    with open(output_file, 'w', encoding='utf-8') as f:  # 確保輸出文件也使用UTF-8編碼
        for package, version in versions.items():
            f.write(f"{package}=={version}\n")
    print(f"Versions file '{output_file}' has been generated successfully.")

# 使用示例
file_path = r"D:\pythonY\pythonProject\New_network_plus_new_loss_function.py"  # 修改為你的Python腳本路徑
imports = extract_imports_from_file(file_path)
versions = get_package_versions(imports)
save_versions_to_file(versions)