Refactor into executable module/package

Dockerfile

@@ -15,10 +15,10 @@ RUN python -m poetry install --without dev --no-root \
  && python -m poetry export -f requirements.txt --output requirements.txt --without-hashes
 
 COPY ${package}/ /app/${package}
-COPY tests /app/${package}/tests
+COPY tests /app/tests
 
 ENV PYTHONPATH /app/${package}
-WORKDIR /app/${package}
+WORKDIR /app/
 
 # Run tests and return coverage analysis
 RUN python -m coverage run tests/test_${package}.py \
@@ -38,9 +38,9 @@ COPY ${package}/ /app/${package}
 # Install all plugin dependencies from the generated requirements.txt file
 RUN python -m pip install -r requirements.txt
 
-WORKDIR /app/${package}
+WORKDIR /app
 
-ENTRYPOINT ["python", "template_python_plugin.py"]
+ENTRYPOINT ["python", "-m", "arcaflow_plugin_template_python"]
 CMD []
 
 LABEL org.opencontainers.image.source="https://github.com/arcalot/arcaflow-plugin-template-python"

arcaflow_plugin_template_python/__main__.py

@@ -0,0 +1,4 @@
+from arcaflow_plugin_template_python import cli
+
+
+cli.main()

arcaflow_plugin_template_python/cli.py

@@ -0,0 +1,17 @@
+import sys
+from arcaflow_plugin_sdk import plugin as plugin_sdk
+import arcaflow_plugin_template_python.template_python_plugin as template_plugin
+
+
+def main():
+    sys.exit(
+        plugin_sdk.run(
+            plugin_sdk.build_schema(
+                # List your step functions here:
+                template_plugin.hello_world,
+            )
+        )
+    )
+
+if __name__ == "__main__":    
+    main()

arcaflow_plugin_template_python/template_python_plugin.py

@@ -3,7 +3,7 @@
 import sys
 import typing
 from arcaflow_plugin_sdk import plugin
-from template_python_schema import (
+from arcaflow_plugin_template_python.template_python_schema import (
     InputParams,
     SuccessOutput,
     ErrorOutput,

pyproject.toml

@@ -5,10 +5,6 @@ description = ""
 authors = ["Arcalot"]
 license = "Apache-2.0+GPL-2.0-only"
 
-packages = [
-   { include="template_python_plugin.py", from="./arcaflow_plugin_template_python"  },
-]
-
 [tool.poetry.dependencies]
 python = "^3.9"
 arcaflow-plugin-sdk = "^0.13.0"

tests/test_arcaflow_plugin_template_python.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 import unittest
-import template_python_plugin
+import arcaflow_plugin_template_python.template_python_plugin as template_python_plugin
 from arcaflow_plugin_sdk import plugin