iOS自动更新版本号脚本

由于经常在公司代码持续集成平台构建代码时候忘记更新版本号，经常浪费一次宝贵的(长长的)构建时间，现在Mac用的工具是 Cornerstone，可以支持 commit 前后执行脚本。所以写了这个脚本，在commit之后执行，用于自动更新版本号到svn。
Usage：

1，保证在 terminal 下面可以正常使用 svn 命令提交目标项目代码

2，修改配置常量 TARGET_SRC_ROOT，TARGET_PROJECT_NAMES，TARGET_PLIST_FILES

3，根据自己项目需要修改 generate_version_code 逻辑

4，在Cornerstone 的 commit 界面将本脚本设置为提交后执行

# -*- coding: utf-8 -*-

import os
import xml.etree.cElementTree as etree
import datetime

TARGET_SRC_ROOT = "../src"
TARGET_PROJECT_NAMES = ["Demo"]
TARGET_PLIST_FILES = ["Info1.plist", "Info2.plist"]


def get_version_value_node(root):
    dict_node = root.find("dict")
    if not dict_node:
        return False

    version_key_node = None
    version_value_node = None
    for key_value_node in dict_node:
        text = key_value_node.text
        if text == "CFBundleVersion":
            version_key_node = key_value_node
            continue
        if version_key_node is not None:
            version_value_node = key_value_node
            break

    if version_value_node is None:
        return

    return version_value_node


def replace_version_code(file_path, old_version_code, new_version_code):
    file_content = None
    with open(file_path, "rb") as file:
        file_content = file.read()

    if not isinstance(file_content, str):
        return False

    file_content = file_content.replace(old_version_code, new_version_code)
    with open(file_path, "wb") as file:
        file.write(file_content)

    return True


# 根据旧的版本号生成新的版本号
# 项目要求格式是 YYMMDD + Build，如 16010102
# 表示 2016 年 1 月 1 日 第二次构建的版本
def generate_version_code(old_version_code):
    if not isinstance(old_version_code, str):
        return None

    if len(old_version_code) != 8:
        return None

    date_code = old_version_code[0:6]
    build_code = old_version_code[6:]
    int_build_code = int(build_code)

    today = datetime.date.today()
    new_date_code = today.strftime("%Y%m%d")
    new_date_code = new_date_code[2:]
    if new_date_code == date_code:
        int_build_code += 1
    else:
        int_build_code = 1

    new_build_code = str(int_build_code).zfill(2)
    new_version_code = new_date_code + new_build_code

    return new_version_code


def update_version(plist_file_path):
    if not isinstance(plist_file_path, str):
        return False

    xml_tree = etree.ElementTree(file=plist_file_path)
    if not xml_tree:
        return False

    root = xml_tree.getroot()
    version_value_node = get_version_value_node(root)
    if version_value_node is None:
        return False

    version_code = version_value_node.text
    if not isinstance(version_code, str):
        return False

    new_version_code = generate_version_code(version_code)
    result = replace_version_code(plist_file_path, version_code, new_version_code)

    return result


def file_log(message):
    current_dir = os.path.realpath(os.path.dirname(__file__))
    path = os.path.join(current_dir, "log.txt")
    with open(path, "a+") as f:
        f.write(message)
        f.write("\r\n")


def commit_files(plist_files):
    commit_command = """svn ci -m "* auto update version script" """
    for plist_file in plist_files:
        commit_command += '"' + plist_file + '" '
    os.system(commit_command)
    file_log("exec command " + commit_command)


def main():
    current_dir = os.path.realpath(os.path.dirname(__file__))
    plist_files = []
    for project_name in TARGET_PROJECT_NAMES:
        path = os.path.join(TARGET_SRC_ROOT, project_name)
        path = os.path.join(current_dir, path)
        for plist_file in TARGET_PLIST_FILES:
            plist_path = os.path.join(path, plist_file)
            plist_path = os.path.abspath(plist_path)
            update_version(plist_path)
            plist_files.append(plist_path)
    commit_files(plist_files)


if __name__ == '__main__':
    main()