1 files changed, 11 insertions, 0 deletions
diff --git a/archivist/cli.py b/archivist/cli.py
index 5f1d8ba..9299c94 100644
--- a/archivist/cli.py
+++ b/archivist/cli.py
@@ -5,6 +5,7 @@ from functools import reduce
 import operator as op
 
 from .virtual_prefixes import query as query_virtual, is_virtual, register_prefixes
+from .utils import ProxyCommand
 
 CONTEXT_SETTINGS = dict(help_option_names=['-h', '--help'])
 
@@ -19,6 +20,16 @@ def enable_debug():
 def cli(debug):
     if debug or 'DEBUG' in os.environ: enable_debug()
 
+
+class ServerCommand(ProxyCommand):
+    def _get_proxy(self):
+        from .server import server_group
+        return server_group
+
+@cli.group(cls=ServerCommand)
+def server():
+    """Flask server handling"""
+
 @cli.group()
 def db():
     """Database Management"""