klmp200
/
Sith
mirror of https://github.com/ae-utbm/sith3.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Better fix

This commit is contained in:
Julien Constant 2023-05-12 11:42:24 +02:00
parent 41e68224ca
commit 46bc00d971
No known key found for this signature in database
GPG Key ID: 816E7C070117E5B7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
core/models.py
View File

@ -185,7 +185,7 @@ def get_group(*, pk: int = None, name: str = None) -> Optional[Group]:
if pk is None and name is None:
raise ValueError("Either pk or name must be set")
pk_or_name: Union[str, int] = pk if pk is not None else name
pk_or_name: Union[str, int] = pk if pk is not None else name.replace(" ", "_")
group = cache.get(f"sith_group_{pk_or_name}")
if group == "not_found":
@ -201,7 +201,7 @@ def get_group(*, pk: int = None, name: str = None) -> Optional[Group]:
group = Group.objects.filter(name=name).first()
if group is not None:
cache.set(f"sith_group_{group.id}", group)
cache.set(f"sith_group_{group.name}", group)
cache.set(f"sith_group_{group.name.replace(' ', '_')}", group)
else:
cache.set(f"sith_group_{pk_or_name}", "not_found")
return group