LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-07-27 08:19:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix type annotation for button type

Browse Source
This commit is contained in:
Jahongir Qurbonov 2024-08-21 21:58:20 +05:00
parent babeba46d3
commit e131421d8e
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
client/src/telethon/_impl/client/types/keyboard.py
View File

@ -1,11 +1,13 @@
from typing import Optional, TypeAlias from typing import Generic, Optional, TypeAlias, TypeVar
from ...tl import abcs, types from ...tl import abcs, types
from .buttons.button import Button from .buttons.button import Button
AnyButton = TypeVar("AnyButton", bound=Button)
def _build_keyboard_rows( def _build_keyboard_rows(
btns: list[Button] | list[list[Button]], btns: list[AnyButton] | list[list[AnyButton]],
) -> list[abcs.KeyboardButtonRow]: ) -> list[abcs.KeyboardButtonRow]:
# list[button] -> list[list[button]] # list[button] -> list[list[button]]
# This does allow for "invalid" inputs (mixing lists and non-lists), but that's acceptable. # This does allow for "invalid" inputs (mixing lists and non-lists), but that's acceptable.
@ -21,12 +23,12 @@ def _build_keyboard_rows(
] ]
class Keyboard: class Keyboard(Generic[AnyButton]):
__slots__ = ("_raw",) __slots__ = ("_raw",)
def __init__( def __init__(
self, self,
buttons: list[Button] | list[list[Button]], buttons: list[AnyButton] | list[list[AnyButton]],
resize: bool, resize: bool,
single_use: bool, single_use: bool,
selective: bool, selective: bool,
@ -43,10 +45,10 @@ class Keyboard:
) )
class InlineKeyboard: class InlineKeyboard(Generic[AnyButton]):
__slots__ = ("_raw",) __slots__ = ("_raw",)
def __init__(self, buttons: list[Button] | list[list[Button]]) -> None: def __init__(self, buttons: list[AnyButton] | list[list[AnyButton]]) -> None:
self._raw = types.ReplyInlineMarkup(rows=_build_keyboard_rows(buttons)) self._raw = types.ReplyInlineMarkup(rows=_build_keyboard_rows(buttons))