From 419bc5878f52ea84c6518b89e5c2d0cc69f1765c Mon Sep 17 00:00:00 2001 From: _run Date: Thu, 30 Jun 2022 17:06:39 +0500 Subject: [PATCH] Fix typehint for ```set_state``` --- telebot/__init__.py | 4 ++-- telebot/async_telebot.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/telebot/__init__.py b/telebot/__init__.py index f034cca..a7f3694 100644 --- a/telebot/__init__.py +++ b/telebot/__init__.py @@ -34,7 +34,7 @@ logger.addHandler(console_output_handler) logger.setLevel(logging.ERROR) from telebot import apihelper, util, types -from telebot.handler_backends import MemoryHandlerBackend, FileHandlerBackend, BaseMiddleware, CancelUpdate, SkipHandler +from telebot.handler_backends import MemoryHandlerBackend, FileHandlerBackend, BaseMiddleware, CancelUpdate, SkipHandler, State from telebot.custom_filters import SimpleCustomFilter, AdvancedCustomFilter @@ -2927,7 +2927,7 @@ class TeleBot: - def set_state(self, user_id: int, state: Union[int, str], chat_id: int=None) -> None: + def set_state(self, user_id: int, state: Union[int, str, State], chat_id: int=None) -> None: """ Sets a new state of a user. diff --git a/telebot/async_telebot.py b/telebot/async_telebot.py index dce9dec..9571c04 100644 --- a/telebot/async_telebot.py +++ b/telebot/async_telebot.py @@ -14,7 +14,7 @@ import telebot.types # storages from telebot.asyncio_storage import StateMemoryStorage, StatePickleStorage -from telebot.asyncio_handler_backends import CancelUpdate, SkipHandler +from telebot.asyncio_handler_backends import CancelUpdate, SkipHandler, State from inspect import signature @@ -3455,7 +3455,7 @@ class AsyncTeleBot: return await asyncio_helper.delete_sticker_from_set(self.token, sticker) - async def set_state(self, user_id: int, state: str, chat_id: int=None): + async def set_state(self, user_id: int, state: Union[State, int, str], chat_id: int=None): """ Sets a new state of a user.