Skip to content

Commit 32e2181

Browse files
committed
fix typing
1 parent b9dfe39 commit 32e2181

File tree

1 file changed

+19
-15
lines changed

1 file changed

+19
-15
lines changed

discord/ext/pages/pagination.py

Lines changed: 19 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -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

187187
class 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

Comments
 (0)