diff options
Diffstat (limited to 'pydis_site/apps/api/viewsets/bot')
-rw-r--r-- | pydis_site/apps/api/viewsets/bot/__init__.py | 20 | ||||
-rw-r--r-- | pydis_site/apps/api/viewsets/bot/user.py | 2 |
2 files changed, 20 insertions, 2 deletions
diff --git a/pydis_site/apps/api/viewsets/bot/__init__.py b/pydis_site/apps/api/viewsets/bot/__init__.py index bb26cb11..8cac6866 100644 --- a/pydis_site/apps/api/viewsets/bot/__init__.py +++ b/pydis_site/apps/api/viewsets/bot/__init__.py @@ -1,4 +1,3 @@ -# flake8: noqa from .aoc_completionist_block import AocCompletionistBlockViewSet from .aoc_link import AocAccountLinkViewSet from .bot_setting import BotSettingViewSet @@ -14,3 +13,22 @@ from .offensive_message import OffensiveMessageViewSet from .reminder import ReminderViewSet from .role import RoleViewSet from .user import UserViewSet + +__all__ = ( + "AocAccountLinkViewSet", + "AocCompletionistBlockViewSet", + "BotSettingViewSet", + "BumpedThreadViewSet", + "DeletedMessageViewSet", + "DocumentationLinkViewSet", + "FilterListViewSet", + "FilterViewSet", + "InfractionViewSet", + "MailingListViewSet", + "NominationViewSet", + "OffTopicChannelNameViewSet", + "OffensiveMessageViewSet", + "ReminderViewSet", + "RoleViewSet", + "UserViewSet", +) diff --git a/pydis_site/apps/api/viewsets/bot/user.py b/pydis_site/apps/api/viewsets/bot/user.py index 79e867c3..ab0a1091 100644 --- a/pydis_site/apps/api/viewsets/bot/user.py +++ b/pydis_site/apps/api/viewsets/bot/user.py @@ -531,6 +531,6 @@ class UserViewSet(ModelViewSet): with Metricity() as metricity: data = metricity.total_messages_in_past_n_days(user_ids, days) - default_data = {user_id: 0 for user_id in user_ids} + default_data = dict.fromkeys(user_ids, 0) response_data = default_data | dict(data) return Response(response_data, status=status.HTTP_200_OK) |