134 lines
2.9 KiB
Python
Executable File
134 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# builtins
|
|
import click
|
|
import uvicorn
|
|
import os
|
|
import sys
|
|
import importlib
|
|
|
|
# database
|
|
import psycopg2
|
|
|
|
# hop-generated classes
|
|
from .models.api.version import Version
|
|
from .models.api.domain import Domain
|
|
from .models.api.route import Route
|
|
from .models.api.acl_function import AclFunction
|
|
from .models.api.acl import Acl
|
|
|
|
|
|
CONTEXT_SETTINGS={
|
|
'default_map':{'run': {'port': 8000}}
|
|
}
|
|
|
|
@click.group(invoke_without_command=True, context_settings=CONTEXT_SETTINGS)
|
|
@click.pass_context
|
|
def cli(ctx):
|
|
if ctx.invoked_subcommand is None:
|
|
return run()
|
|
|
|
@click.option('--host', default='127.0.0.1')
|
|
@click.option('--port', default='8000')
|
|
@click.option('--debug', default=False)
|
|
@click.option('--dev', default=True)
|
|
@cli.command()
|
|
def run(host, port, debug, dev):
|
|
if dev:
|
|
debug = True
|
|
reload = True
|
|
log_level = 'debug'
|
|
else:
|
|
reload = False
|
|
log_level = 'info'
|
|
|
|
click.echo('Launching application with default parameters')
|
|
click.echo(f'''Parameters : \n
|
|
Host : {host}
|
|
Port : {port}
|
|
Debug : {debug}
|
|
Dev : {dev}''')
|
|
|
|
sys.path.insert(0, os.getcwd())
|
|
click.echo(sys.path)
|
|
uvicorn.run('halfapi.app:app',
|
|
host=host,
|
|
port=int(port),
|
|
log_level=log_level,
|
|
reload=reload)
|
|
|
|
def dropdb(dbname, host, port, user, password):
|
|
if not click.confirm(f'Will now drop database {dbname}', default=True):
|
|
return False
|
|
|
|
conn = psycopg2.connect({
|
|
'dbname': dbname,
|
|
'host': host,
|
|
'port': port,
|
|
'user': user,
|
|
'password': password
|
|
})
|
|
|
|
cur = conn.cursor()
|
|
|
|
cur.execute(f'DROP DATABASE {dbname};')
|
|
conn.commit()
|
|
cur.close()
|
|
conn.close()
|
|
|
|
return True
|
|
|
|
@click.option('--domain', default='organigramme')
|
|
@cli.command()
|
|
def delete_domain(domain):
|
|
d = Domain(name=domain)
|
|
if len(d) < 1:
|
|
return False
|
|
|
|
acl = Acl(domain=domain)
|
|
acl.delete()
|
|
|
|
fct = AclFunction(domain=domain)
|
|
fct.delete()
|
|
|
|
route = Route(domain=domain)
|
|
route.delete()
|
|
|
|
d.delete()
|
|
|
|
return True
|
|
|
|
@click.option('--dbname', default='api')
|
|
@click.option('--host', default='127.0.0.1')
|
|
@click.option('--port', default=5432)
|
|
@click.option('--user', default='api')
|
|
@click.option('--password', default='')
|
|
@click.option('--domain', default='organigramme')
|
|
@click.option('--drop', is_flag=True, default=False)
|
|
@cli.command()
|
|
def dbupdate(dbname, host, port, user, password, domain, drop):
|
|
if drop:
|
|
dropdp(dbname, host, port, user, password)
|
|
|
|
delete_domain(domain)
|
|
try:
|
|
ROUTES = importlib.import_module('ROUTES', domain)
|
|
acl_set = set()
|
|
add_acl_set = lambda x: acl_set.add(i)
|
|
[
|
|
map(add_acl_set, ROUTES[route]['acl'])
|
|
for route in ROUTES.keys()
|
|
]
|
|
print(acl_set)
|
|
|
|
except ImportError:
|
|
click.echo(f'The domain {domain} has no *ROUTES* variable', err=True)
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
cli()
|
|
|