ajout des fonctions de DB dans le cli

halfapi/cli.py

@@ -1,8 +1,22 @@
 #!/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}} 
@@ -43,6 +57,77 @@ def run(host, port, debug, dev):
         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()