aboutsummaryrefslogtreecommitdiff
path: root/pyenc/cmdline.py
blob: 099018dafc6eb76af9ef79a920f102686ede3771 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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)