@@ -80,7 +80,7 @@ def __init__(
8080 )
8181 self .button_type = button_type
8282 self .label = label if label or emoji else button_type .capitalize ()
83- self .emoji = emoji
83+ self .emoji : Union [ str , discord . Emoji , discord . PartialEmoji ] = emoji
8484 self .style = style
8585 self .disabled = disabled
8686 self .loop_label = self .label if not loop_label else loop_label
@@ -170,18 +170,18 @@ def __init__(
170170 ):
171171 self .label = label
172172 self .description = description
173- self .emoji = emoji
174- self .pages = pages
173+ self .emoji : Union [ str , discord . Emoji , discord . PartialEmoji ] = emoji
174+ self .pages : Union [ List [ str ], List [ Union [ List [ discord . Embed ], discord . Embed ]]] = pages
175175 self .show_disabled = show_disabled
176176 self .show_indicator = show_indicator
177177 self .author_check = author_check
178178 self .disable_on_timeout = disable_on_timeout
179179 self .use_default_buttons = use_default_buttons
180180 self .default_button_row = default_button_row
181181 self .loop_pages = loop_pages
182- self .custom_view = custom_view
183- self .timeout = timeout
184- self .custom_buttons = custom_buttons
182+ self .custom_view : discord . ui . View = custom_view
183+ self .timeout : float = timeout
184+ self .custom_buttons : List = custom_buttons
185185
186186
187187class Paginator (discord .ui .View ):
@@ -252,27 +252,29 @@ def __init__(
252252 custom_buttons : Optional [List [PaginatorButton ]] = None ,
253253 ) -> None :
254254 super ().__init__ (timeout = timeout )
255- self .timeout = timeout
256- self .pages = pages
255+ self .timeout : float = timeout
256+ self .pages : Union [
257+ List [PageGroup ], List [str ], List [Union [List [discord .Embed ], discord .Embed ]]
258+ ] = pages
257259 self .current_page = 0
258260 self .menu : Optional [PaginatorMenu ] = None
259261 self .show_menu = show_menu
260262 self .page_groups : Optional [List [PageGroup ]] = None
261263
262264 if all (isinstance (pg , PageGroup ) for pg in pages ):
263265 self .page_groups = self .pages if show_menu else None
264- self .pages = self .page_groups [0 ].pages
266+ self .pages : Union [ List [ str ], List [ Union [ List [ discord . Embed ], discord . Embed ]]] = self .page_groups [0 ].pages
265267
266268 self .page_count = len (self .pages ) - 1
267269 self .buttons = {}
268- self .custom_buttons = custom_buttons
270+ self .custom_buttons : List = custom_buttons
269271 self .show_disabled = show_disabled
270272 self .show_indicator = show_indicator
271273 self .disable_on_timeout = disable_on_timeout
272274 self .use_default_buttons = use_default_buttons
273275 self .default_button_row = default_button_row
274276 self .loop_pages = loop_pages
275- self .custom_view = custom_view
277+ self .custom_view : discord . ui . View = custom_view
276278 self .message : Union [discord .Message , discord .WebhookMessage , None ] = None
277279
278280 if self .custom_buttons and not self .use_default_buttons :
@@ -333,7 +335,9 @@ async def update(
333335 """
334336
335337 # Update pages and reset current_page to 0 (default)
336- self .pages = pages if pages is not None else self .pages
338+ self .pages : Union [
339+ List [PageGroup ], List [str ], List [Union [List [discord .Embed ], discord .Embed ]]
340+ ] = pages if pages is not None else self .pages
337341 self .page_count = len (self .pages ) - 1
338342 self .current_page = 0
339343 # Apply config changes, if specified
@@ -360,8 +364,8 @@ async def update(
360364 else self .default_button_row
361365 )
362366 self .loop_pages = loop_pages if loop_pages is not None else self .loop_pages
363- self .custom_view = None if custom_view is None else custom_view
364- self .timeout = timeout if timeout is not None else self .timeout
367+ self .custom_view : discord . ui . View = None if custom_view is None else custom_view
368+ self .timeout : float = timeout if timeout is not None else self .timeout
365369 if custom_buttons and not self .use_default_buttons :
366370 self .buttons = {}
367371 for button in custom_buttons :
@@ -395,7 +399,7 @@ async def disable(
395399 """
396400 page = self .get_page_content (page )
397401 for item in self .children :
398- if item not in self .custom_view_items or include_custom :
402+ if item not in self .custom_view . children or include_custom :
399403 item .disabled = True
400404 if page :
401405 await self .message .edit (
0 commit comments