Commit 7387b018 authored by okhin's avatar okhin 🚴

Update admin.py

parent 75b91752
Pipeline #1242 passed with stage
in 52 seconds
......@@ -33,10 +33,11 @@ class CampaignAdmin(TranslationAdmin):
def formfield_for_foreignkey(self, db_field, request, **kwargs):
if db_field.name == "organization":
if request.user.is_superuser:
kwargs["queryset"] = Organization.objects.all()
orgs = Organization.objects.all()
else:
orgs = request.user.organizations.all()
query = {'id__in': [x.id
for x in request.user.organizations.all()]}
for x in orgs]}
kwargs["queryset"] = Organization.objects.filter(**query)
return super(CampaignAdmin, self).formfield_for_foreignkey(db_field,
request,
......@@ -69,6 +70,9 @@ class ArgumentaryAdmin(TranslationAdmin):
def formfield_for_foreignkey(self, db_field, request, **kwargs):
if db_field.name == "campaign":
if request.user.is_superuser:
orgs = Organization.objects.all()
else:
orgs = request.user.organizations.all()
query = {'organization__in': orgs}
kwargs["queryset"] = Campaign.objects.filter(**query)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment