diff --git a/build_scripts/append_to_version_file.py b/build_scripts/append_to_version_file.py index 3705fe8..99d33ef 100644 --- a/build_scripts/append_to_version_file.py +++ b/build_scripts/append_to_version_file.py @@ -8,14 +8,21 @@ version_file_path = path.join(current_path, "compile_version_string.txt") def get_version_string(): with open(version_file_path, "r") as file: contents = file.readline() - return contents.strip('"') + return contents.strip('"') def set_version_string(new_version_string): remove(version_file_path) with open(version_file_path, "w") as file: file.write("\"" + new_version_string + "\"") -if __name__ == "__main__": +def main(): current_version = get_version_string() - new_version = current_version + "+" + argv[1] - set_version_string(new_version) \ No newline at end of file + if "-release" in current_version.lower(): + current_version.replace("-release", "") + else: + new_version = current_version + "+" + argv[1] + + set_version_string(new_version) + +if __name__ == "__main__": + main()