ajout des fonctions de DB dans le cli
This commit is contained in:
parent
b532aac634
commit
d1e3198441
|
@ -1,8 +1,22 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
# builtins
|
||||||
import click
|
import click
|
||||||
import uvicorn
|
import uvicorn
|
||||||
import os
|
import os
|
||||||
import sys
|
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={
|
CONTEXT_SETTINGS={
|
||||||
'default_map':{'run': {'port': 8000}}
|
'default_map':{'run': {'port': 8000}}
|
||||||
|
@ -43,6 +57,77 @@ def run(host, port, debug, dev):
|
||||||
log_level=log_level,
|
log_level=log_level,
|
||||||
reload=reload)
|
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__':
|
if __name__ == '__main__':
|
||||||
cli()
|
cli()
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue