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 2024-12-22 22:45:09 +01:00
Code Issues Projects Releases Packages Wiki Activity

basic token validation

Browse source 
refs #134
This commit is contained in:
Noam Meltzer 2015-12-26 01:10:15 +02:00
parent 21fdaa4ff7
commit aeca3ca174
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
telegram/bot.py
View file

@ -52,7 +52,7 @@ class Bot(TelegramObject):
def __init__(self, def __init__(self,
token, token,
base_url=None): base_url=None):
self.token = token self.token = self._valid_token(token)
if base_url is None: if base_url is None:
self.base_url = 'https://api.telegram.org/bot%s' % self.token self.base_url = 'https://api.telegram.org/bot%s' % self.token
@ -744,3 +744,11 @@ class Bot(TelegramObject):
def __reduce__(self): def __reduce__(self):
return (self.__class__, (self.token, return (self.__class__, (self.token,
self.base_url.replace(self.token, ''))) self.base_url.replace(self.token, '')))
@staticmethod
def _valid_token(token):
"""a very basic validation on token"""
left, sep, _right = token.partition(':')
if (sep is None) or (not left.isdigit()) or (len(left) < 3):
raise TelegramError('Invalid token')
return token