KatKatKateryna
82 lines
3.1 KiB
Python
82 lines
3.1 KiB
Python
import re
|
|
import sys
|
|
|
|
def patch_installer(tag):
|
|
"""Patches the installer with the correct connector version and specklepy version"""
|
|
iss_file = "speckle-sharp-ci-tools/arcgis.iss"
|
|
setup_whl_file = "setup.py"
|
|
conda_file = "speckle_arcgis_installer/conda_clone_activate.py"
|
|
#toolbox_install_file = "speckle_arcgis_installer/toolbox_install.py"
|
|
toolbox_manual_install_file = "speckle_arcgis_installer/toolbox_install_manual.py"
|
|
plugin_start_file = "speckle_toolbox/esri/toolboxes/speckle/speckle_arcgis.py"
|
|
|
|
#py_tag = get_specklepy_version()
|
|
with open(iss_file, "r") as file:
|
|
lines = file.readlines()
|
|
for i, line in enumerate(lines):
|
|
if "#define AppVersion " in line:
|
|
lines[i] = f'#define AppVersion "{tag.split("-")[0]}"\n'
|
|
if "#define AppInfoVersion " in line:
|
|
lines[i] = f'#define AppInfoVersion "{tag}"\n'
|
|
with open(iss_file, "w") as file:
|
|
file.writelines(lines)
|
|
print(f"Patched installer with connector v{tag} and specklepy ")
|
|
file.close()
|
|
|
|
with open(setup_whl_file, "r") as file:
|
|
lines = file.readlines()
|
|
for i, line in enumerate(lines):
|
|
if "version=" in line:
|
|
lines[i] = f'\t\t\tversion="{tag.split("-")[0]}",\n'
|
|
break
|
|
with open(setup_whl_file, "w") as file:
|
|
file.writelines(lines)
|
|
print(f"Patched whl setup with connector v{tag} and specklepy ")
|
|
file.close()
|
|
|
|
with open(plugin_start_file, "r") as file:
|
|
lines = file.readlines()
|
|
for i, line in enumerate(lines):
|
|
if 'self.version = ' in line:
|
|
lines[i] = lines[i].split("\"")[0] + "\"" + tag.split('-')[0] + "\"" + lines[i].split("\"")[2]
|
|
break
|
|
with open(plugin_start_file, "w") as file:
|
|
file.writelines(lines)
|
|
print(f"Patched GIS start file with connector v{tag} and specklepy ")
|
|
file.close()
|
|
|
|
|
|
def whlFileRename(fileName: str):
|
|
with open(fileName, "r") as file:
|
|
lines = file.readlines()
|
|
for i, line in enumerate(lines):
|
|
if "-py3-none-any.whl" in line and '.sort' not in line:
|
|
p1 = line.split("-py3-none-any.whl")[0].split("-")[0]
|
|
p2 = f'{tag.split("-")[0]}'
|
|
p3 = line.split("-py3-none-any.whl")[1]
|
|
lines[i] = p1+"-"+p2+"-py3-none-any.whl"+p3
|
|
with open(fileName, "w") as file:
|
|
file.writelines(lines)
|
|
print(f"Patched toolbox_installer with connector v{tag} and specklepy ")
|
|
file.close()
|
|
|
|
whlFileRename(conda_file)
|
|
#whlFileRename(toolbox_install_file)
|
|
whlFileRename(toolbox_manual_install_file)
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) < 2:
|
|
return
|
|
|
|
tag = sys.argv[1]
|
|
if not re.match(r"([0-9]+)\.([0-9]+)\.([0-9]+)", tag):
|
|
raise ValueError(f"Invalid tag provided: {tag}")
|
|
|
|
print(f"Patching version: {tag}")
|
|
#patch_connector(tag.split("-")[0]) if I need to edit a connector file
|
|
patch_installer(tag)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |