update_version.py

import subprocess
import json
import re
import os


def get_changed_files():
    # 获取当前分支与主分支的差异文件列表
    changed_files = (
        subprocess.check_output(
            ["git", "diff-tree", "--name-only", "--no-commit-id", "-r", "HEAD"]
        )
        .decode()
        .split()
    )
    return changed_files


def update_version_in_init(file_path, current_version, commit_hash):
    with open(file_path, "r", encoding="utf-8") as file:
        content = file.read()

    new_version = f"{current_version}-{commit_hash}"
    if old_version_match := re.search(r'version\s*=\s*"(.*?)"', content):
        old_version = old_version_match[1]
        old_version = old_version.split("-")[0]
        if old_version != current_version:
            return old_version

        new_content = re.sub(
            r'version\s*=\s*"(.*?)"', f'version="{new_version}"', content
        )

        with open(file_path, "w", encoding="utf-8") as file:
            file.write(new_content)

        return new_version
    return current_version


def process_plugins(plugins, changed_files):
    commit_hash = (
        subprocess.check_output(["git", "rev-parse", "--short=7", "HEAD"])
        .decode()
        .strip()
    )

    for plugin_name, plugin_info in plugins.items():
        module_path = plugin_info["module_path"].replace(".", "/")
        is_dir = plugin_info["is_dir"]
        current_version = plugin_info["version"]
        current_version = current_version.split("-")[0]

        if is_dir:
            for changed_file in changed_files:
                if changed_file.startswith(module_path):
                    init_path = os.path.join(module_path, "__init__.py")
                    if os.path.exists(init_path):
                        new_version = update_version_in_init(
                            init_path, current_version, commit_hash
                        )
                        plugin_info["version"] = new_version
        else:
            file_path = f"{module_path}.py"
            if file_path in changed_files:
                new_version = f"{current_version}-{commit_hash}"
                plugin_info["version"] = new_version


def main():
    plugins_json_path = "plugins.json"
    with open(plugins_json_path, "r", encoding="utf-8") as file:
        plugins = json.load(file)

    changed_files = get_changed_files()
    process_plugins(plugins, changed_files)

    with open(plugins_json_path, "w", encoding="utf-8") as file:
        json.dump(plugins, file, indent=4, ensure_ascii=False)


if __name__ == "__main__":
    main()