diff options
author | Hugo Hörnquist <hugo@lysator.liu.se> | 2022-08-08 04:41:11 +0200 |
---|---|---|
committer | Hugo Hörnquist <hugo@lysator.liu.se> | 2022-08-08 04:41:11 +0200 |
commit | ba1516c255f7167b696ca7f4c6db2846cf6bb046 (patch) | |
tree | c1794abe263034300efce07bb629cb10edc5ddd5 /pyenc/app/cmdline.py | |
parent | work (diff) | |
download | puppet-classifier-ba1516c255f7167b696ca7f4c6db2846cf6bb046.tar.gz puppet-classifier-ba1516c255f7167b696ca7f4c6db2846cf6bb046.tar.xz |
work
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) |