Merge pull request #196 from Birdi7/fix-rename-stacklevel

Fix stacklevel arguments in renamed arguments in the dispatcher
This commit is contained in:
Alex Root Junior 2019-08-24 18:19:55 +03:00 committed by GitHub
commit 172d834e3c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -915,7 +915,7 @@ class Dispatcher(DataMixin, ContextInstanceMixin):
return FSMContext(storage=self.storage, chat=chat, user=user) return FSMContext(storage=self.storage, chat=chat, user=user)
@renamed_argument(old_name='user', new_name='user_id', until_version='3.0', stacklevel=4) @renamed_argument(old_name='user', new_name='user_id', until_version='3.0', stacklevel=3)
@renamed_argument(old_name='chat', new_name='chat_id', until_version='3.0', stacklevel=4) @renamed_argument(old_name='chat', new_name='chat_id', until_version='3.0', stacklevel=4)
async def throttle(self, key, *, rate=None, user_id=None, chat_id=None, no_error=None) -> bool: async def throttle(self, key, *, rate=None, user_id=None, chat_id=None, no_error=None) -> bool:
""" """
@ -974,8 +974,8 @@ class Dispatcher(DataMixin, ContextInstanceMixin):
raise Throttled(key=key, chat=chat_id, user=user_id, **data) raise Throttled(key=key, chat=chat_id, user=user_id, **data)
return result return result
@renamed_argument('user', 'user_id', '3.0') @renamed_argument(old_name='user', new_name='user_id', until_version='3.0', stacklevel=3)
@renamed_argument('chat', 'chat_id', '3.0') @renamed_argument(old_name='chat', new_name='chat_id', until_version='3.0', stacklevel=4)
async def check_key(self, key, chat_id=None, user_id=None): async def check_key(self, key, chat_id=None, user_id=None):
""" """
Get information about key in bucket Get information about key in bucket
@ -996,8 +996,8 @@ class Dispatcher(DataMixin, ContextInstanceMixin):
data = bucket.get(key, {}) data = bucket.get(key, {})
return Throttled(key=key, chat=chat_id, user=user_id, **data) return Throttled(key=key, chat=chat_id, user=user_id, **data)
@renamed_argument('user', 'user_id', '3.0') @renamed_argument(old_name='user', new_name='user_id', until_version='3.0', stacklevel=3)
@renamed_argument('chat', 'chat_id', '3.0') @renamed_argument(old_name='chat', new_name='chat_id', until_version='3.0', stacklevel=4)
async def release_key(self, key, chat_id=None, user_id=None): async def release_key(self, key, chat_id=None, user_id=None):
""" """
Release blocked key Release blocked key