Merge pull request #665 from axsaucedo/template_creation_init

Template creation with cookiecutter for base project initialisation

mlserver/cli/init_project.py

@@ -0,0 +1,14 @@
+import subprocess
+from ..logging import logger
+
+
+def init_cookiecutter_project(template: str):
+    rc = subprocess.call(["which", "cookiecutter"])
+    if rc == 0:
+        cmd = f"cookiecutter {template}"
+        subprocess.run(cmd, check=True, shell=True)
+    else:
+        logger.error(
+            "The cookiecutter command is not found. \n\n"
+            "Please install with 'pip install cookiecutter' and retry"
+        )

mlserver/cli/main.py

@@ -6,6 +6,8 @@
 
 from functools import wraps
 
+from .init_project import init_cookiecutter_project
+
 from ..server import MLServer
 from ..logging import logger, configure_logger
 from ..utils import install_uvloop_event_loop
@@ -58,6 +60,17 @@ async def build(folder: str, tag: str, no_cache: bool = False):
     logger.info(f"Successfully built custom Docker image with tag {tag}")
 
 
+@root.command("init")
+# TODO: Update to have template(s) in the SeldonIO org
+@click.option("-t", "--template", default="https://github.com/EthicalML/sml-security/")
+@click_async
+async def init_project(template: str):
+    """
+    Generate a base project template
+    """
+    init_cookiecutter_project(template)
+
+
 @root.command("dockerfile")
 @click.argument("folder", nargs=1)
 @click.option("-i", "--include-dockerignore", is_flag=True)