Parse the method from UI before passing it to external process

To reduce the chances of the system failing due to the external process being
launched with the wrong parameters, add a parsing stage that converts the
method from the UI into a form acceptable by the CLI script.

* gn3/commands.py: parse the method from UI
* scripts/partial_correlations.py: simplify the acceptable methods

1 files changed, 9 insertions, 1 deletions

diff --git a/gn3/commands.py b/gn3/commands.py
index 0a74b7d..0b8db5b 100644
--- a/gn3/commands.py
+++ b/gn3/commands.py
@@ -67,9 +67,17 @@ def compose_pcorrs_command(
         primary_trait: str, control_traits: Tuple[str, ...], method: str,
         **kwargs):
     """Compose the command to run partias correlations"""
+    def __parse_method__(method):
+        mthd = method.lower().replace("'", "")
+        if "pearsons" in mthd:
+            return "pearsons"
+        if "spearmans" in mthd:
+            return "spearmans"
+        raise Exception(f"Invalid method '{method}'")
+
     prefix_cmd = (
         f"{sys.executable}", "-m", "scripts.partial_correlations",
-        primary_trait, ",".join(control_traits), method)
+        primary_trait, ",".join(control_traits), __parse_method__(method))
     if (
             kwargs.get("target_database") is not None
             and kwargs.get("target_traits") is None):