OpenSource/python-telegram-bot
OpenSource/python-telegram-bot
1
0
Fork 0
mirror of https://github.com/python-telegram-bot/python-telegram-bot.git synced 2025-03-14 19:48:57 +01:00
Code Issues Projects Releases Packages Wiki Activity

Simplified if statement, fixed docs

Browse source
This commit is contained in:
evgfilim1 2017-06-22 13:35:59 +05:00
parent 3bf4399a4f
commit ca9068f2b5
1 changed files with 18 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
telegram/ext/filters.py
View file

@ -325,21 +325,23 @@ class Filters(object):
group = _Group()
class user(BaseFilter):
"""Filters messages to allow only those which are from specified user ID
"""Filters messages to allow only those which are from specified user ID.
Notes:
Only one of chat_id or username must be used here.
Args:
user_id(Optional[int]): which user ID to allow through. Required if username is not
specified
username(Optional[str]): which username to allow through. Required if user_id is not
specified
user_id(Optional[int]): which user ID to allow through.
username(Optional[str]): which username to allow through. If username starts with '@'
symbol, it will be ignored.
Raises:
ValueError
"""
def __init__(self, user_id=None, username=None):
if (user_id is None and username is None) or (bool(user_id) and bool(username)):
raise ValueError('You must specify either user_id or username')
if not (bool(user_id) ^ bool(username)):
raise ValueError('One and only one of user_id or username must be used')
if username is not None and username.startswith('@'):
self.username = username[1:]
else:
@ -355,21 +357,23 @@ class Filters(object):
message.from_user.username == self.username)
class chat(BaseFilter):
"""Filters messages to allow only those which are from specified chat ID
"""Filters messages to allow only those which are from specified chat ID.
Notes:
Only one of chat_id or username must be used here.
Args:
chat_id(Optional[int]): which chat ID to allow through. Required if username is not
specified
username(Optional[str]): which username to allow through. Required if chat_id is not
specified
chat_id(Optional[int]): which chat ID to allow through.
username(Optional[str]): which username to allow through. If username starts with '@'
symbol, it will be ignored.
Raises:
ValueError
"""
def __init__(self, chat_id=None, username=None):
if (chat_id is None and username is None) or (bool(chat_id) and bool(username)):
raise ValueError('You must specify either chat_id or username')
if not (bool(chat_id) ^ bool(username)):
raise ValueError('One and only one of user_id or username must be used')
if username is not None and username.startswith('@'):
self.username = username[1:]
else: