superuser checks over botuser

This commit is contained in:
Corban-Lee Jones 2024-06-26 11:13:23 +01:00
parent 4033ed6e7b
commit 450d40be89

View File

@ -36,9 +36,6 @@ class DefaultPagination(PageNumberPagination):
def is_automated_admin(user):
print("is automated admin", user.user_type == DiscordUser.USER_TYPES.AUTOMATED_USER and user.is_superuser)
print(user.user_type == DiscordUser.USER_TYPES.AUTOMATED_USER)
print(user.is_superuser)
return user.user_type == DiscordUser.USER_TYPES.AUTOMATED_USER and user.is_superuser
@ -117,7 +114,7 @@ class Filter_ListView(generics.ListCreateAPIView):
search_fields = ["name", "keywords", "regex"]
def get_queryset(self):
if is_automated_admin(self.request.user):
if self.request.user.is_superuser:
return Filter.objects.all().order_by("id")
saved_guild_ids = SavedGuilds.objects \
@ -159,7 +156,7 @@ class Filter_DetailView(generics.RetrieveUpdateDestroyAPIView):
serializer_class = FilterSerializer
def get_queryset(self):
if is_automated_admin(self.request.user):
if self.request.user.is_superuser:
return Filter.objects.all().order_by("id")
saved_guild_ids = SavedGuilds.objects \
@ -207,7 +204,7 @@ class Subscription_ListView(generics.ListCreateAPIView):
return self.read_serializer_class
def get_queryset(self):
if is_automated_admin(self.request.user):
if self.request.user.is_superuser:
return Subscription.objects.all().order_by("-creation_datetime")
saved_guild_ids = SavedGuilds.objects \
@ -252,7 +249,7 @@ class Subscription_DetailView(generics.RetrieveUpdateDestroyAPIView):
serializer_class = SubscriptionSerializer_POST
def get_queryset(self):
if is_automated_admin(self.request.user):
if self.request.user.is_superuser:
return Subscription.objects.all().order_by("-creation_datetime")
saved_guild_ids = SavedGuilds.objects \
@ -275,7 +272,7 @@ class Subscription_SubChannelView(generics.DestroyAPIView):
def delete(self, *args, **kwargs):
if is_automated_admin(self.request.user):
if self.request.user.is_superuser:
subscriptions = Subscription.objects.all()
else:
saved_guild_ids = SavedGuilds.objects \
@ -326,7 +323,7 @@ class SavedGuild_ListView(generics.ListCreateAPIView):
search_fields = ["name"]
def get_queryset(self):
if is_automated_admin(self.request.user):
if self.request.user.is_superuser:
return SavedGuilds.objects.all()
return SavedGuilds.objects.filter(added_by=self.request.user)
@ -382,7 +379,7 @@ class SavedGuild_DetailView(generics.RetrieveDestroyAPIView):
serializer_class = SavedGuildSerializer
def get_queryset(self):
if is_automated_admin(self.request.user):
if self.request.user.is_superuser:
return SavedGuilds.objects.all()
return SavedGuilds.objects.filter(added_by=self.request.user)