working cli dbupdate (poetry run halfapi dbupdate)
This commit is contained in:
parent
5cad163a39
commit
37104948ca
100
halfapi/cli.py
100
halfapi/cli.py
|
@ -97,7 +97,17 @@ def run(host, port, debug, dev, dbname, dbhost, dbport, dbuser, dbpassword):
|
||||||
log_level=log_level,
|
log_level=log_level,
|
||||||
reload=reload)
|
reload=reload)
|
||||||
|
|
||||||
def dropdb(dbname, host, port, user, password):
|
@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):
|
||||||
|
|
||||||
|
def dropdb():
|
||||||
if not click.confirm(f'Will now drop database {dbname}', default=True):
|
if not click.confirm(f'Will now drop database {dbname}', default=True):
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
@ -118,7 +128,7 @@ def dropdb(dbname, host, port, user, password):
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def delete_domain(domain):
|
def delete_domain():
|
||||||
d = Domain(name=domain)
|
d = Domain(name=domain)
|
||||||
if len(d) < 1:
|
if len(d) < 1:
|
||||||
return False
|
return False
|
||||||
|
@ -136,72 +146,82 @@ def delete_domain(domain):
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def add_acl_fct(version, domain_name, fct):
|
def add_acl_fct(fct):
|
||||||
acl = Acl()
|
acl = AclFunction()
|
||||||
acl.version = version
|
acl.version = version
|
||||||
acl.domain = domain_name
|
acl.domain = domain
|
||||||
acl.name = fct.__name__
|
acl.name = fct.__name__
|
||||||
|
if len(acl) == 0:
|
||||||
acl.insert()
|
acl.insert()
|
||||||
|
|
||||||
def add_route(version, domain_name, name, **kwargs):
|
def add_acl(name, **kwargs):
|
||||||
|
acl = Acl()
|
||||||
|
acl.version = version
|
||||||
|
acl.domain = domain
|
||||||
|
acl.name = name
|
||||||
|
acl.path = kwargs['path']
|
||||||
|
acl.http_verb = kwargs['verb']
|
||||||
|
for fct in kwargs['acl']:
|
||||||
|
acl.function = fct.__name__
|
||||||
|
|
||||||
|
if len(acl) == 0:
|
||||||
|
if fct is not None:
|
||||||
|
add_acl_fct(fct)
|
||||||
|
|
||||||
|
acl.insert()
|
||||||
|
|
||||||
|
elif fct is None:
|
||||||
|
acl.delete()
|
||||||
|
|
||||||
|
|
||||||
|
def add_route(name, **kwargs):
|
||||||
|
print(f'Adding route {version}/{domain}/{name}')
|
||||||
route = Route()
|
route = Route()
|
||||||
route.version = version
|
route.version = version
|
||||||
route.domain = domain_name
|
route.domain = domain
|
||||||
route.path = kwargs['path']
|
route.path = kwargs['path']
|
||||||
|
if len(route) == 0:
|
||||||
route.insert()
|
route.insert()
|
||||||
|
|
||||||
|
def add_routes_and_acl(routes):
|
||||||
def add_routes(version, domain_name, routes):
|
|
||||||
for name, route_params in routes.items():
|
for name, route_params in routes.items():
|
||||||
print(f'Adding route {version}/{domain_name}/{function_name}')
|
add_route(name, **route_params)
|
||||||
add_route(version, domain_name, name, **route_params)
|
add_acl(name, **route_params)
|
||||||
|
|
||||||
|
|
||||||
def add_domain(version, domain_name):
|
def add_domain():
|
||||||
domain = Domain(name=domain_name)
|
new_domain = Domain(name=domain)
|
||||||
domain.version = version
|
new_domain.version = version
|
||||||
if len(domain) == 0:
|
if len(new_domain) == 0:
|
||||||
print(f'New domain {domain_name}')
|
print(f'New domain {domain}')
|
||||||
domain.insert()
|
new_domain.insert()
|
||||||
|
|
||||||
|
|
||||||
@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:
|
if drop:
|
||||||
dropdb(dbname, host, port, user, password)
|
dropdb()
|
||||||
|
|
||||||
delete_domain(domain)
|
delete_domain()
|
||||||
|
|
||||||
acl_set = set()
|
acl_set = set()
|
||||||
add_route_acl = lambda routes: [
|
|
||||||
[ acl_set.add(acl) for acl in route['acl'] ]
|
|
||||||
for route in routes.keys() ]
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
|
||||||
|
# module retrieval
|
||||||
dom_mod = importlib.import_module(domain)
|
dom_mod = importlib.import_module(domain)
|
||||||
|
|
||||||
API_VERSION = dom_mod.API_VERSION
|
version = dom_mod.API_VERSION
|
||||||
add_domain(API_VERSION, domain)
|
add_domain()
|
||||||
|
|
||||||
|
# add main routes
|
||||||
ROUTES = dom_mod.ROUTES
|
ROUTES = dom_mod.ROUTES
|
||||||
add_route_acl(ROUTES)
|
add_routes_and_acl(dom_mod.ROUTES)
|
||||||
add_routes(API_VERSION, domain, dom_mod.ROUTES)
|
|
||||||
|
|
||||||
|
# add sub routers
|
||||||
ROUTERS = dom_mod.ROUTERS
|
ROUTERS = dom_mod.ROUTERS
|
||||||
|
|
||||||
for router_name in dom_mod.ROUTERS:
|
for router_name in dom_mod.ROUTERS:
|
||||||
router_mod = importlib.import_module(f'.routers.{router_name}', domain)
|
router_mod = importlib.import_module(f'.routers.{router_name}', domain)
|
||||||
add_route_acl(router_mod.ROUTES)
|
add_routes_and_acl(router_mod.ROUTES)
|
||||||
add_routes(API_VERSION, domain, router_mod.ROUTES)
|
|
||||||
|
|
||||||
for acl in acl_set:
|
|
||||||
add_acl_fct(API_VERSION, domain, acl)
|
|
||||||
|
|
||||||
except ImportError:
|
except ImportError:
|
||||||
click.echo(f'The domain {domain} has no *ROUTES* variable', err=True)
|
click.echo(f'The domain {domain} has no *ROUTES* variable', err=True)
|
||||||
|
|
Loading…
Reference in New Issue