Add templates as optional parameter to rest

CHANGELOG.md

@@ -9,7 +9,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## Changes Since Last Release
 
-#### Changed defaults / behaviours
+#### Changed defaults / behaviours 
 
 #### New Features & Functionality
 

superduper/cli/main.py

@@ -25,6 +25,7 @@ def start(
     host: str = 'localhost',
     headless: bool = False,
     data_backend: str | None = None,
+    templates: str | None = None,
 ):
     """Start the rest server and user interface.
 
@@ -39,7 +40,12 @@ def start(
 
     CFG.log_colorize = False
 
-    app = SuperDuperApp('rest', port=remote_port, data_backend=data_backend)
+    app = SuperDuperApp(
+        'rest',
+        port=remote_port,
+        data_backend=data_backend,
+        templates=templates.split(',') if templates else None,
+    )
 
     if host == 'localhost':
         # host frontend and server together

superduper/rest/base.py

@@ -75,6 +75,7 @@ def __init__(
         db: Datalayer = None,
         prefix: str = '',
         data_backend: str | None = None,
+        templates: t.List[str] | None = None,
     ):
         if prefix and not prefix.startswith('/'):
             prefix = f'/{prefix}'
@@ -106,6 +107,7 @@ def __init__(
         )
         self._db = db
         self.data_backend = data_backend
+        self.templates = templates
 
     @cached_property
     def app(self):
@@ -190,6 +192,19 @@ def start(self):
         self.print_routes()
         self.run()
 
+    def _add_templates(self, db):
+        if self.templates:
+            from superduper import templates
+
+            existing = db.show('template')
+            for t in self.templates:
+                if t in existing:
+                    logging.info(f'Found existing template: {t}')
+                    continue
+                logging.info(f'Applying template: {t}')
+                t = getattr(templates, t)
+                db.apply(t, force=True)
+
     def startup(
         self,
         cfg: t.Union[Config, None] = None,
@@ -212,6 +227,9 @@ def startup_db_client():
                     db = build_datalayer(cfg)
             else:
                 db = self._db
+
+            self._add_templates(db)
+
             self._app.state.pool = db
             self._app.state.pool.cluster.initialize()
             self._db = db