fix NotificationLevel being suppressed

MiloTheFox · May 4, 2020 · 5107583 · 5107583
1 parent 6decfd1
commit 5107583
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/discord/guild.py b/discord/guild.py
@@ -1050,12 +1050,10 @@ async def edit(self, *, reason=None, **fields):
         fields['banner'] = banner
         fields['splash'] = splash
 
-        try:
-            default_message_notifications = int(fields.pop('default_notifications'))
-        except (TypeError, KeyError):
-            pass
-        else:
-            fields['default_message_notifications'] = default_message_notifications
+        default_message_notifications = fields.get('default_notifications', self.default_notifications)
+        if not isinstance(default_message_notifications, NotificationLevel):
+            raise InvalidArgument('default_notifications field must be of type NotificationLevel')
+        fields['default_message_notifications'] = default_message_notifications.value
 
         try:
             afk_channel = fields.pop('afk_channel')