155 lines
3.6 KiB
Python
Executable File
155 lines
3.6 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
|
||
|
||
# module libraries
|
||
from halfapi.app import check_conf
|
||
|
||
HALFORM_DSN=''
|
||
HALFORM_SECRET=''
|
||
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}''')
|
||
|
||
HALFORM_DSN=os.environ.get('HALFORM_DSN', '')
|
||
db_params = dsntodict(HALFORM_DSN)
|
||
if not hasattr(db_params, 'dbname'):
|
||
db_params['dbname'] = dbname
|
||
if not hasattr(db_params, 'host'):
|
||
db_params['host'] = dbhost
|
||
if not hasattr(db_params, 'port'):
|
||
db_params['port'] = dbport
|
||
if not hasattr(db_params, 'user'):
|
||
db_params['user'] = dbuser
|
||
if not hasattr(db_params, 'password'):
|
||
db_params['password'] = dbpassword
|
||
|
||
os.environ['HALFORM_DSN'] = dicttodsn(db_params)
|
||
|
||
check_conf()
|
||
|
||
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()
|
||
|