import re
import sys

def set_version(infocs, target_version):
    if not infocs or not target_version:
        raise Exception('invalid param')
        return

    with open(infocs, "r+") as f:
        assemblyinfo_cs = f.read()

        pattern_1 = re.compile(r'AssemblyVersion\("[0-9]+(\.([0-9]+|\*)){1,3}"\)', re.MULTILINE)
        pattern_2 = re.compile(r'AssemblyFileVersion\("[0-9]+(\.([0-9]+|\*)){1,3}"\)', re.MULTILINE)

        sub1 = r'AssemblyVersion("{}")'.format(target_version)
        sub2 = r'AssemblyFileVersion("{}")'.format(target_version)

        phase_1 = re.sub(pattern_1, sub1, assemblyinfo_cs)
        phase_2 = re.sub(pattern_2, sub2, phase_1)

        f.seek(0)
        f.write(phase_2)
        f.truncate()