canaille-globuzma/canaille/commands.py
2021-11-24 13:15:35 +01:00

65 lines
1.3 KiB
Python

import click
import sys
from canaille import create_app
from canaille.models import AuthorizationCode, Token
from flask import current_app
from flask.cli import with_appcontext, FlaskGroup
@click.group(cls=FlaskGroup, create_app=create_app)
def cli():
pass
@cli.command()
@with_appcontext
def clean():
"""
Remove expired tokens and authorization codes.
"""
from canaille import setup_ldap_connection, teardown_ldap_connection
setup_ldap_connection(current_app)
for t in Token.filter():
if t.is_expired():
t.delete()
for a in AuthorizationCode.filter():
if a.is_expired():
a.delete()
teardown_ldap_connection(current_app)
@cli.command()
@with_appcontext
def check():
"""
Check the configuration file.
"""
from canaille.configuration import validate, ConfigurationException
try:
validate(current_app.config, validate_remote=True)
except ConfigurationException as exc:
print(exc)
sys.exit(1)
@cli.command()
@with_appcontext
def install():
"""
Installs canaille elements from the configuration.
"""
from canaille.installation import install
from canaille.configuration import ConfigurationException
try:
install(current_app.config)
except ConfigurationException as exc:
print(exc)
sys.exit(1)