SecStore/update_version.py at master · SECTL/SecStore · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import os
import re

def get_version_from_env():
    # 🌟 星野小贴士：这里获取版本号并确保它是有效的数字格式哦 ~
    original_version = os.getenv('VERSION', 'v0.0.0.0')
    stripped_version = re.sub(r'^v', '', original_version)

    # 🔍 白露检查：如果版本号包含非数字字符，提取数字部分
    numeric_version = re.search(r'^(\d+)(?:\.(\d+))?(?:\.(\d+))?(?:\.(\d+))?', stripped_version)
    if numeric_version:
        # 组装有效的X.X.X.X格式版本号
        parts = numeric_version.groups(default='0')
        stripped_version = '.'.join(parts)
    else:
        # 如果无法提取，使用默认版本
        stripped_version = '0.0.0.0'
        original_version = 'v0.0.0.0'

    return original_version, stripped_version

def update_version_info(version):
    major, minor, patch, build = map(int, version.split('.'))
    version_tuple = (major, minor, patch, build)
    version_str = f"{major}.{minor}.{patch}.{build}"

    with open('version_info.txt', 'r', encoding='utf-8') as f:
        content = f.read()

    content = re.sub(r'filevers=\(\d, \d, \d, \d\)', f'filevers={version_tuple}', content)
    content = re.sub(r'prodvers=\(\d, \d, \d, \d\)', f'prodvers={version_tuple}', content)

    content = re.sub(r'StringStruct\(u\'FileVersion\', u\'\d+\.\d+\.\d+\.\d+\'\)', f'StringStruct(u\'FileVersion\', u\'{version_str}\')', content)
    content = re.sub(r'StringStruct\(u\'ProductVersion\', u\'\d+\.\d+\.\d+\.\d+\'\)', f'StringStruct(u\'ProductVersion\', u\'{version_str}\')', content)

    with open('version_info.txt', 'w', encoding='utf-8') as f:
        f.write(content)

def update_config_py(version):
    with open('app/common/config.py', 'r', encoding='utf-8') as f:
        content = f.read()

    content = re.sub(r'VERSION = "v?\d+\.\d+\.\d+\.\d+"', f'VERSION = "{version}"', content)

    with open('app/common/config.py', 'w', encoding='utf-8') as f:
        f.write(content)

if __name__ == '__main__':
    original_version, stripped_version = get_version_from_env()
    update_version_info(stripped_version)
    update_config_py(original_version)