feat: script for psql test before migration

perun/proxy/utils/run_probes_jpmu_test.py

+#!/usr/bin/python3
+import re
+import subprocess
+import sys
+import time
+from datetime import datetime
+from threading import Thread
+
+import yaml
+
+
+def open_file(filepath):
+    try:
+        with open(filepath) as f:
+            return f.read()
+    except OSError as e:
+        print(
+            f"Cannot open config with path: {filepath}, error: {e.strerror}",
+            file=sys.stderr,
+        )
+        sys.exit(2)
+
+
+def get_metrics_and_new_output(output):
+    """
+    Parses metrics from output, metrics must be in one of (or combination of)
+    the following formats:
+        1) |metric1=val;val;;;|metric2=val (delimiter |)
+        2) |metric1=val;metric2=val2; (delimiter ;)
+        3) |metric1=val metric2=val2 (delimiter ' ')
+
+        Values must be int or float
+    """
+    metrics_pattern = r"(\s\|\s|\|)(\w+=[\d.;]+(;|\s|$))+"
+
+    match = re.search(metrics_pattern, output)
+    if match:
+        output = re.sub(metrics_pattern, " ", output)
+        metrics = re.sub(r"\s", "", match.group())
+        metrics = re.sub(r"^\|", "", metrics)
+        metrics = re.sub(r"(\d;?)([a-zA-Z])", r"\1|\2", metrics)
+        return metrics.strip(), output.strip()
+    return None, output
+
+
+def run_probe(probe_name, command, timeout, identifier):
+    """
+    Runs nagios monitoring probe and prints output in following formats:
+        1) return_code probe_name metrics output
+        2) return_code probe_name - output
+
+    metrics output format:
+        metric1=val;|metric2=val2|metric3=val3;val3;;;|metric4=val4
+    """
+    exec_time = datetime.now()
+    try:
+        result = subprocess.run(
+            command,
+            text=True,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.STDOUT,
+            timeout=timeout,
+        )
+    except subprocess.TimeoutExpired:
+        print(f"exectime: {exec_time} - 3 {probe_name} - probe TIMED OUT after {timeout}s")
+        return 3
+    output = re.sub("[ \t\n]+", " ", result.stdout)
+    search = re.search(r" - .*", output)
+    if search:
+        output = re.sub(r"^ - ", "", search.group())
+    metrics, new_output = get_metrics_and_new_output(output)
+    if metrics:
+        print(f"exectime: {exec_time} - {result.returncode} {probe_name} {metrics} {new_output}")
+    else:
+        print(f"exectime: {exec_time} - {result.returncode} {probe_name} - {output}")
+    return result.returncode
+
+
+def main():
+    config_filepath = "/etc/run_probes_cfg.yaml"
+    config = yaml.safe_load(open_file(config_filepath))
+    if not config:
+        return
+
+    global_timeout = config["default_timeout"]
+    for _, options in config["checks"].items():
+        module = options["module"]
+        for i in range(30):
+            for name, args in options.get("runs").items():
+                command = ["python3", "-m", module]
+                timeout = global_timeout
+                if args is not None:
+                    for arg_name, arg_val in args.items():
+                        if arg_name == "timeout":
+                            timeout = arg_val
+                            continue
+                        if len(arg_name) == 1:
+                            arg_name = "-" + arg_name
+                        else:
+                            arg_name = "--" + arg_name
+                        if arg_val is True:
+                            arg_val = "true"
+                        elif arg_val is False:
+                            arg_val = "false"
+                        command.append(arg_name)
+                        if arg_val is not None:
+                            command.append(str(arg_val))
+                Thread(target=run_probe, args=[name, command, timeout, i]).start()
+            time.sleep(5)
+
+
+if __name__ == "__main__":
+    main()

setup.py

@@ -38,6 +38,7 @@ setuptools.setup(
     entry_points={
         "console_scripts": [
             "run_probes=perun.proxy.utils.run_probes:main",
+            "run_probes_jpmu_test=perun.proxy.utils.run_probes_jpmu_test:main",
             "check_custom_command=perun.proxy.utils.nagios.check_custom_command:main",
             "check_dockers=perun.proxy.utils.nagios.check_dockers:main",
             "check_exabgp_propagation="