bc9e7e8b93
* build(deps): bump structlog from 20.1.0 to 20.2.0 Bumps [structlog](https://github.com/hynek/structlog) from 20.1.0 to 20.2.0. - [Release notes](https://github.com/hynek/structlog/releases) - [Changelog](https://github.com/hynek/structlog/blob/master/CHANGELOG.rst) - [Commits](https://github.com/hynek/structlog/compare/20.1.0...20.2.0) Signed-off-by: dependabot[bot] <support@github.com> * *: use structlog.stdlib instead of structlog for type-hints Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Jens Langhammer <jens.langhammer@beryju.org>
38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
"""authentik flow signals"""
|
|
from django.core.cache import cache
|
|
from django.db.models.signals import post_save
|
|
from django.dispatch import receiver
|
|
from structlog.stdlib import get_logger
|
|
|
|
LOGGER = get_logger()
|
|
|
|
|
|
def delete_cache_prefix(prefix: str) -> int:
|
|
"""Delete keys prefixed with `prefix` and return count of deleted keys."""
|
|
keys = cache.keys(prefix)
|
|
cache.delete_many(keys)
|
|
return len(keys)
|
|
|
|
|
|
@receiver(post_save)
|
|
# pylint: disable=unused-argument
|
|
def invalidate_flow_cache(sender, instance, **_):
|
|
"""Invalidate flow cache when flow is updated"""
|
|
from authentik.flows.models import Flow, FlowStageBinding, Stage
|
|
from authentik.flows.planner import cache_key
|
|
|
|
if isinstance(instance, Flow):
|
|
total = delete_cache_prefix(f"{cache_key(instance)}*")
|
|
LOGGER.debug("Invalidating Flow cache", flow=instance, len=total)
|
|
if isinstance(instance, FlowStageBinding):
|
|
total = delete_cache_prefix(f"{cache_key(instance.target)}*")
|
|
LOGGER.debug(
|
|
"Invalidating Flow cache from FlowStageBinding", binding=instance, len=total
|
|
)
|
|
if isinstance(instance, Stage):
|
|
total = 0
|
|
for binding in FlowStageBinding.objects.filter(stage=instance):
|
|
prefix = cache_key(binding.target)
|
|
total += delete_cache_prefix(f"{prefix}*")
|
|
LOGGER.debug("Invalidating Flow cache from Stage", stage=instance, len=total)
|