Skip to content
Snippets Groups Projects
Commit 9fb1c20d authored by Hangzhi Yu's avatar Hangzhi Yu
Browse files

Remove duplicated code

parent abee08ee
No related branches found
No related tags found
1 merge request!318Resolve "Add helper function to filter a queryset by a rule"
Pipeline #2894 failed
......@@ -358,14 +358,11 @@ def queryset_rules_filter(obj: Union[HttpRequest, Model], queryset: QuerySet, pe
"""Filter queryset by user and permission."""
wanted_objects = set()
if isinstance(obj, HttpRequest):
if hasattr(obj, "user"):
for item in queryset:
if obj.user.has_perm(perm, item):
wanted_objects.add(item.pk)
else:
for item in queryset:
if obj.has_perm(perm, item):
wanted_objects.add(item.pk)
if isinstance(obj, HttpRequest) and hasattr(obj, "user"):
obj = obj.user
for item in queryset:
if obj.has_perm(perm, item):
wanted_objects.add(item.pk)
return queryset.filter(pk__in = wanted_objects)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment