diff options
Diffstat (limited to '')
-rw-r--r-- | pyenc/app/cmdline.py | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/pyenc/app/cmdline.py b/pyenc/app/cmdline.py new file mode 100644 index 0000000..4e7e33f --- /dev/null +++ b/pyenc/app/cmdline.py @@ -0,0 +1,37 @@ +import click +from flask.cli import AppGroup + +app_group = AppGroup('user', help="Testt") + + +@app_group.command('enc') +@click.argument('fqdn') +def enc(fqdn): + from pyenc import enc + enc.run_enc(fqdn) + + +@app_group.command('init-db') +def initialize_database(): + from pyenc.app import model + model.db.create_all() + # model.db.session.add(model.Misc(key='db-version', value='0.1')) + # model.db.session.add(model.Misc(key='default-table', value='production')) + # model.db.session.commit() + + +@app_group.command('enumerate-classes') +@click.argument('environment') +def enumerate_classes(environment): + from pyenc import enumerate_classes as enumerate_classes_ + environment_name = environment + # TODO path should be an argument + 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) |