habsi/authentik
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

*: ensure unittests wait on tasks

Browse Source
This commit is contained in:
Jens Langhammer
2020-10-16 14:53:14 +02:00
parent 7154f19668
commit 3b9524cdfc
5 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
passbook/policies/reputation/tasks.py
View File

@ -18,7 +18,7 @@ LOGGER = get_logger()
def save_ip_reputation(self: MonitoredTask):
"""Save currently cached reputation to database"""
objects_to_update = []
for key, score in cache.get_many(CACHE_KEY_IP_PREFIX + "*").items():
for key, score in cache.get_many(cache.keys(CACHE_KEY_IP_PREFIX + "*")).items():
remote_ip = key.replace(CACHE_KEY_IP_PREFIX, "")
rep, _ = IPReputation.objects.get_or_create(ip=remote_ip)
rep.score = score
@ -33,7 +33,7 @@ def save_ip_reputation(self: MonitoredTask):
def save_user_reputation(self: MonitoredTask):
"""Save currently cached reputation to database"""
objects_to_update = []
for key, score in cache.get_many(CACHE_KEY_USER_PREFIX + "*").items():
for key, score in cache.get_many(cache.keys(CACHE_KEY_USER_PREFIX + "*")).items():
username = key.replace(CACHE_KEY_USER_PREFIX, "")
users = User.objects.filter(username=username)
if not users.exists():