feat(#324): entrypoint

Author: h1alexbel

sr-filter/pyproject.toml

@@ -28,6 +28,9 @@ license = "MIT"
 readme = "README.md"
 packages = [{include = "sr", from = "src"}]
 
+[tool.poetry.scripts]
+sr-filter = "sr.sr:main"
+
 [tool.poetry.dependencies]
 sr-data = "0.0.0"
 python = "==3.10.* || ==3.11.* || ==3.12.*"

sr-filter/src/sr/sr.py

@@ -25,20 +25,37 @@
 
 import argparse
 import toml
+import json
 from loguru import logger
 
-parser = argparse.ArgumentParser(description="SR toolchain")
-parser.add_argument(
-    "--version",
-    action="version",
-    version=f"%(prog)s {toml.load("pyproject.toml")["tool"]["poetry"]["version"]}"
-)
-parser.add_argument(
-    "--steps",
-    type=str,
-    default="mine,pulls,filter,workflows,junit,package,cluster,stats",
-    help="SR steps to execute"
-)
 
-args = parser.parse_args()
-logger.info(f"Registering steps: {args.steps}")
+"""
+Register steps.
+"""
+def register(steps):
+    logger.info(f"Registering steps: {steps.replace(",", ", ")}")
+    with open("resources/toolchain.json", "r") as spec:
+        tlc = json.load(spec)
+        defined = tlc["goal"]
+    for step in steps.split(","):
+        if not step in defined:
+            logger.error(f"Step '{step}' cannot be recongnized by sr-filter");
+            exit(-1);
+    logger.info("Steps registered");
+
+
+def main():
+    parser = argparse.ArgumentParser(description="SR toolchain")
+    parser.add_argument(
+        "--version",
+        action="version",
+        version=f"%(prog)s {toml.load("pyproject.toml")["tool"]["poetry"]["version"]}"
+    )
+    parser.add_argument(
+        "--steps",
+        type=str,
+        default="mine,pulls,filter,workflows,junit,package,cluster,stats",
+        help="Comma separeted steps to execute"
+    )
+    args = parser.parse_args()
+    register(args.steps)