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 + '"')