oguzhankoral
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
import re
|
|
import sys
|
|
import os
|
|
|
|
|
|
def sanitize_version(tag):
|
|
"""Extracts the first three numeric segments from a tag string, because PowerBI is..."""
|
|
parts = re.findall(r"\d+", tag)
|
|
return ".".join(parts[:3]) if len(parts) >= 3 else tag
|
|
|
|
def patch_connector(tag):
|
|
"""Patches the connector version within the data connector file"""
|
|
sanitized_tag = sanitize_version(tag)
|
|
pq_file = os.path.join(os.path.dirname(__file__), "src", "powerbi-data-connector", "Speckle.pq")
|
|
|
|
with open(pq_file, "r") as file:
|
|
lines = file.readlines()
|
|
|
|
for (index, line) in enumerate(lines):
|
|
if '[Version = "3.0.0"]' in line:
|
|
lines[index] = f'[Version = "{sanitized_tag}"]\n'
|
|
print(f"Patched connector version number in {pq_file}")
|
|
break
|
|
|
|
with open(pq_file, "w") as file:
|
|
file.writelines(lines)
|
|
|
|
|
|
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)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|