@@ -80,7 +80,7 @@ def __init__(
80
80
)
81
81
self .button_type = button_type
82
82
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
84
84
self .style = style
85
85
self .disabled = disabled
86
86
self .loop_label = self .label if not loop_label else loop_label
@@ -170,18 +170,18 @@ def __init__(
170
170
):
171
171
self .label = label
172
172
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
175
175
self .show_disabled = show_disabled
176
176
self .show_indicator = show_indicator
177
177
self .author_check = author_check
178
178
self .disable_on_timeout = disable_on_timeout
179
179
self .use_default_buttons = use_default_buttons
180
180
self .default_button_row = default_button_row
181
181
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
185
185
186
186
187
187
class Paginator (discord .ui .View ):
@@ -252,27 +252,29 @@ def __init__(
252
252
custom_buttons : Optional [List [PaginatorButton ]] = None ,
253
253
) -> None :
254
254
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
257
259
self .current_page = 0
258
260
self .menu : Optional [PaginatorMenu ] = None
259
261
self .show_menu = show_menu
260
262
self .page_groups : Optional [List [PageGroup ]] = None
261
263
262
264
if all (isinstance (pg , PageGroup ) for pg in pages ):
263
265
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
265
267
266
268
self .page_count = len (self .pages ) - 1
267
269
self .buttons = {}
268
- self .custom_buttons = custom_buttons
270
+ self .custom_buttons : List = custom_buttons
269
271
self .show_disabled = show_disabled
270
272
self .show_indicator = show_indicator
271
273
self .disable_on_timeout = disable_on_timeout
272
274
self .use_default_buttons = use_default_buttons
273
275
self .default_button_row = default_button_row
274
276
self .loop_pages = loop_pages
275
- self .custom_view = custom_view
277
+ self .custom_view : discord . ui . View = custom_view
276
278
self .message : Union [discord .Message , discord .WebhookMessage , None ] = None
277
279
278
280
if self .custom_buttons and not self .use_default_buttons :
@@ -333,7 +335,9 @@ async def update(
333
335
"""
334
336
335
337
# 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
337
341
self .page_count = len (self .pages ) - 1
338
342
self .current_page = 0
339
343
# Apply config changes, if specified
@@ -360,8 +364,8 @@ async def update(
360
364
else self .default_button_row
361
365
)
362
366
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
365
369
if custom_buttons and not self .use_default_buttons :
366
370
self .buttons = {}
367
371
for button in custom_buttons :
@@ -395,7 +399,7 @@ async def disable(
395
399
"""
396
400
page = self .get_page_content (page )
397
401
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 :
399
403
item .disabled = True
400
404
if page :
401
405
await self .message .edit (
0 commit comments