import os

import re

import argparse

def replace_in_file(filepath, regex, sub):

with open(filepath, 'r') as f:

content = f.read()

content_new = re.sub(regex, sub, content, flags=re.M)

with open(filepath, "w") as f:

f.write(content_new)

dir_path = os.path.dirname(os.path.realpath(__file__))

parser = argparse.ArgumentParser()

parser.add_argument('--version', help='version to replace')

args = parser.parse_args()

ver = args.version

print("using version " + ver)

about_file = os.path.join(dir_path, os.pardir, "mergin", "version.py")

print("patching " + about_file)

replace_in_file(about_file, "__version__\s=\s\".*", "__version__ = \"" + ver + "\"")

setup_file = os.path.join(dir_path, os.pardir, "setup.py")

print("patching " + setup_file)

replace_in_file(setup_file, "version='.*", "version='" + ver + "',")