canaille-globuzma/canaille/oidc/commands.py
2023-04-09 21:58:11 +02:00

25 lines
549 B
Python

import click
from canaille.app.commands import with_backendcontext
from canaille.oidc.models import AuthorizationCode
from canaille.oidc.models import Token
from flask.cli import with_appcontext
@click.command()
@with_appcontext
@with_backendcontext
def clean():
"""
Remove expired tokens and authorization codes.
"""
for t in Token.query():
if t.is_expired():
t.delete()
for a in AuthorizationCode.query():
if a.is_expired():
a.delete()
def register(cli):
cli.add_command(clean)