version_set.py

# /usr/bin/env python3
# -*- coding: utf-8 -*-

import os
import re
import sys
import codecs
import traceback
import chardet

MAJOR_VERSION = '0'
MINOR_VERSION = '6'
FIX_VERSION = '0'

'''
    git checkout HEAD -- CommandTrayHost/CommandTrayHost.rc CommandTrayHost/stdafx.h
'''


def local_time(zone='Asia/Shanghai'):
    from datetime import datetime
    from pytz import timezone
    other_zone = timezone(zone)
    other_zone_time = datetime.now(other_zone)
    return other_zone_time.strftime('%Y-%m-%d %H:%M:%S %Z%z')


def main():
    global MAJOR_VERSION, MINOR_VERSION, FIX_VERSION
    print(sys.argv)
    if len(sys.argv) < 2:
        print("must input build version")
        return False
    argc = len(sys.argv)
    if(argc > 1):
        build_version_number = sys.argv[1]
    print("build_version_number: ", build_version_number)
    if(argc > 2):
        MAJOR_VERSION = sys.argv[2]
    if(argc > 3):
        MINOR_VERSION = sys.argv[3]
    if(argc > 4):
        FIX_VERSION = sys.argv[4]
    # \r\n will have trouble with $ to match end
    pattern_rc = re.compile(
        r'^VALUE "ProductVersion", "\d+, \d+, \d+, \d+[\\]0"', re.M)
    rc_string = r'VALUE "ProductVersion", "{}, {}, {}, {}\\0"'.format(
        MAJOR_VERSION, MINOR_VERSION, FIX_VERSION, build_version_number)  # .encode('utf-8')

    pattern2_rc = re.compile(r'^VALUE "FileVersion", "\d+, \d+, \d+, \d+[\\]0"', re.M)
    rc_string2 = r'VALUE "FileVersion", "{}, {}, {}, {}\\0"'.format(
        MAJOR_VERSION, MINOR_VERSION, FIX_VERSION, build_version_number)

    pattern3_rc = re.compile(r'^FILEVERSION \d+, \d+, \d+, \d+', re.M)
    rc_string3 = r'FILEVERSION {}, {}, {}, {}'.format(
        MAJOR_VERSION, MINOR_VERSION, FIX_VERSION, build_version_number)

    pattern4_rc = re.compile(r'^PRODUCTVERSION \d+, \d+, \d+, \d+', re.M)
    rc_string4 = r'PRODUCTVERSION {}, {}, {}, {}'.format(
        MAJOR_VERSION, MINOR_VERSION, FIX_VERSION, build_version_number)

    pattern_stdafx_h = re.compile(
        r'^#define VERSION_NUMS L"\d+[.]\d+[.][0-9b-]+"', re.M)
    stdafx_h_string = r'#define VERSION_NUMS L"{}.{}.{}-b{}"'.format(
        MAJOR_VERSION, MINOR_VERSION, FIX_VERSION, build_version_number)  # .encode('utf-16le')

    base_dir = os.path.dirname(os.path.abspath(__file__))
    rc_file = os.path.join(base_dir, "CommandTrayHost", "CommandTrayHost.rc")
    stdafx_h_file = os.path.join(base_dir, "CommandTrayHost", "stdafx.h")

    print(stdafx_h_string, '\n', rc_string, '\n', rc_string2)

    for file_name, pattern_re, replace_string, encoding in (
        (stdafx_h_file, pattern_stdafx_h, stdafx_h_string, 'utf-8'),
        (rc_file, pattern_rc, rc_string, 'utf-16le'),
        (rc_file, pattern2_rc, rc_string2, 'utf-16le'),
        (rc_file, pattern3_rc, rc_string3, 'utf-16le'),
        (rc_file, pattern4_rc, rc_string4, 'utf-16le'),
    ):
        # CommandTrayHost.rc # stdafx.h
        print(f'opening {file_name}')
        try:
            with open(file_name, "rb") as f:
                content = f.read()
        except:
            print(f"open {file_name} failed!")
            return False
        print(len(content), end=" ")

        detect = chardet.detect(content)
        print(detect, end=' ')
        encoding = detect['encoding']

        if encoding.lower().startswith('utf-16'):
            bom = codecs.BOM_UTF16_LE  # ''.encode('utf-16')
        elif encoding.lower().startswith('utf-8'):
            bom = codecs.BOM_UTF8  # ''.encode('utf-8-sig')

        if content.startswith(bom):
            print("bom:", repr(bom), end=' ')
            # content = content[len(bom):]

        try:
            content = content.decode(encoding)
        except:
            traceback.print_exc()
            print("binary.decode failed!")

        if pattern_re.search(content) is None:
            print("search failed")
            return False
        print(len(content), end=" sub-> ")
        content = pattern_re.sub(replace_string, content)
        if file_name.endswith("stdafx.h"):
            content = content.replace(r'#define BUILD_TIME_CN __TIMESTAMP__',
                                      f'#define BUILD_TIME_CN "{local_time()}"'
                                      )
            content = content.replace(r'#define BUILD_TIME_EN __TIMESTAMP__',
                                      f'#define BUILD_TIME_EN "{local_time("UTC")}"'
                                      )
        print(len(content), end=" ")

        if file_name.endswith(".rc"):
            encoding = 'utf-16'

        if encoding == 'utf-16le':
            content = bom + content.encode(encoding)
        else:
            content = content.encode(encoding)

        print(len(content), encoding)
        with open(file_name, "wb") as f:
            f.write(content)
        print(file_name, "repalced!\n")

    return True


if __name__ == '__main__':
    print(f"""{__file__} <build_number> [MAJOR_VERSION] [MINOR_VERSION] [FIX_VERSION]
[MAJOR_VERSION] [MINOR_VERSION] [FIX_VERSION] are optional.""")
    print(sys.version_info)
    if (sys.version_info < (3, 0)):
        sys.exit(2)
    if not main():
        sys.exit(1)