MegEngine/imperative/python/gen_version.py

import argparse
import os
import subprocess

def get_git_commit(src_dir):
    try:
        return subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=src_dir).decode('ascii').strip()
    except Exception:
        return 'unknown'

def get_mge_version(version_txt_path):
    v = {}
    with open(version_txt_path) as fp:
        exec(fp.read(), v)
    return v

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="generate version.py to build path")
    parser.add_argument("--output", type=str, required=True)
    parser.add_argument("--major", type=int, required=True)
    parser.add_argument("--minor", type=int, required=True)
    parser.add_argument("--patch", type=int, required=True)
    parser.add_argument("--rc", type=str, required=False)
    parser.add_argument("--internal", action='store_true')
    args = parser.parse_args()
    python_dir = os.path.dirname(__file__)
    commit_id = get_git_commit(python_dir)
    mge_ver = str(args.major) + "." + str(args.minor) + "." + str(args.patch)
    if args.rc is not None:
        mge_ver += args.rc
    with open(args.output, 'w') as f:
        f.write("__version__ = '{}'\n".format(mge_ver))
        f.write("git_version = {}\n".format(repr(commit_id)))
        if args.internal:
            f.write("__internal__ = True\n")