corbz/PYRSS-Website
corbz/PYRSS-Website
1
0
Fork 0
Code Issues 21 Pull Requests Actions Releases 1 Wiki Activity

superuser checks over botuser

Browse Source
This commit is contained in:
Corban-Lee Jones 2024-06-26 11:13:23 +01:00
parent 4033ed6e7b
commit 450d40be89
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
apps/api/views.py
View File

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