masq
/
halfapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Replace dbupdate command with routes. 

Allows to use --update, --list, ...
This commit is contained in:
Joël Maïzi 2020-07-29 16:40:25 +02:00
parent e289f6ad6b
commit 527d5c2e93
1 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
halfapi/cli.py
View File

@ -57,18 +57,28 @@ def delete_domain(domain):
@click.option('--domain', default=None) @click.option('--domain', default=None)
@click.option('--update', default=False, is_flag=True)
@click.option('--list', default=False, is_flag=True)
@cli.command() @cli.command()
def dbupdate(domain): def routes(domain, list, update):
if domain is None: domains = DOMAINS if domain is None else [domain]
if update:
if not domain:
click.echo('No domain name given, will update all active domains') click.echo('No domain name given, will update all active domains')
for domain in DOMAINS: for domain in domains:
dbupdate_fct(domain) update_db(domain)
sys.exit(0) if list:
list_routes(domains)
return dbupdate_fct(domain)
def dbupdate_fct(domain=None): def list_routes(domains):
for domain in domains:
print(f'\nDomain {domain}')
routes = Acl(domain=domain)
for route in routes.select():
print('-', route)
def update_db(domain=None):
if domain is None: if domain is None:
click.echo('Missing domain', err=True) click.echo('Missing domain', err=True)
sys.exit(1) sys.exit(1)