work

1 files changed, 29 insertions, 0 deletions

diff --git a/pyenc/cmdline.py b/pyenc/cmdline.py
new file mode 100644
index 0000000..099018d
--- /dev/null
+++ b/pyenc/cmdline.py
@@ -0,0 +1,29 @@
+import click
+from flask.cli import AppGroup
+
+app_group = AppGroup('user', help="Testt")
+
+@app_group.command('enc')
+@click.argument('fqdn')
+def enc(fqdn):
+    from . import enc
+    enc.run_enc(fqdn)
+
+@app_group.command('init-db')
+def initialize_database():
+    from . import model
+    model.db.create_all()
+
+@app_group.command('enumerate-classes')
+def enumerate_classes():
+    from . import enumerate_classes
+    environment_name = 'production'
+    path_base = '/var/lib/machines/busting/etc/puppetlabs/code/environments/'
+    enumerate_classes.run(
+            path_base=path_base,
+            environment_name=environment_name)
+
+def init_app(app):
+    """Add command line options to current flask app."""
+    app.cli.add_command(app_group)
+