Typehint WelcomeScreen

Author: izxxr

discord/guild.py

@@ -2969,6 +2969,22 @@ async def welcome_screen(self):
         """
         data = await self._state.http.get_welcome_screen(self.id)
         return WelcomeScreen(data=data, guild=self)
+
+
+    @overload
+    async def edit_welcome_screen(
+        self,
+        *,
+        description: Optional[str] = ...,
+        welcome_channels: Optional[List[WelcomeChannel]] = ...,
+        enabled: Optional[bool] = ...,
+    ) -> WelcomeScreen:
+        ...
+
+    @overload
+    async def edit_welcome_screen(self) -> None:
+        ...        
+    
 
     async def edit_welcome_screen(self, **options):
         """|coro|
@@ -3012,7 +3028,7 @@ async def edit_welcome_screen(self, **options):
 
         for channel in welcome_channels:
             if not isinstance(channel, WelcomeScreenChannel):
-                raise InvalidArgument('welcome_channels parameter must be a list of WelcomeScreenChannel.')
+                raise TypeError('welcome_channels parameter must be a list of WelcomeScreenChannel.')
 
             welcome_channels_data.append(channel.to_dict())
 

discord/welcome_screen.py

@@ -140,6 +140,21 @@ def enabled(self) -> bool:
     def guild(self) -> Guild:
         """:class:`Guild`: The guild this welcome screen belongs to."""
         return self._guild
+    
+    
+    @overload
+    async def edit(
+        self,
+        *,
+        description: Optional[str] = ...,
+        welcome_channels: Optional[List[WelcomeChannel]] = ...,
+        enabled: Optional[bool] = ...,
+    ) -> None:
+        ...
+
+    @overload
+    async def edit(self) -> None:
+        ...
 
     async def edit(self, **options):
         """|coro|
@@ -184,19 +199,14 @@ async def edit(self, **options):
         NotFound
             This welcome screen does not exist.
         
-        Returns
-        --------
-        
-        :class:`WelcomeScreen`
-            The updated welcome screen.
         """
 
         welcome_channels = options.get('welcome_channels', [])
         welcome_channels_data = []
 
         for channel in welcome_channels:
             if not isinstance(channel, WelcomeScreenChannel):
-                raise InvalidArgument('welcome_channels parameter must be a list of WelcomeScreenChannel.')
+                raise TypeError('welcome_channels parameter must be a list of WelcomeScreenChannel.')
 
             welcome_channels_data.append(channel.to_dict())