diff options
Diffstat (limited to 'pyenc/cmdline.py')
-rw-r--r-- | pyenc/cmdline.py | 29 |
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) + |