diff --git a/daemon/scripts/core-service-update b/daemon/scripts/core-service-update new file mode 100755 index 00000000..6d0be06c --- /dev/null +++ b/daemon/scripts/core-service-update @@ -0,0 +1,51 @@ +#!/usr/bin/env python +import argparse +import re +from io import TextIOWrapper + + +def parse_args() -> argparse.Namespace: + parser = argparse.ArgumentParser( + description=f"Helps transition older CORE services to work with newer versions") + parser.add_argument("-f", "--file", dest="file", type=argparse.FileType("r"), + help=f"service file to update") + return parser.parse_args() + + +def update_service(service_file: TextIOWrapper) -> None: + update = [] + for line in service_file.readlines(): + # update service attributes + line = re.sub(r"^(\s+)_([a-z])", r"\1\2", line) + # rename dirs to directories + line = re.sub(r"^(\s+)dirs", r"\1directories", line) + # fix import states for service + line = re.sub(r"^.+import.+CoreService.+$", + r"from core.services.coreservices import CoreService", line) + # fix method signatures + line = re.sub(r"def generateconfig\(cls, node, filename, services\)", + r"def generate_config(cls, node, filename)", line) + line = re.sub(r"def getvalidate\(cls, node, services\)", + r"def get_validate(cls, node)", line) + line = re.sub(r"def getstartup\(cls, node, services\)", + r"def get_startup(cls, node)", line) + line = re.sub(r"def getconfigfilenames\(cls, nodenum, services\)", + r"def get_configs(cls, node)", line) + # remove unwanted lines + if re.search(r"addservice\(", line): + continue + if re.search(r"from.+\.ipaddr|import ipaddr", line): + continue + if re.search(r"from.+\.ipaddress|import ipaddress", line): + continue + # add modified line to make updated copy + update.append(line) + service_file.close() + + with open(f"{service_file.name}.update", "w") as f: + f.writelines(update) + + +if __name__ == "__main__": + args = parse_args() + update_service(args.file)