python-telegram-bot/telegram/bot.py

1234 lines
40 KiB
Python
Raw Normal View History

2015-07-07 21:50:36 +02:00
#!/usr/bin/env python
# pylint: disable=E0611,E0213,E1102,C0103,E1101,W0613,R0913,R0904
2015-08-11 21:58:17 +02:00
#
# A library that provides a Python interface to the Telegram Bot API
# Copyright (C) 2015-2016
# Leandro Toledo de Souza <devs@python-telegram-bot.org>
2015-08-11 21:58:17 +02:00
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser Public License for more details.
#
# You should have received a copy of the GNU Lesser Public License
# along with this program. If not, see [http://www.gnu.org/licenses/].
2015-08-10 18:57:31 +02:00
"""This module contains a object that represents a Telegram Bot."""
2015-07-07 21:50:36 +02:00
import functools
2015-07-20 12:53:58 +02:00
import logging
2015-07-07 21:50:36 +02:00
2015-09-20 17:28:10 +02:00
from telegram import (User, Message, Update, UserProfilePhotos, File,
TelegramError, ReplyMarkup, TelegramObject, NullHandler)
from telegram.error import InvalidToken
from telegram.utils import request
from telegram.utils.validate import validate_string
logging.getLogger(__name__).addHandler(NullHandler())
2015-07-07 21:50:36 +02:00
2015-07-20 12:53:58 +02:00
class Bot(TelegramObject):
"""This object represents a Telegram Bot.
Attributes:
id (int):
first_name (str):
last_name (str):
username (str):
name (str):
Args:
token (str):
**kwargs: Arbitrary keyword arguments.
Keyword Args:
base_url (Optional[str]):
"""
2015-07-09 18:19:58 +02:00
2015-07-07 21:50:36 +02:00
def __init__(self,
token,
base_url=None):
2015-12-26 00:10:15 +01:00
self.token = self._valid_token(token)
2015-07-08 22:23:18 +02:00
2015-07-07 21:50:36 +02:00
if base_url is None:
2015-07-08 22:23:18 +02:00
self.base_url = 'https://api.telegram.org/bot%s' % self.token
2015-07-07 21:50:36 +02:00
else:
2015-07-08 22:23:18 +02:00
self.base_url = base_url + self.token
2015-09-20 17:28:10 +02:00
self.base_file_url = 'https://api.telegram.org/file/bot%s' % self.token
self.bot = None
self.logger = logging.getLogger(__name__)
2015-07-20 13:36:08 +02:00
def info(func):
"""
Returns:
"""
2016-04-14 07:01:05 +02:00
@functools.wraps(func)
def decorator(self, *args, **kwargs):
"""
decorator
"""
if not self.bot:
self.getMe()
result = func(self, *args, **kwargs)
return result
2016-04-14 07:01:05 +02:00
return decorator
@property
@info
def id(self):
"""int: """
return self.bot.id
2015-07-08 22:23:18 +02:00
@property
@info
def first_name(self):
"""str: """
return self.bot.first_name
2015-07-08 22:23:18 +02:00
@property
@info
def last_name(self):
"""str: """
return self.bot.last_name
2015-07-20 13:59:41 +02:00
@property
@info
def username(self):
"""str: """
return self.bot.username
2015-07-08 22:23:18 +02:00
@property
def name(self):
"""str: """
2015-07-20 12:53:58 +02:00
return '@%s' % self.username
2015-07-08 22:23:18 +02:00
2015-07-20 12:53:58 +02:00
def log(func):
"""
Returns:
A telegram.Message instance representing the message posted.
"""
2015-07-20 12:53:58 +02:00
logger = logging.getLogger(func.__module__)
2015-07-07 21:50:36 +02:00
2015-07-20 12:53:58 +02:00
@functools.wraps(func)
def decorator(self, *args, **kwargs):
"""
decorator
"""
logger.debug('Entering: %s', func.__name__)
2015-07-20 12:53:58 +02:00
result = func(self, *args, **kwargs)
logger.debug(result)
logger.debug('Exiting: %s', func.__name__)
2015-07-20 12:53:58 +02:00
return result
2016-04-14 07:01:05 +02:00
2015-07-20 12:53:58 +02:00
return decorator
2015-07-07 21:50:36 +02:00
def message(func):
"""
Returns:
A telegram.Message instance representing the message posted.
"""
2016-04-14 07:01:05 +02:00
2015-07-20 12:53:58 +02:00
@functools.wraps(func)
def decorator(self, *args, **kwargs):
"""
decorator
"""
url, data = func(self, *args, **kwargs)
return Bot._post_message(url, data, kwargs)
2016-04-14 07:01:05 +02:00
return decorator
@staticmethod
def _post_message(url, data, kwargs, timeout=None, network_delay=2.):
"""Posts a message to the telegram servers.
2015-09-04 22:15:44 +02:00
Returns:
telegram.Message
"""
if not data.get('chat_id'):
raise TelegramError('Invalid chat_id')
if kwargs.get('reply_to_message_id'):
reply_to_message_id = kwargs.get('reply_to_message_id')
data['reply_to_message_id'] = reply_to_message_id
if kwargs.get('disable_notification'):
disable_notification = kwargs.get('disable_notification')
data['disable_notification'] = disable_notification
if kwargs.get('reply_markup'):
reply_markup = kwargs.get('reply_markup')
if isinstance(reply_markup, ReplyMarkup):
data['reply_markup'] = reply_markup.to_json()
else:
data['reply_markup'] = reply_markup
result = request.post(url, data, timeout=timeout,
network_delay=network_delay)
if result is True:
return result
return Message.de_json(result)
2015-07-20 12:53:58 +02:00
@log
def getMe(self):
"""A simple method for testing your bot's auth token.
Returns:
A telegram.User instance representing that bot if the
credentials are valid, None otherwise.
"""
url = '%s/getMe' % self.base_url
result = request.get(url)
2015-07-20 12:53:58 +02:00
self.bot = User.de_json(result)
2015-08-14 21:30:30 +02:00
return self.bot
2015-07-20 12:53:58 +02:00
@log
@message
2015-07-07 21:50:36 +02:00
def sendMessage(self,
chat_id,
text,
parse_mode=None,
2015-07-07 21:50:36 +02:00
disable_web_page_preview=None,
**kwargs):
2015-07-08 00:54:00 +02:00
"""Use this method to send text messages.
Args:
chat_id:
2015-12-16 15:31:02 +01:00
Unique identifier for the message recipient - telegram.Chat id.
parse_mode:
2016-03-11 20:32:10 +01:00
Send 'Markdown', if you want Telegram apps to show bold, italic and
inline URLs in your bot's message. [Optional]
2015-07-08 00:54:00 +02:00
text:
2016-03-11 20:32:10 +01:00
Text of the message to be sent. The current maximum length is 4096
UTF8 characters.
2015-07-08 00:54:00 +02:00
disable_web_page_preview:
Disables link previews for links in this message. [Optional]
disable_notification:
Sends the message silently. iOS users will not receive
a notification, Android users will receive a notification
with no sound. Other apps coming soon. [Optional]
2015-07-08 00:54:00 +02:00
reply_to_message_id:
If the message is a reply, ID of the original message. [Optional]
reply_markup:
Additional interface options. A JSON-serialized object for a custom
reply keyboard, instructions to hide keyboard or to force a reply
from the user. [Optional]
2015-07-08 21:58:18 +02:00
2015-07-08 00:54:00 +02:00
Returns:
A telegram.Message instance representing the message posted.
"""
2015-07-07 21:50:36 +02:00
url = '%s/sendMessage' % self.base_url
2015-07-07 21:50:36 +02:00
data = {'chat_id': chat_id,
'text': text}
if parse_mode:
data['parse_mode'] = parse_mode
2015-07-07 21:50:36 +02:00
if disable_web_page_preview:
data['disable_web_page_preview'] = disable_web_page_preview
return url, data
2015-07-07 21:50:36 +02:00
2015-07-20 12:53:58 +02:00
@log
@message
2015-07-07 21:50:36 +02:00
def forwardMessage(self,
chat_id,
from_chat_id,
message_id):
2015-07-08 00:54:00 +02:00
"""Use this method to forward messages of any kind.
Args:
chat_id:
2015-12-16 15:31:02 +01:00
Unique identifier for the message recipient - Chat id.
2015-07-08 00:54:00 +02:00
from_chat_id:
Unique identifier for the chat where the original message was sent
2015-12-16 15:31:02 +01:00
- Chat id.
2015-07-08 00:54:00 +02:00
message_id:
Unique message identifier.
disable_notification:
Sends the message silently. iOS users will not receive
a notification, Android users will receive a notification
with no sound. Other apps coming soon. [Optional]
2015-07-08 21:58:18 +02:00
2015-07-08 00:54:00 +02:00
Returns:
A telegram.Message instance representing the message forwarded.
"""
2015-07-07 21:50:36 +02:00
url = '%s/forwardMessage' % self.base_url
2015-07-07 21:50:36 +02:00
data = {}
if chat_id:
data['chat_id'] = chat_id
if from_chat_id:
data['from_chat_id'] = from_chat_id
if message_id:
data['message_id'] = message_id
return url, data
2015-07-07 21:50:36 +02:00
2015-07-20 12:53:58 +02:00
@log
@message
2015-07-07 21:50:36 +02:00
def sendPhoto(self,
chat_id,
photo,
caption=None,
**kwargs):
2015-07-08 00:54:00 +02:00
"""Use this method to send photos.
Args:
chat_id:
2015-12-16 15:31:02 +01:00
Unique identifier for the message recipient - Chat id.
2015-07-08 00:54:00 +02:00
photo:
Photo to send. You can either pass a file_id as String to resend a
photo that is already on the Telegram servers, or upload a new
photo using multipart/form-data.
caption:
Photo caption (may also be used when resending photos by file_id).
[Optional]
disable_notification:
Sends the message silently. iOS users will not receive
a notification, Android users will receive a notification
with no sound. Other apps coming soon. [Optional]
2015-07-08 00:54:00 +02:00
reply_to_message_id:
If the message is a reply, ID of the original message. [Optional]
reply_markup:
Additional interface options. A JSON-serialized object for a custom
reply keyboard, instructions to hide keyboard or to force a reply
from the user. [Optional]
2015-07-08 21:58:18 +02:00
2015-07-08 00:54:00 +02:00
Returns:
A telegram.Message instance representing the message posted.
"""
2015-07-07 21:50:36 +02:00
url = '%s/sendPhoto' % self.base_url
2015-07-07 21:50:36 +02:00
data = {'chat_id': chat_id,
'photo': photo}
2015-07-07 23:46:32 +02:00
2015-07-07 21:50:36 +02:00
if caption:
data['caption'] = caption
return url, data
2015-07-07 21:50:36 +02:00
2015-07-20 12:53:58 +02:00
@log
@message
2015-07-08 01:10:43 +02:00
def sendAudio(self,
chat_id,
audio,
duration=None,
performer=None,
title=None,
**kwargs):
2015-07-08 01:10:43 +02:00
"""Use this method to send audio files, if you want Telegram clients to
display them in the music player. Your audio must be in an .mp3 format.
On success, the sent Message is returned. Bots can currently send audio
files of up to 50 MB in size, this limit may be changed in the future.
For backward compatibility, when both fields title and description are
empty and mime-type of the sent file is not "audio/mpeg", file is sent
as playable voice message. In this case, your audio must be in an .ogg
file encoded with OPUS. This will be removed in the future. You need to
use sendVoice method instead.
2015-07-08 01:10:43 +02:00
Args:
chat_id:
2015-12-16 15:31:02 +01:00
Unique identifier for the message recipient - Chat id.
2015-07-08 01:10:43 +02:00
audio:
Audio file to send. You can either pass a file_id as String to
resend an audio that is already on the Telegram servers, or upload
a new audio file using multipart/form-data.
duration:
Duration of sent audio in seconds. [Optional]
performer:
Performer of sent audio. [Optional]
title:
Title of sent audio. [Optional]
disable_notification:
Sends the message silently. iOS users will not receive
a notification, Android users will receive a notification
with no sound. Other apps coming soon. [Optional]
2015-07-08 01:10:43 +02:00
reply_to_message_id:
If the message is a reply, ID of the original message. [Optional]
reply_markup:
Additional interface options. A JSON-serialized object for a
custom reply keyboard, instructions to hide keyboard or to force a
reply from the user. [Optional]
2015-07-08 21:58:18 +02:00
2015-07-08 01:10:43 +02:00
Returns:
A telegram.Message instance representing the message posted.
"""
url = '%s/sendAudio' % self.base_url
2015-07-07 21:50:36 +02:00
2015-07-08 01:10:43 +02:00
data = {'chat_id': chat_id,
'audio': audio}
if duration:
data['duration'] = duration
if performer:
data['performer'] = performer
if title:
data['title'] = title
return url, data
2015-07-08 01:10:43 +02:00
2015-07-20 12:53:58 +02:00
@log
@message
2015-07-08 02:12:51 +02:00
def sendDocument(self,
chat_id,
document,
filename=None,
**kwargs):
"""Use this method to send general files.
2015-07-08 02:12:51 +02:00
Args:
chat_id:
2015-12-16 15:31:02 +01:00
Unique identifier for the message recipient - Chat id.
2015-07-08 02:12:51 +02:00
document:
File to send. You can either pass a file_id as String to resend a
file that is already on the Telegram servers, or upload a new file
using multipart/form-data.
filename:
File name that shows in telegram message (it is usefull when you
send file generated by temp module, for example). [Optional]
disable_notification:
Sends the message silently. iOS users will not receive
a notification, Android users will receive a notification
with no sound. Other apps coming soon. [Optional]
2015-07-08 02:12:51 +02:00
reply_to_message_id:
If the message is a reply, ID of the original message. [Optional]
reply_markup:
Additional interface options. A JSON-serialized object for a
custom reply keyboard, instructions to hide keyboard or to force a
reply from the user. [Optional]
2015-07-08 21:58:18 +02:00
2015-07-08 02:12:51 +02:00
Returns:
A telegram.Message instance representing the message posted.
"""
url = '%s/sendDocument' % self.base_url
2015-07-07 21:50:36 +02:00
2015-07-08 02:12:51 +02:00
data = {'chat_id': chat_id,
'document': document}
if filename:
data['filename'] = filename
return url, data
2015-07-08 02:12:51 +02:00
2015-07-20 12:53:58 +02:00
@log
@message
2015-07-08 04:52:12 +02:00
def sendSticker(self,
chat_id,
sticker,
**kwargs):
2015-07-08 14:17:18 +02:00
"""Use this method to send .webp stickers.
Args:
chat_id:
2015-12-16 15:31:02 +01:00
Unique identifier for the message recipient - Chat id.
2015-07-08 14:17:18 +02:00
sticker:
Sticker to send. You can either pass a file_id as String to resend
a sticker that is already on the Telegram servers, or upload a new
sticker using multipart/form-data.
disable_notification:
Sends the message silently. iOS users will not receive
a notification, Android users will receive a notification
with no sound. Other apps coming soon. [Optional]
2015-07-08 14:17:18 +02:00
reply_to_message_id:
If the message is a reply, ID of the original message. [Optional]
reply_markup:
Additional interface options. A JSON-serialized object for a
custom reply keyboard, instructions to hide keyboard or to force a
reply from the user. [Optional]
2015-07-08 21:58:18 +02:00
2015-07-08 14:17:18 +02:00
Returns:
A telegram.Message instance representing the message posted.
"""
2015-07-08 04:52:12 +02:00
url = '%s/sendSticker' % self.base_url
2015-07-07 21:50:36 +02:00
2015-07-08 04:52:12 +02:00
data = {'chat_id': chat_id,
'sticker': sticker}
return url, data
2015-07-08 04:52:12 +02:00
2015-07-20 12:53:58 +02:00
@log
2015-07-08 14:17:18 +02:00
def sendVideo(self,
chat_id,
video,
2015-08-11 22:32:06 +02:00
duration=None,
caption=None,
timeout=None,
**kwargs):
2015-07-08 14:17:18 +02:00
"""Use this method to send video files, Telegram clients support mp4
videos (other formats may be sent as telegram.Document).
Args:
chat_id:
2015-12-16 15:31:02 +01:00
Unique identifier for the message recipient - Chat id.
2015-07-08 14:17:18 +02:00
video:
Video to send. You can either pass a file_id as String to resend a
video that is already on the Telegram servers, or upload a new
video file using multipart/form-data.
2015-08-11 22:32:06 +02:00
duration:
Duration of sent video in seconds. [Optional]
caption:
Video caption (may also be used when resending videos by file_id).
[Optional]
timeout:
float. If this value is specified, use it as the definitive timeout
(in seconds) for urlopen() operations. [Optional]
disable_notification:
Sends the message silently. iOS users will not receive
a notification, Android users will receive a notification
with no sound. Other apps coming soon. [Optional]
2015-07-08 14:17:18 +02:00
reply_to_message_id:
If the message is a reply, ID of the original message. [Optional]
reply_markup:
Additional interface options. A JSON-serialized object for a
custom reply keyboard, instructions to hide keyboard or to force a
reply from the user. [Optional]
2015-07-08 21:58:18 +02:00
2015-07-08 14:17:18 +02:00
Returns:
A telegram.Message instance representing the message posted.
"""
url = '%s/sendVideo' % self.base_url
2015-07-07 21:50:36 +02:00
2015-07-08 14:17:18 +02:00
data = {'chat_id': chat_id,
'video': video}
2015-08-11 22:32:06 +02:00
if duration:
data['duration'] = duration
if caption:
data['caption'] = caption
return self._post_message(url, data, kwargs, timeout=timeout)
2015-07-08 14:17:18 +02:00
@log
@message
def sendVoice(self,
chat_id,
voice,
duration=None,
**kwargs):
"""Use this method to send audio files, if you want Telegram clients to
display the file as a playable voice message. For this to work, your
audio must be in an .ogg file encoded with OPUS (other formats may be
sent as Audio or Document). On success, the sent Message is returned.
Bots can currently send audio files of up to 50 MB in size, this limit
may be changed in the future.
Args:
chat_id:
2015-12-16 15:31:02 +01:00
Unique identifier for the message recipient - Chat id.
voice:
Audio file to send. You can either pass a file_id as String to
resend an audio that is already on the Telegram servers, or upload
a new audio file using multipart/form-data.
duration:
Duration of sent audio in seconds. [Optional]
disable_notification:
Sends the message silently. iOS users will not receive
a notification, Android users will receive a notification
with no sound. Other apps coming soon. [Optional]
reply_to_message_id:
If the message is a reply, ID of the original message. [Optional]
reply_markup:
Additional interface options. A JSON-serialized object for a
custom reply keyboard, instructions to hide keyboard or to force a
reply from the user. [Optional]
Returns:
A telegram.Message instance representing the message posted.
"""
url = '%s/sendVoice' % self.base_url
data = {'chat_id': chat_id,
'voice': voice}
if duration:
data['duration'] = duration
return url, data
2015-07-20 12:53:58 +02:00
@log
@message
def sendLocation(self,
chat_id,
latitude,
longitude,
**kwargs):
"""Use this method to send point on the map.
Args:
chat_id:
2015-12-16 15:31:02 +01:00
Unique identifier for the message recipient - Chat id.
latitude:
Latitude of location.
longitude:
Longitude of location.
disable_notification:
Sends the message silently. iOS users will not receive
a notification, Android users will receive a notification
with no sound. Other apps coming soon. [Optional]
reply_to_message_id:
If the message is a reply, ID of the original message. [Optional]
reply_markup:
Additional interface options. A JSON-serialized object for a
custom reply keyboard, instructions to hide keyboard or to force a
reply from the user. [Optional]
2015-07-08 21:58:18 +02:00
Returns:
A telegram.Message instance representing the message posted.
"""
url = '%s/sendLocation' % self.base_url
2015-07-07 21:50:36 +02:00
data = {'chat_id': chat_id,
'latitude': latitude,
'longitude': longitude}
return url, data
@log
@message
def sendVenue(self,
chat_id,
latitude,
longitude,
title,
address,
foursquare_id=None,
**kwargs):
"""
Use this method to send information about a venue.
Args:
chat_id:
Unique identifier for the target chat or username of the target
channel (in the format @channelusername).
latitude:
Latitude of the venue.
longitude:
Longitude of the venue.
title:
Name of the venue.
address:
Address of the venue.
foursquare_id:
Foursquare identifier of the venue.
disable_notification:
Sends the message silently. iOS users will not receive a
notification, Android users will receive a notification with no
sound.
reply_to_message_id:
If the message is a reply, ID of the original message.
reply_markup:
Additional interface options. A JSON-serialized object for an
inline keyboard, custom reply keyboard, instructions to hide
reply keyboard or to force a reply from the user.
Returns:
A telegram.Message instance representing the message posted.
"""
url = '%s/sendVenue' % self.base_url
data = {'chat_id': chat_id,
'latitude': latitude,
'longitude': longitude,
'address': address,
'title': title}
if foursquare_id:
data['foursquare_id'] = foursquare_id
return url, data
@log
@message
def sendContact(self,
chat_id,
phone_number,
first_name,
last_name=None,
**kwargs):
"""
Use this method to send phone contacts.
Args:
chat_id:
Unique identifier for the target chat or username of the target
channel (in the format @channelusername).
phone_number:
Contact's phone number.
first_name:
Contact's first name.
last_name:
Contact's last name.
disable_notification:
Sends the message silently. iOS users will not receive a
notification, Android users will receive a notification with no
sound.
reply_to_message_id:
If the message is a reply, ID of the original message.
reply_markup:
Additional interface options. A JSON-serialized object for an
inline keyboard, custom reply keyboard, instructions to hide
reply keyboard or to force a reply from the user.
Returns:
A telegram.Message instance representing the message posted.
"""
url = '%s/sendContact' % self.base_url
data = {'chat_id': chat_id,
'phone_number': phone_number,
'first_name': first_name}
if last_name:
data['last_name'] = last_name
return url, data
2015-07-20 12:53:58 +02:00
@log
@message
def sendChatAction(self,
chat_id,
action):
"""Use this method when you need to tell the user that something is
happening on the bot's side. The status is set for 5 seconds or less
(when a message arrives from your bot, Telegram clients clear its
typing status).
Args:
chat_id:
2015-12-16 15:31:02 +01:00
Unique identifier for the message recipient - Chat id.
action:
Type of action to broadcast. Choose one, depending on what the user
is about to receive:
- ChatAction.TYPING for text messages,
- ChatAction.UPLOAD_PHOTO for photos,
2015-08-25 13:45:02 +02:00
- ChatAction.UPLOAD_VIDEO for videos,
- ChatAction.UPLOAD_AUDIO for audio files,
- ChatAction.UPLOAD_DOCUMENT for general files,
- ChatAction.FIND_LOCATION for location data.
"""
url = '%s/sendChatAction' % self.base_url
2015-07-07 21:50:36 +02:00
data = {'chat_id': chat_id,
'action': action}
return url, data
2016-01-04 17:31:06 +01:00
@log
def answerInlineQuery(self,
inline_query_id,
results,
cache_time=None,
is_personal=None,
next_offset=None,
switch_pm_text=None,
switch_pm_parameter=None):
2016-01-04 17:31:06 +01:00
"""Use this method to reply to an inline query.
Args:
inline_query_id (str):
2016-01-04 17:31:06 +01:00
Unique identifier for answered query
results (list[InlineQueryResult]):
A list of results for the inline query
Keyword Args:
cache_time (Optional[int]):
The maximum amount of time the result of the inline query
may be cached on the server
is_personal (Optional[bool]):
Pass True, if results may be cached on the server side only
for the user that sent the query. By default, results may be
returned to any user who sends the same query.
next_offset (Optional[str]):
Pass the offset that a client should send in the next query
with the same text to receive more results. Pass an empty
string if there are no more results or if you don't support
pagination. Offset length can't exceed 64 bytes.
switch_pm_text (Optional[str]):
If passed, clients will display a button with specified text
that switches the user to a private chat with the bot and
sends the bot a start message with the parameter
switch_pm_parameter.
switch_pm_parameter (Optional[str]):
Parameter for the start message sent to the bot when user
presses the switch button.
2016-01-04 17:31:06 +01:00
Returns:
A boolean if answering was successful
"""
validate_string(inline_query_id, 'inline_query_id')
validate_string(inline_query_id, 'next_offset')
2016-01-04 17:31:06 +01:00
url = '%s/answerInlineQuery' % self.base_url
results = [res.to_dict() for res in results]
data = {'inline_query_id': inline_query_id,
'results': results}
2016-04-19 00:08:47 +02:00
if cache_time or cache_time == 0:
data['cache_time'] = int(cache_time)
2016-04-14 07:38:51 +02:00
if is_personal:
data['is_personal'] = bool(is_personal)
2016-04-14 07:38:51 +02:00
if next_offset:
2016-01-04 17:31:06 +01:00
data['next_offset'] = next_offset
if switch_pm_text:
data['switch_pm_text'] = switch_pm_text
if switch_pm_parameter:
data['switch_pm_parameter'] = switch_pm_parameter
2016-01-04 17:31:06 +01:00
result = request.post(url, data)
return result
2015-07-20 12:53:58 +02:00
@log
2015-07-08 15:10:08 +02:00
def getUserProfilePhotos(self,
user_id,
offset=None,
limit=100):
"""Use this method to get a list of profile pictures for a user.
Args:
user_id:
Unique identifier of the target user.
offset:
Sequential number of the first photo to be returned. By default,
all photos are returned. [Optional]
limit:
2015-08-10 18:57:31 +02:00
Limits the number of photos to be retrieved. Values between 1-100
are accepted. Defaults to 100. [Optional]
2015-07-08 21:58:18 +02:00
Returns:
Returns a telegram.UserProfilePhotos object.
"""
2015-07-08 15:10:08 +02:00
url = '%s/getUserProfilePhotos' % self.base_url
2015-07-07 21:50:36 +02:00
2015-07-08 15:10:08 +02:00
data = {'user_id': user_id}
if offset:
data['offset'] = offset
if limit:
data['limit'] = limit
result = request.post(url, data)
2015-07-08 15:10:08 +02:00
return UserProfilePhotos.de_json(result)
2015-07-08 15:10:08 +02:00
2015-09-20 17:28:10 +02:00
@log
def getFile(self,
file_id):
"""Use this method to get basic info about a file and prepare it for
downloading. For the moment, bots can download files of up to 20MB in
size.
Args:
file_id:
File identifier to get info about.
Returns:
Returns a telegram.File object
"""
url = '%s/getFile' % self.base_url
data = {'file_id': file_id}
result = request.post(url, data)
if result.get('file_path'):
result['file_path'] = '%s/%s' % (self.base_file_url,
result['file_path'])
return File.de_json(result)
@log
def kickChatMember(self,
chat_id,
user_id):
"""Use this method to kick a user from a group or a supergroup. In the
case of supergroups, the user will not be able to return to the group
on their own using invite links, etc., unless unbanned first. The bot
must be an administrator in the group for this to work.
Args:
chat_id:
Unique identifier for the target group or username of the target
supergroup (in the format @supergroupusername).
user_id:
Unique identifier of the target user.
Returns:
True on success.
"""
url = '%s/kickChatMember' % self.base_url
data = {'chat_id': chat_id,
'user_id': user_id}
result = request.post(url, data)
return result
@log
def unbanChatMember(self,
chat_id,
user_id):
"""Use this method to unban a previously kicked user in a supergroup.
The user will not return to the group automatically, but will be able
to join via link, etc. The bot must be an administrator in the group
for this to work.
Args:
chat_id:
Unique identifier for the target group or username of the target
supergroup (in the format @supergroupusername).
user_id:
Unique identifier of the target user.
Returns:
True on success.
"""
url = '%s/unbanChatMember' % self.base_url
data = {'chat_id': chat_id,
'user_id': user_id}
result = request.post(url, data)
return result
2016-04-14 02:25:26 +02:00
@log
def answerCallbackQuery(self,
callback_query_id,
text=None,
show_alert=None):
"""Use this method to send answers to callback queries sent from inline
keyboards. The answer will be displayed to the user as a notification
at the top of the chat screen or as an alert.
Args:
callback_query_id:
Unique identifier for the query to be answered.
text:
Text of the notification. If not specified, nothing will be shown
to the user.
show_alert:
If true, an alert will be shown by the client instead of a
notification at the top of the chat screen. Defaults to false.
Returns:
True on success.
"""
url = '%s/answerCallbackQuery' % self.base_url
data = {'callback_query_id': callback_query_id}
if text:
data['text'] = text
if show_alert:
data['show_alert'] = show_alert
result = request.post(url, data)
return result
@log
def editMessageText(self,
text,
chat_id=None,
message_id=None,
inline_message_id=None,
parse_mode=None,
disable_web_page_preview=None,
reply_markup=None):
"""Use this method to edit text messages sent by the bot or via the bot
(for inline bots).
Args:
text:
New text of the message.
chat_id:
Required if inline_message_id is not specified. Unique identifier
for the target chat or username of the target channel (in the
format @channelusername).
message_id:
Required if inline_message_id is not specified. Unique identifier
of the sent message.
inline_message_id:
Required if chat_id and message_id are not specified. Identifier of
the inline message.
parse_mode:
Send Markdown or HTML, if you want Telegram apps to show bold,
italic, fixed-width text or inline URLs in your bot's message.
disable_web_page_preview:
Disables link previews for links in this message.
reply_markup:
A JSON-serialized object for an inline keyboard.
Returns:
Returns a telegram.Message object.
"""
2016-04-16 20:29:43 +02:00
url = '%s/editMessageText' % self.base_url
data = {'text': text}
if chat_id:
data['chat_id'] = chat_id
if message_id:
data['message_id'] = message_id
if inline_message_id:
data['inline_message_id'] = inline_message_id
if reply_markup:
data['parse_mode'] = parse_mode
if reply_markup:
data['disable_web_page_preview'] = disable_web_page_preview
if reply_markup:
2016-04-17 22:11:29 +02:00
if isinstance(reply_markup, ReplyMarkup):
data['reply_markup'] = reply_markup.to_json()
else:
data['reply_markup'] = reply_markup
result = request.post(url, data)
return Message.de_json(result)
@log
def editMessageCaption(self,
caption,
chat_id=None,
message_id=None,
inline_message_id=None,
reply_markup=None):
"""Use this method to edit captions of messages sent by the bot or via
the bot (for inline bots).
Args:
caption:
New caption of the message.
chat_id:
Required if inline_message_id is not specified. Unique identifier
for the target chat or username of the target channel (in the
format @channelusername).
message_id:
Required if inline_message_id is not specified. Unique identifier
of the sent message.
inline_message_id:
Required if chat_id and message_id are not specified. Identifier of
the inline message.
reply_markup:
A JSON-serialized object for an inline keyboard.
Returns:
Returns a telegram.Message object.
"""
url = '%s/editMessageCaption' % self.base_url
data = {'caption': caption}
if chat_id:
data['chat_id'] = chat_id
if message_id:
data['message_id'] = message_id
if inline_message_id:
data['inline_message_id'] = inline_message_id
if reply_markup:
2016-04-17 22:11:29 +02:00
if isinstance(reply_markup, ReplyMarkup):
data['reply_markup'] = reply_markup.to_json()
else:
data['reply_markup'] = reply_markup
result = request.post(url, data)
return Message.de_json(result)
@log
def editMessageReplyMarkup(self,
reply_markup,
chat_id=None,
message_id=None,
inline_message_id=None):
"""Use this method to edit only the reply markup of messages sent by
the bot or via the bot (for inline bots).
Args:
reply_markup:
A JSON-serialized object for an inline keyboard.
chat_id:
Required if inline_message_id is not specified. Unique identifier
for the target chat or username of the target channel (in the
format @channelusername).
message_id:
Required if inline_message_id is not specified. Unique identifier
of the sent message.
inline_message_id:
Required if chat_id and message_id are not specified. Identifier of
the inline message.
Returns:
Returns a telegram.Message object.
"""
url = '%s/editMessageReplyMarkup' % self.base_url
2016-04-17 22:11:29 +02:00
if isinstance(reply_markup, ReplyMarkup):
reply_markup = reply_markup.to_json()
data = {'reply_markup': reply_markup}
if chat_id:
data['chat_id'] = chat_id
if message_id:
data['message_id'] = message_id
if inline_message_id:
data['inline_message_id'] = inline_message_id
result = request.post(url, data)
return Message.de_json(result)
2015-07-20 12:53:58 +02:00
@log
2015-07-07 21:50:36 +02:00
def getUpdates(self,
offset=None,
limit=100,
timeout=0,
network_delay=2.):
2015-07-08 14:22:31 +02:00
"""Use this method to receive incoming updates using long polling.
Args:
offset:
Identifier of the first update to be returned. Must be greater by
one than the highest among the identifiers of previously received
updates. By default, updates starting with the earliest unconfirmed
update are returned. An update is considered confirmed as soon as
getUpdates is called with an offset higher than its update_id.
limit:
2015-08-10 18:57:31 +02:00
Limits the number of updates to be retrieved. Values between 1-100
2015-07-08 14:22:31 +02:00
are accepted. Defaults to 100.
timeout:
Timeout in seconds for long polling. Defaults to 0, i.e. usual
short polling.
network_delay:
2015-11-10 23:31:51 +01:00
Additional timeout in seconds to allow the response from Telegram
to take some time when using long polling. Defaults to 2, which
should be enough for most connections. Increase it if it takes very
long for data to be transmitted from and to the Telegram servers.
2015-07-08 21:58:18 +02:00
2015-07-08 14:22:31 +02:00
Returns:
A list of telegram.Update objects are returned.
"""
2015-07-07 21:50:36 +02:00
url = '%s/getUpdates' % self.base_url
2015-07-07 21:50:36 +02:00
data = {}
if offset:
data['offset'] = offset
if limit:
data['limit'] = limit
if timeout:
data['timeout'] = timeout
result = request.post(url, data, network_delay=network_delay)
2015-07-07 21:50:36 +02:00
if result:
self.logger.debug(
'Getting updates: %s', [u['update_id'] for u in result])
2015-07-20 13:59:41 +02:00
else:
self.logger.debug('No new updates found.')
2015-07-20 13:59:41 +02:00
return [Update.de_json(x) for x in result]
2015-07-07 21:50:36 +02:00
2015-07-20 12:53:58 +02:00
@log
def setWebhook(self,
webhook_url=None,
certificate=None):
2015-07-09 18:19:58 +02:00
"""Use this method to specify a url and receive incoming updates via an
outgoing webhook. Whenever there is an update for the bot, we will send
an HTTPS POST request to the specified url, containing a
JSON-serialized Update. In case of an unsuccessful request, we will
give up after a reasonable amount of attempts.
Args:
url:
HTTPS url to send updates to.
Use an empty string to remove webhook integration
Returns:
True if successful else TelegramError was raised
"""
url = '%s/setWebhook' % self.base_url
2015-07-07 21:50:36 +02:00
data = {}
if webhook_url:
data['url'] = webhook_url
if certificate:
data['certificate'] = certificate
2015-07-09 18:19:58 +02:00
result = request.post(url, data)
2015-07-09 18:19:58 +02:00
return result
2015-07-20 12:53:58 +02:00
@staticmethod
def de_json(data):
pass
2015-07-20 13:36:08 +02:00
def to_dict(self):
"""
Returns:
dict:
"""
2015-07-20 12:53:58 +02:00
data = {'id': self.id,
'username': self.username,
'first_name': self.username}
if self.last_name:
data['last_name'] = self.last_name
return data
def __reduce__(self):
2015-08-12 04:46:09 +02:00
return (self.__class__, (self.token,
2015-08-12 04:43:40 +02:00
self.base_url.replace(self.token, '')))
2015-12-26 00:10:15 +01:00
@staticmethod
def _valid_token(token):
"""a very basic validation on token"""
left, sep, _right = token.partition(':')
2015-12-26 18:34:05 +01:00
if (not sep) or (not left.isdigit()) or (len(left) < 3):
raise InvalidToken()
2015-12-26 00:10:15 +01:00
return token