mirror of
https://github.com/eternnoir/pyTelegramBotAPI.git
synced 2023-08-10 21:12:57 +03:00
e811163b5f
Added the field file_unique_id to the objects Animation, Audio, Document, PassportFile, PhotoSize, Sticker, Video, VideoNote, Voice, File and the fields small_file_unique_id and big_file_unique_id to the object ChatPhoto. (Bot API 4.5)
2547 lines
98 KiB
Python
2547 lines
98 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import logging
|
|
|
|
try:
|
|
import ujson as json
|
|
except ImportError:
|
|
import json
|
|
|
|
from telebot import util
|
|
|
|
DISABLE_KEYLEN_ERROR = False
|
|
|
|
logger = logging.getLogger('TeleBot')
|
|
|
|
|
|
class JsonSerializable(object):
|
|
"""
|
|
Subclasses of this class are guaranteed to be able to be converted to JSON format.
|
|
All subclasses of this class must override to_json.
|
|
"""
|
|
|
|
def to_json(self):
|
|
"""
|
|
Returns a JSON string representation of this class.
|
|
|
|
This function must be overridden by subclasses.
|
|
:return: a JSON formatted string.
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
|
|
class Dictionaryable(object):
|
|
"""
|
|
Subclasses of this class are guaranteed to be able to be converted to dictionary.
|
|
All subclasses of this class must override to_dict.
|
|
"""
|
|
|
|
def to_dict(self):
|
|
"""
|
|
Returns a DICT with class field values
|
|
|
|
This function must be overridden by subclasses.
|
|
:return: a DICT
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
|
|
class JsonDeserializable(object):
|
|
"""
|
|
Subclasses of this class are guaranteed to be able to be created from a json-style dict or json formatted string.
|
|
All subclasses of this class must override de_json.
|
|
"""
|
|
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
"""
|
|
Returns an instance of this class from the given json dict or string.
|
|
|
|
This function must be overridden by subclasses.
|
|
:return: an instance of this class created from the given json dict or string.
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
@staticmethod
|
|
def check_json(json_type):
|
|
"""
|
|
Checks whether json_type is a dict or a string. If it is already a dict, it is returned as-is.
|
|
If it is not, it is converted to a dict by means of json.loads(json_type)
|
|
:param json_type:
|
|
:return:
|
|
"""
|
|
if util.is_dict(json_type):
|
|
return json_type
|
|
elif util.is_string(json_type):
|
|
return json.loads(json_type)
|
|
else:
|
|
raise ValueError("json_type should be a json dict or string.")
|
|
|
|
def __str__(self):
|
|
d = {}
|
|
for x, y in self.__dict__.items():
|
|
if hasattr(y, '__dict__'):
|
|
d[x] = y.__dict__
|
|
else:
|
|
d[x] = y
|
|
|
|
return str(d)
|
|
|
|
|
|
class Update(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
update_id = obj['update_id']
|
|
message = Message.de_json(obj.get('message'))
|
|
edited_message = Message.de_json(obj.get('edited_message'))
|
|
channel_post = Message.de_json(obj.get('channel_post'))
|
|
edited_channel_post = Message.de_json(obj.get('edited_channel_post'))
|
|
inline_query = InlineQuery.de_json(obj.get('inline_query'))
|
|
chosen_inline_result = ChosenInlineResult.de_json(obj.get('chosen_inline_result'))
|
|
callback_query = CallbackQuery.de_json(obj.get('callback_query'))
|
|
shipping_query = ShippingQuery.de_json(obj.get('shipping_query'))
|
|
pre_checkout_query = PreCheckoutQuery.de_json(obj.get('pre_checkout_query'))
|
|
poll = Poll.de_json(obj.get('poll'))
|
|
poll_answer = PollAnswer.de_json(obj.get('poll_answer'))
|
|
return cls(update_id, message, edited_message, channel_post, edited_channel_post, inline_query,
|
|
chosen_inline_result, callback_query, shipping_query, pre_checkout_query, poll, poll_answer)
|
|
|
|
def __init__(self, update_id, message, edited_message, channel_post, edited_channel_post, inline_query,
|
|
chosen_inline_result, callback_query, shipping_query, pre_checkout_query, poll, poll_answer):
|
|
self.update_id = update_id
|
|
self.message = message
|
|
self.edited_message = edited_message
|
|
self.channel_post = channel_post
|
|
self.edited_channel_post = edited_channel_post
|
|
self.inline_query = inline_query
|
|
self.chosen_inline_result = chosen_inline_result
|
|
self.callback_query = callback_query
|
|
self.shipping_query = shipping_query
|
|
self.pre_checkout_query = pre_checkout_query
|
|
self.poll = poll
|
|
self.poll_answer = poll_answer
|
|
|
|
|
|
class WebhookInfo(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
url = obj['url']
|
|
has_custom_certificate = obj['has_custom_certificate']
|
|
pending_update_count = obj['pending_update_count']
|
|
last_error_date = obj.get('last_error_date')
|
|
last_error_message = obj.get('last_error_message')
|
|
max_connections = obj.get('max_connections')
|
|
allowed_updates = obj.get('allowed_updates')
|
|
return cls(url, has_custom_certificate, pending_update_count, last_error_date, last_error_message,
|
|
max_connections, allowed_updates)
|
|
|
|
def __init__(self, url, has_custom_certificate, pending_update_count, last_error_date, last_error_message,
|
|
max_connections, allowed_updates):
|
|
self.url = url
|
|
self.has_custom_certificate = has_custom_certificate
|
|
self.pending_update_count = pending_update_count
|
|
self.last_error_date = last_error_date
|
|
self.last_error_message = last_error_message
|
|
self.max_connections = max_connections
|
|
self.allowed_updates = allowed_updates
|
|
|
|
|
|
class User(JsonDeserializable, Dictionaryable, JsonSerializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
id = obj['id']
|
|
is_bot = obj['is_bot']
|
|
first_name = obj['first_name']
|
|
last_name = obj.get('last_name')
|
|
username = obj.get('username')
|
|
language_code = obj.get('language_code')
|
|
can_join_groups = obj.get('can_join_groups')
|
|
can_read_all_group_messages = obj.get('can_read_all_group_messages')
|
|
supports_inline_queries = obj.get('supports_inline_queries')
|
|
return cls(id, is_bot, first_name, last_name, username, language_code, can_join_groups, can_read_all_group_messages, supports_inline_queries)
|
|
|
|
def __init__(self, id, is_bot, first_name, last_name=None, username=None, language_code=None, can_join_groups=None, can_read_all_group_messages=None, supports_inline_queries=None):
|
|
self.id = id
|
|
self.is_bot = is_bot
|
|
self.first_name = first_name
|
|
self.username = username
|
|
self.last_name = last_name
|
|
self.language_code = language_code
|
|
self.can_join_groups = can_join_groups
|
|
self.can_read_all_group_messages = can_read_all_group_messages
|
|
self.supports_inline_queries = supports_inline_queries
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
return {'id': self.id,
|
|
'is_bot': self.is_bot,
|
|
'first_name': self.first_name,
|
|
'last_name': self.last_name,
|
|
'username': self.username,
|
|
'language_code': self.language_code,
|
|
'can_join_groups': self.can_join_groups,
|
|
'can_read_all_group_messages': self.can_read_all_group_messages,
|
|
'supports_inline_queries': self.supports_inline_queries}
|
|
|
|
|
|
class GroupChat(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
id = obj['id']
|
|
title = obj['title']
|
|
return cls(id, title)
|
|
|
|
def __init__(self, id, title):
|
|
self.id = id
|
|
self.title = title
|
|
|
|
|
|
class Chat(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if json_string is None:
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
id = obj['id']
|
|
type = obj['type']
|
|
title = obj.get('title')
|
|
username = obj.get('username')
|
|
first_name = obj.get('first_name')
|
|
last_name = obj.get('last_name')
|
|
all_members_are_administrators = obj.get('all_members_are_administrators')
|
|
photo = ChatPhoto.de_json(obj.get('photo'))
|
|
description = obj.get('description')
|
|
invite_link = obj.get('invite_link')
|
|
pinned_message = Message.de_json(obj.get('pinned_message'))
|
|
permissions = ChatPermissions.de_json(obj.get('permissions'))
|
|
slow_mode_delay = obj.get('slow_mode_delay')
|
|
sticker_set_name = obj.get('sticker_set_name')
|
|
can_set_sticker_set = obj.get('can_set_sticker_set')
|
|
return cls(
|
|
id, type, title, username, first_name, last_name,
|
|
all_members_are_administrators, photo, description, invite_link,
|
|
pinned_message, permissions, slow_mode_delay, sticker_set_name,
|
|
can_set_sticker_set)
|
|
|
|
def __init__(self, id, type, title=None, username=None, first_name=None,
|
|
last_name=None, all_members_are_administrators=None,
|
|
photo=None, description=None, invite_link=None,
|
|
pinned_message=None, permissions=None, slow_mode_delay=None,
|
|
sticker_set_name=None, can_set_sticker_set=None):
|
|
self.id = id
|
|
self.type = type
|
|
self.title = title
|
|
self.username = username
|
|
self.first_name = first_name
|
|
self.last_name = last_name
|
|
self.all_members_are_administrators = all_members_are_administrators
|
|
self.photo = photo
|
|
self.description = description
|
|
self.invite_link = invite_link
|
|
self.pinned_message = pinned_message
|
|
self.permissions = permissions
|
|
self.slow_mode_delay = slow_mode_delay
|
|
self.sticker_set_name = sticker_set_name
|
|
self.can_set_sticker_set = can_set_sticker_set
|
|
|
|
|
|
class Message(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
message_id = obj['message_id']
|
|
from_user = User.de_json(obj.get('from'))
|
|
date = obj['date']
|
|
chat = Chat.de_json(obj['chat'])
|
|
content_type = None
|
|
opts = {}
|
|
if 'forward_from' in obj:
|
|
opts['forward_from'] = User.de_json(obj['forward_from'])
|
|
if 'forward_from_chat' in obj:
|
|
opts['forward_from_chat'] = Chat.de_json(obj['forward_from_chat'])
|
|
if 'forward_from_message_id' in obj:
|
|
opts['forward_from_message_id'] = obj.get('forward_from_message_id')
|
|
if 'forward_signature' in obj:
|
|
opts['forward_signature'] = obj.get('forward_signature')
|
|
if 'forward_sender_name' in obj:
|
|
opts['forward_sender_name'] = obj.get('forward_sender_name')
|
|
if 'forward_date' in obj:
|
|
opts['forward_date'] = obj.get('forward_date')
|
|
if 'reply_to_message' in obj:
|
|
opts['reply_to_message'] = Message.de_json(obj['reply_to_message'])
|
|
if 'edit_date' in obj:
|
|
opts['edit_date'] = obj.get('edit_date')
|
|
if 'media_group_id' in obj:
|
|
opts['media_group_id'] = obj.get('media_group_id')
|
|
if 'author_signature' in obj:
|
|
opts['author_signature'] = obj.get('author_signature')
|
|
if 'text' in obj:
|
|
opts['text'] = obj['text']
|
|
content_type = 'text'
|
|
if 'entities' in obj:
|
|
opts['entities'] = Message.parse_entities(obj['entities'])
|
|
if 'caption_entities' in obj:
|
|
opts['caption_entities'] = Message.parse_entities(obj['caption_entities'])
|
|
if 'audio' in obj:
|
|
opts['audio'] = Audio.de_json(obj['audio'])
|
|
content_type = 'audio'
|
|
if 'animation' in obj:
|
|
opts['animation'] = Animation.de_json(obj['animation'])
|
|
content_type = 'animation'
|
|
if 'document' in obj:
|
|
opts['document'] = Document.de_json(obj['document'])
|
|
content_type = 'document'
|
|
if 'game' in obj:
|
|
opts['game'] = Game.de_json(obj['game'])
|
|
content_type = 'game'
|
|
if 'photo' in obj:
|
|
opts['photo'] = Message.parse_photo(obj['photo'])
|
|
content_type = 'photo'
|
|
if 'sticker' in obj:
|
|
opts['sticker'] = Sticker.de_json(obj['sticker'])
|
|
content_type = 'sticker'
|
|
if 'video' in obj:
|
|
opts['video'] = Video.de_json(obj['video'])
|
|
content_type = 'video'
|
|
if 'video_note' in obj:
|
|
opts['video_note'] = VideoNote.de_json(obj['video_note'])
|
|
content_type = 'video_note'
|
|
if 'voice' in obj:
|
|
opts['voice'] = Audio.de_json(obj['voice'])
|
|
content_type = 'voice'
|
|
if 'caption' in obj:
|
|
opts['caption'] = obj['caption']
|
|
if 'contact' in obj:
|
|
opts['contact'] = Contact.de_json(json.dumps(obj['contact']))
|
|
content_type = 'contact'
|
|
if 'location' in obj:
|
|
opts['location'] = Location.de_json(obj['location'])
|
|
content_type = 'location'
|
|
if 'venue' in obj:
|
|
opts['venue'] = Venue.de_json(obj['venue'])
|
|
content_type = 'venue'
|
|
if 'dice' in obj:
|
|
opts['dice'] = Dice.de_json(obj['dice'])
|
|
content_type = 'dice'
|
|
if 'new_chat_members' in obj:
|
|
new_chat_members = []
|
|
for member in obj['new_chat_members']:
|
|
new_chat_members.append(User.de_json(member))
|
|
opts['new_chat_members'] = new_chat_members
|
|
content_type = 'new_chat_members'
|
|
if 'left_chat_member' in obj:
|
|
opts['left_chat_member'] = User.de_json(obj['left_chat_member'])
|
|
content_type = 'left_chat_member'
|
|
if 'new_chat_title' in obj:
|
|
opts['new_chat_title'] = obj['new_chat_title']
|
|
content_type = 'new_chat_title'
|
|
if 'new_chat_photo' in obj:
|
|
opts['new_chat_photo'] = Message.parse_photo(obj['new_chat_photo'])
|
|
content_type = 'new_chat_photo'
|
|
if 'delete_chat_photo' in obj:
|
|
opts['delete_chat_photo'] = obj['delete_chat_photo']
|
|
content_type = 'delete_chat_photo'
|
|
if 'group_chat_created' in obj:
|
|
opts['group_chat_created'] = obj['group_chat_created']
|
|
content_type = 'group_chat_created'
|
|
if 'supergroup_chat_created' in obj:
|
|
opts['supergroup_chat_created'] = obj['supergroup_chat_created']
|
|
content_type = 'supergroup_chat_created'
|
|
if 'channel_chat_created' in obj:
|
|
opts['channel_chat_created'] = obj['channel_chat_created']
|
|
content_type = 'channel_chat_created'
|
|
if 'migrate_to_chat_id' in obj:
|
|
opts['migrate_to_chat_id'] = obj['migrate_to_chat_id']
|
|
content_type = 'migrate_to_chat_id'
|
|
if 'migrate_from_chat_id' in obj:
|
|
opts['migrate_from_chat_id'] = obj['migrate_from_chat_id']
|
|
content_type = 'migrate_from_chat_id'
|
|
if 'pinned_message' in obj:
|
|
opts['pinned_message'] = Message.de_json(obj['pinned_message'])
|
|
content_type = 'pinned_message'
|
|
if 'invoice' in obj:
|
|
opts['invoice'] = Invoice.de_json(obj['invoice'])
|
|
content_type = 'invoice'
|
|
if 'successful_payment' in obj:
|
|
opts['successful_payment'] = SuccessfulPayment.de_json(obj['successful_payment'])
|
|
content_type = 'successful_payment'
|
|
if 'connected_website' in obj:
|
|
opts['connected_website'] = obj['connected_website']
|
|
content_type = 'connected_website'
|
|
if 'poll' in obj:
|
|
opts['poll'] = Poll.de_json(obj['poll'])
|
|
content_type = 'poll'
|
|
if 'passport_data' in obj:
|
|
opts['passport_data'] = obj['passport_data']
|
|
content_type = 'passport_data'
|
|
return cls(message_id, from_user, date, chat, content_type, opts, json_string)
|
|
|
|
@classmethod
|
|
def parse_chat(cls, chat):
|
|
if 'first_name' not in chat:
|
|
return GroupChat.de_json(chat)
|
|
else:
|
|
return User.de_json(chat)
|
|
|
|
@classmethod
|
|
def parse_photo(cls, photo_size_array):
|
|
ret = []
|
|
for ps in photo_size_array:
|
|
ret.append(PhotoSize.de_json(ps))
|
|
return ret
|
|
|
|
@classmethod
|
|
def parse_entities(cls, message_entity_array):
|
|
ret = []
|
|
for me in message_entity_array:
|
|
ret.append(MessageEntity.de_json(me))
|
|
return ret
|
|
|
|
def __init__(self, message_id, from_user, date, chat, content_type, options, json_string):
|
|
self.content_type = content_type
|
|
self.message_id = message_id
|
|
self.from_user = from_user
|
|
self.date = date
|
|
self.chat = chat
|
|
self.forward_from = None
|
|
self.forward_from_chat = None
|
|
self.forward_from_message_id = None
|
|
self.forward_signature = None
|
|
self.forward_sender_name = None
|
|
self.forward_date = None
|
|
self.reply_to_message = None
|
|
self.edit_date = None
|
|
self.media_group_id = None
|
|
self.author_signature = None
|
|
self.text = None
|
|
self.entities = None
|
|
self.caption_entities = None
|
|
self.audio = None
|
|
self.document = None
|
|
self.photo = None
|
|
self.sticker = None
|
|
self.video = None
|
|
self.video_note = None
|
|
self.voice = None
|
|
self.caption = None
|
|
self.contact = None
|
|
self.location = None
|
|
self.venue = None
|
|
self.animation = None
|
|
self.dice = None
|
|
self.new_chat_member = None # Deprecated since Bot API 3.0. Not processed anymore
|
|
self.new_chat_members = None
|
|
self.left_chat_member = None
|
|
self.new_chat_title = None
|
|
self.new_chat_photo = None
|
|
self.delete_chat_photo = None
|
|
self.group_chat_created = None
|
|
self.supergroup_chat_created = None
|
|
self.channel_chat_created = None
|
|
self.migrate_to_chat_id = None
|
|
self.migrate_from_chat_id = None
|
|
self.pinned_message = None
|
|
self.invoice = None
|
|
self.successful_payment = None
|
|
self.connected_website = None
|
|
for key in options:
|
|
setattr(self, key, options[key])
|
|
self.json = json_string
|
|
|
|
def __html_text(self, text, entities):
|
|
"""
|
|
Author: @sviat9440
|
|
Updaters: @badiboy
|
|
Message: "*Test* parse _formatting_, [url](https://example.com), [text_mention](tg://user?id=123456) and mention @username"
|
|
|
|
Example:
|
|
message.html_text
|
|
>> "<b>Test</b> parse <i>formatting</i>, <a href=\"https://example.com\">url</a>, <a href=\"tg://user?id=123456\">text_mention</a> and mention @username"
|
|
|
|
Cusom subs:
|
|
You can customize the substitutes. By default, there is no substitute for the entities: hashtag, bot_command, email. You can add or modify substitute an existing entity.
|
|
Example:
|
|
message.custom_subs = {"bold": "<strong class=\"example\">{text}</strong>", "italic": "<i class=\"example\">{text}</i>", "mention": "<a href={url}>{text}</a>"}
|
|
message.html_text
|
|
>> "<strong class=\"example\">Test</strong> parse <i class=\"example\">formatting</i>, <a href=\"https://example.com\">url</a> and <a href=\"tg://user?id=123456\">text_mention</a> and mention <a href=\"https://t.me/username\">@username</a>"
|
|
"""
|
|
|
|
if not entities:
|
|
return text
|
|
|
|
_subs = {
|
|
"bold" : "<b>{text}</b>",
|
|
"italic" : "<i>{text}</i>",
|
|
"pre" : "<pre>{text}</pre>",
|
|
"code" : "<code>{text}</code>",
|
|
#"url" : "<a href=\"{url}\">{text}</a>", # @badiboy plain URLs have no text and do not need tags
|
|
"text_link": "<a href=\"{url}\">{text}</a>",
|
|
"strikethrough": "<s>{text}</s>",
|
|
"underline": "<u>{text}</u>"
|
|
}
|
|
|
|
if hasattr(self, "custom_subs"):
|
|
for key, value in self.custom_subs.items():
|
|
_subs[key] = value
|
|
utf16_text = text.encode("utf-16-le")
|
|
html_text = ""
|
|
|
|
def func(upd_text, subst_type=None, url=None, user=None):
|
|
upd_text = upd_text.decode("utf-16-le")
|
|
if subst_type == "text_mention":
|
|
subst_type = "url"
|
|
url = "tg://user?id={0}".format(user.id)
|
|
elif subst_type == "mention":
|
|
url = "https://t.me/{0}".format(upd_text[1:])
|
|
upd_text = upd_text.replace("&", "&").replace("<", "<").replace(">", ">")
|
|
if not subst_type or not _subs.get(subst_type):
|
|
return upd_text
|
|
subs = _subs.get(subst_type)
|
|
return subs.format(text=upd_text, url=url)
|
|
|
|
offset = 0
|
|
for entity in entities:
|
|
if entity.offset > offset:
|
|
html_text += func(utf16_text[offset * 2 : entity.offset * 2])
|
|
offset = entity.offset
|
|
html_text += func(utf16_text[offset * 2 : (offset + entity.length) * 2], entity.type, entity.url, entity.user)
|
|
offset += entity.length
|
|
elif entity.offset == offset:
|
|
html_text += func(utf16_text[offset * 2 : (offset + entity.length) * 2], entity.type, entity.url, entity.user)
|
|
offset += entity.length
|
|
else:
|
|
# TODO: process nested entities from Bot API 4.5
|
|
# Now ignoring them
|
|
pass
|
|
if offset * 2 < len(utf16_text):
|
|
html_text += func(utf16_text[offset * 2:])
|
|
return html_text
|
|
|
|
@property
|
|
def html_text(self):
|
|
return self.__html_text(self.text, self.entities)
|
|
|
|
@property
|
|
def html_caption(self):
|
|
return self.__html_text(self.caption, self.caption_entities)
|
|
|
|
|
|
class MessageEntity(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
type = obj['type']
|
|
offset = obj['offset']
|
|
length = obj['length']
|
|
url = obj.get('url')
|
|
user = User.de_json(obj.get('user'))
|
|
language = obj.get('language')
|
|
return cls(type, offset, length, url, user, language)
|
|
|
|
def __init__(self, type, offset, length, url=None, user=None, language=None):
|
|
self.type = type
|
|
self.offset = offset
|
|
self.length = length
|
|
self.url = url
|
|
self.user = user
|
|
self.language = language
|
|
|
|
|
|
class Dice(JsonSerializable, Dictionaryable, JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
value = obj['value']
|
|
emoji = obj['emoji']
|
|
return cls(value, emoji)
|
|
|
|
def __init__(self, value, emoji):
|
|
self.value = value
|
|
self.emoji = emoji
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
return {'value': self.value,
|
|
'emoji': self.emoji}
|
|
|
|
|
|
class PhotoSize(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
file_id = obj['file_id']
|
|
file_unique_id = obj['file_unique_id']
|
|
width = obj['width']
|
|
height = obj['height']
|
|
file_size = obj.get('file_size')
|
|
return cls(file_id, file_unique_id, width, height, file_size)
|
|
|
|
def __init__(self, file_id, file_unique_id, width, height, file_size=None):
|
|
self.file_size = file_size
|
|
self.file_unique_id = file_unique_id
|
|
self.height = height
|
|
self.width = width
|
|
self.file_id = file_id
|
|
|
|
|
|
class Audio(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
file_id = obj['file_id']
|
|
file_unique_id = obj['file_unique_id']
|
|
duration = obj['duration']
|
|
performer = obj.get('performer')
|
|
title = obj.get('title')
|
|
mime_type = obj.get('mime_type')
|
|
file_size = obj.get('file_size')
|
|
return cls(file_id, file_unique_id, duration, performer, title, mime_type, file_size)
|
|
|
|
def __init__(self, file_id, file_unique_id, duration, performer=None, title=None, mime_type=None, file_size=None):
|
|
self.file_id = file_id
|
|
self.file_unique_id = file_unique_id
|
|
self.duration = duration
|
|
self.performer = performer
|
|
self.title = title
|
|
self.mime_type = mime_type
|
|
self.file_size = file_size
|
|
|
|
|
|
class Voice(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
file_id = obj['file_id']
|
|
file_unique_id = obj['file_unique_id']
|
|
duration = obj['duration']
|
|
mime_type = obj.get('mime_type')
|
|
file_size = obj.get('file_size')
|
|
return cls(file_id, file_unique_id, duration, mime_type, file_size)
|
|
|
|
def __init__(self, file_id, file_unique_id, duration, mime_type=None, file_size=None):
|
|
self.file_id = file_id
|
|
self.file_unique_id = file_unique_id
|
|
self.duration = duration
|
|
self.mime_type = mime_type
|
|
self.file_size = file_size
|
|
|
|
|
|
class Document(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
file_id = obj['file_id']
|
|
file_unique_id = obj['file_unique_id']
|
|
thumb = None
|
|
if 'thumb' in obj and 'file_id' in obj['thumb']:
|
|
thumb = PhotoSize.de_json(obj['thumb'])
|
|
file_name = obj.get('file_name')
|
|
mime_type = obj.get('mime_type')
|
|
file_size = obj.get('file_size')
|
|
return cls(file_id, file_unique_id, thumb, file_name, mime_type, file_size)
|
|
|
|
def __init__(self, file_id, file_unique_id, thumb=None, file_name=None, mime_type=None, file_size=None):
|
|
self.file_id = file_id
|
|
self.file_unique_id = file_unique_id
|
|
self.thumb = thumb
|
|
self.file_name = file_name
|
|
self.mime_type = mime_type
|
|
self.file_size = file_size
|
|
|
|
|
|
class Video(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None):
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
file_id = obj['file_id']
|
|
file_unique_id = obj['file_unique_id']
|
|
width = obj['width']
|
|
height = obj['height']
|
|
duration = obj['duration']
|
|
thumb = PhotoSize.de_json(obj.get('thumb'))
|
|
mime_type = obj.get('mime_type')
|
|
file_size = obj.get('file_size')
|
|
return cls(file_id, file_unique_id, width, height, duration, thumb, mime_type, file_size)
|
|
|
|
def __init__(self, file_id, file_unique_id, width, height, duration, thumb=None, mime_type=None, file_size=None):
|
|
self.file_id = file_id
|
|
self.file_unique_id = file_unique_id
|
|
self.width = width
|
|
self.height = height
|
|
self.duration = duration
|
|
self.thumb = thumb
|
|
self.mime_type = mime_type
|
|
self.file_size = file_size
|
|
|
|
|
|
class VideoNote(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None):
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
file_id = obj['file_id']
|
|
file_unique_id = obj['file_unique_id']
|
|
length = obj['length']
|
|
duration = obj['duration']
|
|
thumb = PhotoSize.de_json(obj.get('thumb'))
|
|
file_size = obj.get('file_size')
|
|
return cls(file_id, file_unique_id, length, duration, thumb, file_size)
|
|
|
|
def __init__(self, file_id, file_unique_id, length, duration, thumb=None, file_size=None):
|
|
self.file_id = file_id
|
|
self.file_unique_id = file_unique_id
|
|
self.length = length
|
|
self.duration = duration
|
|
self.thumb = thumb
|
|
self.file_size = file_size
|
|
|
|
|
|
class Contact(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None):
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
phone_number = obj['phone_number']
|
|
first_name = obj['first_name']
|
|
last_name = obj.get('last_name')
|
|
user_id = obj.get('user_id')
|
|
vcard = obj.get('vcard')
|
|
return cls(phone_number, first_name, last_name, user_id, vcard)
|
|
|
|
def __init__(self, phone_number, first_name, last_name=None, user_id=None, vcard=None):
|
|
self.phone_number = phone_number
|
|
self.first_name = first_name
|
|
self.last_name = last_name
|
|
self.user_id = user_id
|
|
self.vcard = vcard
|
|
|
|
|
|
class Location(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None):
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
longitude = obj['longitude']
|
|
latitude = obj['latitude']
|
|
return cls(longitude, latitude)
|
|
|
|
def __init__(self, longitude, latitude):
|
|
self.longitude = longitude
|
|
self.latitude = latitude
|
|
|
|
|
|
class Venue(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None):
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
location = Location.de_json(obj['location'])
|
|
title = obj['title']
|
|
address = obj['address']
|
|
foursquare_id = obj.get('foursquare_id')
|
|
foursquare_type = obj.get('foursquare_type')
|
|
return cls(location, title, address, foursquare_id, foursquare_type)
|
|
|
|
def __init__(self, location, title, address, foursquare_id=None, foursquare_type=None):
|
|
self.location = location
|
|
self.title = title
|
|
self.address = address
|
|
self.foursquare_id = foursquare_id
|
|
self.foursquare_type = foursquare_type
|
|
|
|
|
|
class UserProfilePhotos(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None):
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
total_count = obj['total_count']
|
|
photos = [[PhotoSize.de_json(y) for y in x] for x in obj['photos']]
|
|
return cls(total_count, photos)
|
|
|
|
def __init__(self, total_count, photos):
|
|
self.total_count = total_count
|
|
self.photos = photos
|
|
|
|
|
|
class File(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None):
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
file_id = obj['file_id']
|
|
file_unique_id = obj['file_unique_id']
|
|
file_size = obj.get('file_size')
|
|
file_path = obj.get('file_path')
|
|
return cls(file_id, file_unique_id, file_size, file_path)
|
|
|
|
def __init__(self, file_id, file_unique_id, file_size, file_path):
|
|
self.file_id = file_id
|
|
self.file_unique_id = file_unique_id
|
|
self.file_size = file_size
|
|
self.file_path = file_path
|
|
|
|
|
|
class ForceReply(JsonSerializable):
|
|
def __init__(self, selective=None):
|
|
self.selective = selective
|
|
|
|
def to_json(self):
|
|
json_dict = {'force_reply': True}
|
|
if self.selective:
|
|
json_dict['selective'] = True
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class ReplyKeyboardRemove(JsonSerializable):
|
|
def __init__(self, selective=None):
|
|
self.selective = selective
|
|
|
|
def to_json(self):
|
|
json_dict = {'remove_keyboard': True}
|
|
if self.selective:
|
|
json_dict['selective'] = True
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class ReplyKeyboardMarkup(JsonSerializable):
|
|
max_row_keys = 12
|
|
|
|
def __init__(self, resize_keyboard=None, one_time_keyboard=None, selective=None, row_width=3):
|
|
if row_width > self.max_row_keys:
|
|
# Todo: Will be replaced with Exception in future releases
|
|
if not DISABLE_KEYLEN_ERROR:
|
|
logger.error('Telegram does not support reply keyboard row width over %d.' % self.max_row_keys)
|
|
row_width = self.max_row_keys
|
|
|
|
self.resize_keyboard = resize_keyboard
|
|
self.one_time_keyboard = one_time_keyboard
|
|
self.selective = selective
|
|
self.row_width = row_width
|
|
self.keyboard = []
|
|
|
|
def add(self, *args, row_width=None):
|
|
"""
|
|
This function adds strings to the keyboard, while not exceeding row_width.
|
|
E.g. ReplyKeyboardMarkup#add("A", "B", "C") yields the json result {keyboard: [["A"], ["B"], ["C"]]}
|
|
when row_width is set to 1.
|
|
When row_width is set to 2, the following is the result of this function: {keyboard: [["A", "B"], ["C"]]}
|
|
See https://core.telegram.org/bots/api#replykeyboardmarkup
|
|
:param args: KeyboardButton to append to the keyboard
|
|
:param row_width: width of row
|
|
:return: self, to allow function chaining.
|
|
"""
|
|
if row_width is None:
|
|
row_width = self.row_width
|
|
|
|
|
|
if row_width > self.max_row_keys:
|
|
# Todo: Will be replaced with Exception in future releases
|
|
if not DISABLE_KEYLEN_ERROR:
|
|
logger.error('Telegram does not support reply keyboard row width over %d.' % self.max_row_keys)
|
|
row_width = self.max_row_keys
|
|
|
|
for row in util.chunks(args, row_width):
|
|
button_array = []
|
|
for button in row:
|
|
if util.is_string(button):
|
|
button_array.append({'text': button})
|
|
elif util.is_bytes(button):
|
|
button_array.append({'text': button.decode('utf-8')})
|
|
else:
|
|
button_array.append(button.to_dict())
|
|
self.keyboard.append(button_array)
|
|
|
|
return self
|
|
|
|
def row(self, *args):
|
|
"""
|
|
Adds a list of KeyboardButton to the keyboard. This function does not consider row_width.
|
|
ReplyKeyboardMarkup#row("A")#row("B", "C")#to_json() outputs '{keyboard: [["A"], ["B", "C"]]}'
|
|
See https://core.telegram.org/bots/api#replykeyboardmarkup
|
|
:param args: strings
|
|
:return: self, to allow function chaining.
|
|
"""
|
|
|
|
return self.add(*args, row_width=self.max_row_keys)
|
|
|
|
def to_json(self):
|
|
"""
|
|
Converts this object to its json representation following the Telegram API guidelines described here:
|
|
https://core.telegram.org/bots/api#replykeyboardmarkup
|
|
:return:
|
|
"""
|
|
json_dict = {'keyboard': self.keyboard}
|
|
if self.one_time_keyboard:
|
|
json_dict['one_time_keyboard'] = True
|
|
if self.resize_keyboard:
|
|
json_dict['resize_keyboard'] = True
|
|
if self.selective:
|
|
json_dict['selective'] = True
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class KeyboardButton(Dictionaryable, JsonSerializable):
|
|
def __init__(self, text, request_contact=None, request_location=None, request_poll=None):
|
|
self.text = text
|
|
self.request_contact = request_contact
|
|
self.request_location = request_location
|
|
self.request_poll = request_poll
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
json_dict = {'text': self.text}
|
|
if self.request_contact:
|
|
json_dict['request_contact'] = self.request_contact
|
|
if self.request_location:
|
|
json_dict['request_location'] = self.request_location
|
|
if self.request_poll:
|
|
json_dict['request_poll'] = self.request_poll.to_dict()
|
|
return json_dict
|
|
|
|
|
|
class KeyboardButtonPollType(Dictionaryable):
|
|
def __init__(self, type=''):
|
|
self.type = type
|
|
|
|
def to_dict(self):
|
|
return {'type': self.type}
|
|
|
|
|
|
class InlineKeyboardMarkup(Dictionaryable, JsonSerializable):
|
|
max_row_keys = 8
|
|
|
|
def __init__(self, row_width=3):
|
|
"""
|
|
This object represents an inline keyboard that appears
|
|
right next to the message it belongs to.
|
|
|
|
:return:
|
|
"""
|
|
if row_width > self.max_row_keys:
|
|
# Todo: Will be replaced with Exception in future releases
|
|
logger.error('Telegram does not support inline keyboard row width over %d.' % self.max_row_keys)
|
|
row_width = self.max_row_keys
|
|
|
|
self.row_width = row_width
|
|
self.keyboard = []
|
|
|
|
def add(self, *args, row_width=None):
|
|
"""
|
|
This method adds buttons to the keyboard without exceeding row_width.
|
|
|
|
E.g. InlineKeyboardMarkup#add("A", "B", "C") yields the json result:
|
|
{keyboard: [["A"], ["B"], ["C"]]}
|
|
when row_width is set to 1.
|
|
When row_width is set to 2, the result:
|
|
{keyboard: [["A", "B"], ["C"]]}
|
|
See https://core.telegram.org/bots/api#inlinekeyboardmarkup
|
|
|
|
:param args: Array of InlineKeyboardButton to append to the keyboard
|
|
:param row_width: width of row
|
|
:return: self, to allow function chaining.
|
|
"""
|
|
if row_width is None:
|
|
row_width = self.row_width
|
|
|
|
if row_width > self.max_row_keys:
|
|
# Todo: Will be replaced with Exception in future releases
|
|
logger.error('Telegram does not support inline keyboard row width over %d.' % self.max_row_keys)
|
|
row_width = self.max_row_keys
|
|
|
|
for row in util.chunks(args, row_width):
|
|
button_array = [button.to_dict() for button in row]
|
|
self.keyboard.append(button_array)
|
|
|
|
return self
|
|
|
|
def row(self, *args):
|
|
"""
|
|
Adds a list of InlineKeyboardButton to the keyboard.
|
|
This metod does not consider row_width.
|
|
|
|
InlineKeyboardMarkup.row("A").row("B", "C").to_json() outputs:
|
|
'{keyboard: [["A"], ["B", "C"]]}'
|
|
See https://core.telegram.org/bots/api#inlinekeyboardmarkup
|
|
|
|
:param args: Array of InlineKeyboardButton to append to the keyboard
|
|
:return: self, to allow function chaining.
|
|
"""
|
|
|
|
return self.add(*args, row_width=self.max_row_keys)
|
|
|
|
def to_json(self):
|
|
"""
|
|
Converts this object to its json representation
|
|
following the Telegram API guidelines described here:
|
|
https://core.telegram.org/bots/api#inlinekeyboardmarkup
|
|
:return:
|
|
"""
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
json_dict = {'inline_keyboard': self.keyboard}
|
|
return json_dict
|
|
|
|
|
|
class LoginUrl(Dictionaryable, JsonSerializable):
|
|
def __init__(self, url, forward_text=None, bot_username=None, request_write_access=None):
|
|
self.url = url
|
|
self.forward_text = forward_text
|
|
self.bot_username = bot_username
|
|
self.request_write_access = request_write_access
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
json_dict = {'url': self.url}
|
|
if self.forward_text:
|
|
json_dict['forward_text'] = self.forward_text
|
|
if self.bot_username:
|
|
json_dict['bot_username'] = self.bot_username
|
|
if self.request_write_access:
|
|
json_dict['request_write_access'] = self.request_write_access
|
|
return json_dict
|
|
|
|
|
|
class InlineKeyboardButton(Dictionaryable, JsonSerializable):
|
|
def __init__(self, text, url=None, callback_data=None, switch_inline_query=None,
|
|
switch_inline_query_current_chat=None, callback_game=None, pay=None, login_url=None):
|
|
self.text = text
|
|
self.url = url
|
|
self.callback_data = callback_data
|
|
self.switch_inline_query = switch_inline_query
|
|
self.switch_inline_query_current_chat = switch_inline_query_current_chat
|
|
self.callback_game = callback_game
|
|
self.pay = pay
|
|
self.login_url = login_url
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
json_dict = {'text': self.text}
|
|
if self.url:
|
|
json_dict['url'] = self.url
|
|
if self.callback_data:
|
|
json_dict['callback_data'] = self.callback_data
|
|
if self.switch_inline_query is not None:
|
|
json_dict['switch_inline_query'] = self.switch_inline_query
|
|
if self.switch_inline_query_current_chat is not None:
|
|
json_dict['switch_inline_query_current_chat'] = self.switch_inline_query_current_chat
|
|
if self.callback_game is not None:
|
|
json_dict['callback_game'] = self.callback_game
|
|
if self.pay is not None:
|
|
json_dict['pay'] = self.pay
|
|
if self.login_url is not None:
|
|
json_dict['login_url'] = self.login_url.to_dict()
|
|
return json_dict
|
|
|
|
|
|
class CallbackQuery(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
id = obj['id']
|
|
from_user = User.de_json(obj['from'])
|
|
message = Message.de_json(obj.get('message'))
|
|
inline_message_id = obj.get('inline_message_id')
|
|
chat_instance = obj['chat_instance']
|
|
data = obj.get('data')
|
|
game_short_name = obj.get('game_short_name')
|
|
return cls(id, from_user, data, chat_instance, message, inline_message_id, game_short_name)
|
|
|
|
def __init__(self, id, from_user, data, chat_instance, message=None, inline_message_id=None, game_short_name=None):
|
|
self.game_short_name = game_short_name
|
|
self.chat_instance = chat_instance
|
|
self.id = id
|
|
self.from_user = from_user
|
|
self.message = message
|
|
self.data = data
|
|
self.inline_message_id = inline_message_id
|
|
|
|
|
|
class ChatPhoto(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None):
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
small_file_id = obj['small_file_id']
|
|
small_file_unique_id = obj['small_file_unique_id']
|
|
big_file_id = obj['big_file_id']
|
|
big_file_unique_id = obj['big_file_unique_id']
|
|
return cls(small_file_id, small_file_unique_id, big_file_id, big_file_unique_id)
|
|
|
|
def __init__(self, small_file_id, small_file_unique_id, big_file_id, big_file_unique_id):
|
|
self.small_file_id = small_file_id
|
|
self.small_file_unique_id = small_file_unique_id
|
|
self.big_file_id = big_file_id
|
|
self.big_file_unique_id = big_file_unique_id
|
|
|
|
|
|
class ChatMember(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if json_string is None:
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
user = User.de_json(obj['user'])
|
|
status = obj['status']
|
|
custom_title = obj.get('custom_title')
|
|
until_date = obj.get('until_date')
|
|
can_be_edited = obj.get('can_be_edited')
|
|
can_post_messages = obj.get('can_post_messages')
|
|
can_edit_messages = obj.get('can_edit_messages')
|
|
can_delete_messages = obj.get('can_delete_messages')
|
|
can_restrict_members = obj.get('can_restrict_members')
|
|
can_promote_members = obj.get('can_promote_members')
|
|
can_change_info = obj.get('can_change_info')
|
|
can_invite_users = obj.get('can_invite_users')
|
|
can_pin_messages = obj.get('can_pin_messages')
|
|
is_member = obj.get('is_member')
|
|
can_send_messages = obj.get('can_send_messages')
|
|
can_send_media_messages = obj.get('can_send_media_messages')
|
|
can_send_polls = obj.get('can_send_polls')
|
|
can_send_other_messages = obj.get('can_send_other_messages')
|
|
can_add_web_page_previews = obj.get('can_add_web_page_previews')
|
|
return cls(
|
|
user, status, custom_title, until_date, can_be_edited, can_post_messages,
|
|
can_edit_messages, can_delete_messages, can_restrict_members,
|
|
can_promote_members, can_change_info, can_invite_users, can_pin_messages,
|
|
is_member, can_send_messages, can_send_media_messages, can_send_polls,
|
|
can_send_other_messages, can_add_web_page_previews)
|
|
|
|
def __init__(self, user, status, custom_title=None, until_date=None, can_be_edited=None,
|
|
can_post_messages=None, can_edit_messages=None, can_delete_messages=None,
|
|
can_restrict_members=None, can_promote_members=None, can_change_info=None,
|
|
can_invite_users=None, can_pin_messages=None, is_member=None,
|
|
can_send_messages=None, can_send_media_messages=None, can_send_polls=None,
|
|
can_send_other_messages=None, can_add_web_page_previews=None):
|
|
self.user = user
|
|
self.status = status
|
|
self.custom_title = custom_title
|
|
self.until_date = until_date
|
|
self.can_be_edited = can_be_edited
|
|
self.can_post_messages = can_post_messages
|
|
self.can_edit_messages = can_edit_messages
|
|
self.can_delete_messages = can_delete_messages
|
|
self.can_restrict_members = can_restrict_members
|
|
self.can_promote_members = can_promote_members
|
|
self.can_change_info = can_change_info
|
|
self.can_invite_users = can_invite_users
|
|
self.can_pin_messages = can_pin_messages
|
|
self.is_member = is_member
|
|
self.can_send_messages = can_send_messages
|
|
self.can_send_media_messages = can_send_media_messages
|
|
self.can_send_polls = can_send_polls
|
|
self.can_send_other_messages = can_send_other_messages
|
|
self.can_add_web_page_previews = can_add_web_page_previews
|
|
|
|
|
|
class ChatPermissions(JsonDeserializable, JsonSerializable, Dictionaryable):
|
|
def __init__(self, can_send_messages=None, can_send_media_messages=None,
|
|
can_send_polls=None, can_send_other_messages=None,
|
|
can_add_web_page_previews=None, can_change_info=None,
|
|
can_invite_users=None, can_pin_messages=None):
|
|
self.can_send_messages = can_send_messages
|
|
self.can_send_media_messages = can_send_media_messages
|
|
self.can_send_polls = can_send_polls
|
|
self.can_send_other_messages = can_send_other_messages
|
|
self.can_add_web_page_previews = can_add_web_page_previews
|
|
self.can_change_info = can_change_info
|
|
self.can_invite_users = can_invite_users
|
|
self.can_pin_messages = can_pin_messages
|
|
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if json_string is None:
|
|
return json_string
|
|
obj = cls.check_json(json_string)
|
|
can_send_messages = obj.get('can_send_messages')
|
|
can_send_media_messages = obj.get('can_send_media_messages')
|
|
can_send_polls = obj.get('can_send_polls')
|
|
can_send_other_messages = obj.get('can_send_other_messages')
|
|
can_add_web_page_previews = obj.get('can_add_web_page_previews')
|
|
can_change_info = obj.get('can_change_info')
|
|
can_invite_users = obj.get('can_invite_users')
|
|
can_pin_messages = obj.get('can_pin_messages')
|
|
return cls(
|
|
can_send_messages, can_send_media_messages, can_send_polls,
|
|
can_send_other_messages, can_add_web_page_previews,
|
|
can_change_info, can_invite_users, can_pin_messages)
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
json_dict = dict()
|
|
if self.can_send_messages is not None:
|
|
json_dict['can_send_messages'] = self.can_send_messages
|
|
if self.can_send_media_messages is not None:
|
|
json_dict['can_send_media_messages'] = self.can_send_media_messages
|
|
if self.can_send_polls is not None:
|
|
json_dict['can_send_polls'] = self.can_send_polls
|
|
if self.can_send_other_messages is not None:
|
|
json_dict['can_send_other_messages'] = self.can_send_other_messages
|
|
if self.can_add_web_page_previews is not None:
|
|
json_dict['can_add_web_page_previews'] = self.can_add_web_page_previews
|
|
if self.can_change_info is not None:
|
|
json_dict['can_change_info'] = self.can_change_info
|
|
if self.can_invite_users is not None:
|
|
json_dict['can_invite_users'] = self.can_invite_users
|
|
if self.can_pin_messages is not None:
|
|
json_dict['can_pin_messages'] = self.can_pin_messages
|
|
return json_dict
|
|
|
|
|
|
class BotCommand(JsonSerializable):
|
|
def __init__(self, command, description):
|
|
"""
|
|
This object represents a bot command.
|
|
:param command: Text of the command, 1-32 characters.
|
|
Can contain only lowercase English letters, digits and underscores.
|
|
:param description: Description of the command, 3-256 characters.
|
|
:return:
|
|
"""
|
|
self.command = command
|
|
self.description = description
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
return {'command': self.command, 'description': self.description}
|
|
|
|
|
|
# InlineQuery
|
|
|
|
class InlineQuery(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None):
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
id = obj['id']
|
|
from_user = User.de_json(obj['from'])
|
|
location = Location.de_json(obj.get('location'))
|
|
query = obj['query']
|
|
offset = obj['offset']
|
|
return cls(id, from_user, location, query, offset)
|
|
|
|
def __init__(self, id, from_user, location, query, offset):
|
|
"""
|
|
This object represents an incoming inline query.
|
|
When the user sends an empty query, your bot could
|
|
return some default or trending results.
|
|
:param id: string Unique identifier for this query
|
|
:param from_user: User Sender
|
|
:param location: Sender location, only for bots that request user location
|
|
:param query: String Text of the query
|
|
:param offset: String Offset of the results to be returned, can be controlled by the bot
|
|
:return: InlineQuery Object
|
|
"""
|
|
self.id = id
|
|
self.from_user = from_user
|
|
self.location = location
|
|
self.query = query
|
|
self.offset = offset
|
|
|
|
|
|
class InputTextMessageContent(Dictionaryable):
|
|
def __init__(self, message_text, parse_mode=None, disable_web_page_preview=None):
|
|
self.message_text = message_text
|
|
self.parse_mode = parse_mode
|
|
self.disable_web_page_preview = disable_web_page_preview
|
|
|
|
def to_dict(self):
|
|
json_dic = {'message_text': self.message_text}
|
|
if self.parse_mode:
|
|
json_dic['parse_mode'] = self.parse_mode
|
|
if self.disable_web_page_preview:
|
|
json_dic['disable_web_page_preview'] = self.disable_web_page_preview
|
|
return json_dic
|
|
|
|
|
|
class InputLocationMessageContent(Dictionaryable):
|
|
def __init__(self, latitude, longitude, live_period=None):
|
|
self.latitude = latitude
|
|
self.longitude = longitude
|
|
self.live_period = live_period
|
|
|
|
def to_dict(self):
|
|
json_dic = {'latitude': self.latitude, 'longitude': self.longitude}
|
|
if self.live_period:
|
|
json_dic['live_period'] = self.live_period
|
|
return json_dic
|
|
|
|
|
|
class InputVenueMessageContent(Dictionaryable):
|
|
def __init__(self, latitude, longitude, title, address, foursquare_id=None, foursquare_type=None):
|
|
self.latitude = latitude
|
|
self.longitude = longitude
|
|
self.title = title
|
|
self.address = address
|
|
self.foursquare_id = foursquare_id
|
|
self.foursquare_type = foursquare_type
|
|
|
|
def to_dict(self):
|
|
json_dict = {
|
|
'latitude': self.latitude,
|
|
'longitude': self.longitude,
|
|
'title': self.title,
|
|
'address' : self.address
|
|
}
|
|
if self.foursquare_id:
|
|
json_dict['foursquare_id'] = self.foursquare_id
|
|
if self.foursquare_type:
|
|
json_dict['foursquare_type'] = self.foursquare_type
|
|
return json_dict
|
|
|
|
|
|
class InputContactMessageContent(Dictionaryable):
|
|
def __init__(self, phone_number, first_name, last_name=None, vcard=None):
|
|
self.phone_number = phone_number
|
|
self.first_name = first_name
|
|
self.last_name = last_name
|
|
self.vcard = vcard
|
|
|
|
def to_dict(self):
|
|
json_dict = {'phone_numbe': self.phone_number, 'first_name': self.first_name}
|
|
if self.last_name:
|
|
json_dict['last_name'] = self.last_name
|
|
if self.vcard:
|
|
json_dict['vcard'] = self.vcard
|
|
return json_dict
|
|
|
|
|
|
class ChosenInlineResult(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None):
|
|
return None
|
|
obj = cls.check_json(json_string)
|
|
result_id = obj['result_id']
|
|
from_user = User.de_json(obj['from'])
|
|
query = obj['query']
|
|
location = Location.de_json(obj.get('location'))
|
|
inline_message_id = obj.get('inline_message_id')
|
|
return cls(result_id, from_user, query, location, inline_message_id)
|
|
|
|
def __init__(self, result_id, from_user, query, location=None, inline_message_id=None):
|
|
"""
|
|
This object represents a result of an inline query
|
|
that was chosen by the user and sent to their chat partner.
|
|
:param result_id: string The unique identifier for the result that was chosen.
|
|
:param from_user: User The user that chose the result.
|
|
:param query: String The query that was used to obtain the result.
|
|
:return: ChosenInlineResult Object.
|
|
"""
|
|
self.result_id = result_id
|
|
self.from_user = from_user
|
|
self.query = query
|
|
self.location = location
|
|
self.inline_message_id = inline_message_id
|
|
|
|
|
|
class InlineQueryResultArticle(JsonSerializable):
|
|
def __init__(self, id, title, input_message_content, reply_markup=None, url=None,
|
|
hide_url=None, description=None, thumb_url=None, thumb_width=None, thumb_height=None):
|
|
"""
|
|
Represents a link to an article or web page.
|
|
:param id: Unique identifier for this result, 1-64 Bytes.
|
|
:param title: Title of the result.
|
|
:param input_message_content: InputMessageContent : Content of the message to be sent
|
|
:param reply_markup: InlineKeyboardMarkup : Inline keyboard attached to the message
|
|
:param url: URL of the result.
|
|
:param hide_url: Pass True, if you don't want the URL to be shown in the message.
|
|
:param description: Short description of the result.
|
|
:param thumb_url: Url of the thumbnail for the result.
|
|
:param thumb_width: Thumbnail width.
|
|
:param thumb_height: Thumbnail height
|
|
:return:
|
|
"""
|
|
self.type = 'article'
|
|
self.id = id
|
|
self.title = title
|
|
self.input_message_content = input_message_content
|
|
self.reply_markup = reply_markup
|
|
self.url = url
|
|
self.hide_url = hide_url
|
|
self.description = description
|
|
self.thumb_url = thumb_url
|
|
self.thumb_width = thumb_width
|
|
self.thumb_height = thumb_height
|
|
|
|
def to_json(self):
|
|
json_dict = {
|
|
'type': self.type,
|
|
'id': self.id,
|
|
'title': self.title,
|
|
'input_message_content': self.input_message_content.to_dict()}
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.url:
|
|
json_dict['url'] = self.url
|
|
if self.hide_url:
|
|
json_dict['hide_url'] = self.hide_url
|
|
if self.description:
|
|
json_dict['description'] = self.description
|
|
if self.thumb_url:
|
|
json_dict['thumb_url'] = self.thumb_url
|
|
if self.thumb_width:
|
|
json_dict['thumb_width'] = self.thumb_width
|
|
if self.thumb_height:
|
|
json_dict['thumb_height'] = self.thumb_height
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultPhoto(JsonSerializable):
|
|
def __init__(self, id, photo_url, thumb_url, photo_width=None, photo_height=None, title=None,
|
|
description=None, caption=None, parse_mode=None, reply_markup=None, input_message_content=None):
|
|
"""
|
|
Represents a link to a photo.
|
|
:param id: Unique identifier for this result, 1-64 bytes
|
|
:param photo_url: A valid URL of the photo. Photo must be in jpeg format. Photo size must not exceed 5MB
|
|
:param thumb_url: URL of the thumbnail for the photo
|
|
:param photo_width: Width of the photo.
|
|
:param photo_height: Height of the photo.
|
|
:param title: Title for the result.
|
|
:param description: Short description of the result.
|
|
:param caption: Caption of the photo to be sent, 0-200 characters.
|
|
:param parse_mode: Send Markdown or HTML, if you want Telegram apps to show bold, italic, fixed-width text or
|
|
inline URLs in the media caption.
|
|
:param reply_markup: InlineKeyboardMarkup : Inline keyboard attached to the message
|
|
:param input_message_content: InputMessageContent : Content of the message to be sent instead of the photo
|
|
:return:
|
|
"""
|
|
self.type = 'photo'
|
|
self.id = id
|
|
self.photo_url = photo_url
|
|
self.photo_width = photo_width
|
|
self.photo_height = photo_height
|
|
self.thumb_url = thumb_url
|
|
self.title = title
|
|
self.description = description
|
|
self.caption = caption
|
|
self.parse_mode = parse_mode
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
|
|
def to_json(self):
|
|
json_dict = {'type': self.type, 'id': self.id, 'photo_url': self.photo_url, 'thumb_url': self.thumb_url}
|
|
if self.photo_width:
|
|
json_dict['photo_width'] = self.photo_width
|
|
if self.photo_height:
|
|
json_dict['photo_height'] = self.photo_height
|
|
if self.title:
|
|
json_dict['title'] = self.title
|
|
if self.description:
|
|
json_dict['description'] = self.description
|
|
if self.caption:
|
|
json_dict['caption'] = self.caption
|
|
if self.parse_mode:
|
|
json_dict['parse_mode'] = self.parse_mode
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultGif(JsonSerializable):
|
|
def __init__(self, id, gif_url, thumb_url, gif_width=None, gif_height=None, title=None, caption=None,
|
|
reply_markup=None, input_message_content=None, gif_duration=None):
|
|
"""
|
|
Represents a link to an animated GIF file.
|
|
:param id: Unique identifier for this result, 1-64 bytes.
|
|
:param gif_url: A valid URL for the GIF file. File size must not exceed 1MB
|
|
:param thumb_url: URL of the static thumbnail (jpeg or gif) for the result.
|
|
:param gif_width: Width of the GIF.
|
|
:param gif_height: Height of the GIF.
|
|
:param title: Title for the result.
|
|
:param caption: Caption of the GIF file to be sent, 0-200 characters
|
|
:param reply_markup: InlineKeyboardMarkup : Inline keyboard attached to the message
|
|
:param input_message_content: InputMessageContent : Content of the message to be sent instead of the photo
|
|
:return:
|
|
"""
|
|
self.type = 'gif'
|
|
self.id = id
|
|
self.gif_url = gif_url
|
|
self.gif_width = gif_width
|
|
self.gif_height = gif_height
|
|
self.thumb_url = thumb_url
|
|
self.title = title
|
|
self.caption = caption
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.gif_duration = gif_duration
|
|
|
|
def to_json(self):
|
|
json_dict = {'type': self.type, 'id': self.id, 'gif_url': self.gif_url, 'thumb_url': self.thumb_url}
|
|
if self.gif_height:
|
|
json_dict['gif_height'] = self.gif_height
|
|
if self.gif_width:
|
|
json_dict['gif_width'] = self.gif_width
|
|
if self.title:
|
|
json_dict['title'] = self.title
|
|
if self.caption:
|
|
json_dict['caption'] = self.caption
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
if self.gif_duration:
|
|
json_dict['gif_duration'] = self.gif_duration
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultMpeg4Gif(JsonSerializable):
|
|
def __init__(self, id, mpeg4_url, thumb_url, mpeg4_width=None, mpeg4_height=None, title=None, caption=None,
|
|
parse_mode=None, reply_markup=None, input_message_content=None, mpeg4_duration=None):
|
|
"""
|
|
Represents a link to a video animation (H.264/MPEG-4 AVC video without sound).
|
|
:param id: Unique identifier for this result, 1-64 bytes
|
|
:param mpeg4_url: A valid URL for the MP4 file. File size must not exceed 1MB
|
|
:param thumb_url: URL of the static thumbnail (jpeg or gif) for the result
|
|
:param mpeg4_width: Video width
|
|
:param mpeg4_height: Video height
|
|
:param title: Title for the result
|
|
:param caption: Caption of the MPEG-4 file to be sent, 0-200 characters
|
|
:param parse_mode: Send Markdown or HTML, if you want Telegram apps to show bold, italic, fixed-width text
|
|
or inline URLs in the media caption.
|
|
:param reply_markup: InlineKeyboardMarkup : Inline keyboard attached to the message
|
|
:param input_message_content: InputMessageContent : Content of the message to be sent instead of the photo
|
|
:return:
|
|
"""
|
|
self.type = 'mpeg4_gif'
|
|
self.id = id
|
|
self.mpeg4_url = mpeg4_url
|
|
self.mpeg4_width = mpeg4_width
|
|
self.mpeg4_height = mpeg4_height
|
|
self.thumb_url = thumb_url
|
|
self.title = title
|
|
self.caption = caption
|
|
self.parse_mode = parse_mode
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.mpeg4_duration = mpeg4_duration
|
|
|
|
def to_json(self):
|
|
json_dict = {'type': self.type, 'id': self.id, 'mpeg4_url': self.mpeg4_url, 'thumb_url': self.thumb_url}
|
|
if self.mpeg4_width:
|
|
json_dict['mpeg4_width'] = self.mpeg4_width
|
|
if self.mpeg4_height:
|
|
json_dict['mpeg4_height'] = self.mpeg4_height
|
|
if self.title:
|
|
json_dict['title'] = self.title
|
|
if self.caption:
|
|
json_dict['caption'] = self.caption
|
|
if self.parse_mode:
|
|
json_dict['parse_mode'] = self.parse_mode
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
if self.mpeg4_duration:
|
|
json_dict['mpeg4_duration '] = self.mpeg4_duration
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultVideo(JsonSerializable):
|
|
def __init__(self, id, video_url, mime_type, thumb_url, title,
|
|
caption=None, parse_mode=None, video_width=None, video_height=None, video_duration=None,
|
|
description=None, reply_markup=None, input_message_content=None):
|
|
"""
|
|
Represents link to a page containing an embedded video player or a video file.
|
|
:param id: Unique identifier for this result, 1-64 bytes
|
|
:param video_url: A valid URL for the embedded video player or video file
|
|
:param mime_type: Mime type of the content of video url, “text/html” or “video/mp4”
|
|
:param thumb_url: URL of the thumbnail (jpeg only) for the video
|
|
:param title: Title for the result
|
|
:param parse_mode: Send Markdown or HTML, if you want Telegram apps to show bold, italic, fixed-width text or
|
|
inline URLs in the media caption.
|
|
:param video_width: Video width
|
|
:param video_height: Video height
|
|
:param video_duration: Video duration in seconds
|
|
:param description: Short description of the result
|
|
:return:
|
|
"""
|
|
self.type = 'video'
|
|
self.id = id
|
|
self.video_url = video_url
|
|
self.mime_type = mime_type
|
|
self.video_width = video_width
|
|
self.video_height = video_height
|
|
self.video_duration = video_duration
|
|
self.thumb_url = thumb_url
|
|
self.title = title
|
|
self.caption = caption
|
|
self.parse_mode = parse_mode
|
|
self.description = description
|
|
self.input_message_content = input_message_content
|
|
self.reply_markup = reply_markup
|
|
|
|
def to_json(self):
|
|
json_dict = {'type': self.type, 'id': self.id, 'video_url': self.video_url, 'mime_type': self.mime_type,
|
|
'thumb_url': self.thumb_url, 'title': self.title}
|
|
if self.video_width:
|
|
json_dict['video_width'] = self.video_width
|
|
if self.video_height:
|
|
json_dict['video_height'] = self.video_height
|
|
if self.video_duration:
|
|
json_dict['video_duration'] = self.video_duration
|
|
if self.description:
|
|
json_dict['description'] = self.description
|
|
if self.caption:
|
|
json_dict['caption'] = self.caption
|
|
if self.parse_mode:
|
|
json_dict['parse_mode'] = self.parse_mode
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultAudio(JsonSerializable):
|
|
def __init__(self, id, audio_url, title, caption=None, parse_mode=None, performer=None, audio_duration=None,
|
|
reply_markup=None, input_message_content=None):
|
|
self.type = 'audio'
|
|
self.id = id
|
|
self.audio_url = audio_url
|
|
self.title = title
|
|
self.caption = caption
|
|
self.parse_mode = parse_mode
|
|
self.performer = performer
|
|
self.audio_duration = audio_duration
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
|
|
def to_json(self):
|
|
json_dict = {'type': self.type, 'id': self.id, 'audio_url': self.audio_url, 'title': self.title}
|
|
if self.caption:
|
|
json_dict['caption'] = self.caption
|
|
if self.parse_mode:
|
|
json_dict['parse_mode'] = self.parse_mode
|
|
if self.performer:
|
|
json_dict['performer'] = self.performer
|
|
if self.audio_duration:
|
|
json_dict['audio_duration'] = self.audio_duration
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultVoice(JsonSerializable):
|
|
def __init__(self, id, voice_url, title, caption=None, parse_mode=None, performer=None, voice_duration=None,
|
|
reply_markup=None, input_message_content=None):
|
|
self.type = 'voice'
|
|
self.id = id
|
|
self.voice_url = voice_url
|
|
self.title = title
|
|
self.caption = caption
|
|
self.parse_mode = parse_mode
|
|
self.performer = performer
|
|
self.voice_duration = voice_duration
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
|
|
def to_json(self):
|
|
json_dict = {'type': self.type, 'id': self.id, 'voice_url': self.voice_url, 'title': self.title}
|
|
if self.caption:
|
|
json_dict['caption'] = self.caption
|
|
if self.parse_mode:
|
|
json_dict['parse_mode'] = self.parse_mode
|
|
if self.performer:
|
|
json_dict['performer'] = self.performer
|
|
if self.voice_duration:
|
|
json_dict['voice_duration'] = self.voice_duration
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultDocument(JsonSerializable):
|
|
def __init__(self, id, title, document_url, mime_type, caption=None, parse_mode=None, description=None,
|
|
reply_markup=None, input_message_content=None, thumb_url=None, thumb_width=None, thumb_height=None):
|
|
self.type = 'document'
|
|
self.id = id
|
|
self.title = title
|
|
self.document_url = document_url
|
|
self.mime_type = mime_type
|
|
self.caption = caption
|
|
self.parse_mode = parse_mode
|
|
self.description = description
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.thumb_url = thumb_url
|
|
self.thumb_width = thumb_width
|
|
self.thumb_height = thumb_height
|
|
|
|
def to_json(self):
|
|
json_dict = {'type': self.type, 'id': self.id, 'title': self.title, 'document_url': self.document_url,
|
|
'mime_type': self.mime_type}
|
|
if self.caption:
|
|
json_dict['caption'] = self.caption
|
|
if self.parse_mode:
|
|
json_dict['parse_mode'] = self.parse_mode
|
|
if self.description:
|
|
json_dict['description'] = self.description
|
|
if self.thumb_url:
|
|
json_dict['thumb_url'] = self.thumb_url
|
|
if self.thumb_width:
|
|
json_dict['thumb_width'] = self.thumb_width
|
|
if self.thumb_height:
|
|
json_dict['thumb_height'] = self.thumb_height
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultLocation(JsonSerializable):
|
|
def __init__(self, id, title, latitude, longitude, live_period=None, reply_markup=None,
|
|
input_message_content=None, thumb_url=None, thumb_width=None, thumb_height=None):
|
|
self.type = 'location'
|
|
self.id = id
|
|
self.title = title
|
|
self.latitude = latitude
|
|
self.longitude = longitude
|
|
self.live_period = live_period
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.thumb_url = thumb_url
|
|
self.thumb_width = thumb_width
|
|
self.thumb_height = thumb_height
|
|
|
|
def to_json(self):
|
|
json_dict = {'type': self.type, 'id': self.id, 'latitude': self.latitude, 'longitude': self.longitude,
|
|
'title': self.title}
|
|
if self.live_period:
|
|
json_dict['live_period'] = self.live_period
|
|
if self.thumb_url:
|
|
json_dict['thumb_url'] = self.thumb_url
|
|
if self.thumb_width:
|
|
json_dict['thumb_width'] = self.thumb_width
|
|
if self.thumb_height:
|
|
json_dict['thumb_height'] = self.thumb_height
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultVenue(JsonSerializable):
|
|
def __init__(self, id, title, latitude, longitude, address, foursquare_id=None, foursquare_type=None,
|
|
reply_markup=None, input_message_content=None, thumb_url=None, thumb_width=None, thumb_height=None):
|
|
self.type = 'venue'
|
|
self.id = id
|
|
self.title = title
|
|
self.latitude = latitude
|
|
self.longitude = longitude
|
|
self.address = address
|
|
self.foursquare_id = foursquare_id
|
|
self.foursquare_type = foursquare_type
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.thumb_url = thumb_url
|
|
self.thumb_width = thumb_width
|
|
self.thumb_height = thumb_height
|
|
|
|
def to_json(self):
|
|
json_dict = {'type': self.type, 'id': self.id, 'title': self.title, 'latitude': self.latitude,
|
|
'longitude': self.longitude, 'address': self.address}
|
|
if self.foursquare_id:
|
|
json_dict['foursquare_id'] = self.foursquare_id
|
|
if self.foursquare_type:
|
|
json_dict['foursquare_type'] = self.foursquare_type
|
|
if self.thumb_url:
|
|
json_dict['thumb_url'] = self.thumb_url
|
|
if self.thumb_width:
|
|
json_dict['thumb_width'] = self.thumb_width
|
|
if self.thumb_height:
|
|
json_dict['thumb_height'] = self.thumb_height
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultContact(JsonSerializable):
|
|
def __init__(self, id, phone_number, first_name, last_name=None, vcard=None,
|
|
reply_markup=None, input_message_content=None,
|
|
thumb_url=None, thumb_width=None, thumb_height=None):
|
|
self.type = 'contact'
|
|
self.id = id
|
|
self.phone_number = phone_number
|
|
self.first_name = first_name
|
|
self.last_name = last_name
|
|
self.vcard = vcard
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.thumb_url = thumb_url
|
|
self.thumb_width = thumb_width
|
|
self.thumb_height = thumb_height
|
|
|
|
def to_json(self):
|
|
json_dict = {'type': self.type, 'id': self.id, 'phone_number': self.phone_number, 'first_name': self.first_name}
|
|
if self.last_name:
|
|
json_dict['last_name'] = self.last_name
|
|
if self.vcard:
|
|
json_dict['vcard'] = self.vcard
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
if self.thumb_url:
|
|
json_dict['thumb_url'] = self.thumb_url
|
|
if self.thumb_width:
|
|
json_dict['thumb_width'] = self.thumb_width
|
|
if self.thumb_height:
|
|
json_dict['thumb_height'] = self.thumb_height
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class BaseInlineQueryResultCached(JsonSerializable):
|
|
def __init__(self):
|
|
self.type = None
|
|
self.id = None
|
|
self.title = None
|
|
self.description = None
|
|
self.caption = None
|
|
self.reply_markup = None
|
|
self.input_message_content = None
|
|
self.parse_mode = None
|
|
self.payload_dic = {}
|
|
|
|
def to_json(self):
|
|
json_dict = self.payload_dic
|
|
json_dict['type'] = self.type
|
|
json_dict['id'] = self.id
|
|
if self.title:
|
|
json_dict['title'] = self.title
|
|
if self.description:
|
|
json_dict['description'] = self.description
|
|
if self.caption:
|
|
json_dict['caption'] = self.caption
|
|
if self.reply_markup:
|
|
json_dict['reply_markup'] = self.reply_markup.to_dict()
|
|
if self.input_message_content:
|
|
json_dict['input_message_content'] = self.input_message_content.to_dict()
|
|
if self.parse_mode:
|
|
json_dict['parse_mode'] = self.parse_mode
|
|
return json.dumps(json_dict)
|
|
|
|
|
|
class InlineQueryResultCachedPhoto(BaseInlineQueryResultCached):
|
|
def __init__(self, id, photo_file_id, title=None, description=None, caption=None, parse_mode=None,
|
|
reply_markup=None, input_message_content=None):
|
|
BaseInlineQueryResultCached.__init__(self)
|
|
self.type = 'photo'
|
|
self.id = id
|
|
self.photo_file_id = photo_file_id
|
|
self.title = title
|
|
self.description = description
|
|
self.caption = caption
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.parse_mode = parse_mode
|
|
self.payload_dic['photo_file_id'] = photo_file_id
|
|
|
|
|
|
class InlineQueryResultCachedGif(BaseInlineQueryResultCached):
|
|
def __init__(self, id, gif_file_id, title=None, description=None, caption=None, parse_mode=None, reply_markup=None,
|
|
input_message_content=None):
|
|
BaseInlineQueryResultCached.__init__(self)
|
|
self.type = 'gif'
|
|
self.id = id
|
|
self.gif_file_id = gif_file_id
|
|
self.title = title
|
|
self.description = description
|
|
self.caption = caption
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.parse_mode = parse_mode
|
|
self.payload_dic['gif_file_id'] = gif_file_id
|
|
|
|
|
|
class InlineQueryResultCachedMpeg4Gif(BaseInlineQueryResultCached):
|
|
def __init__(self, id, mpeg4_file_id, title=None, description=None, caption=None, parse_mode=None,
|
|
reply_markup=None, input_message_content=None):
|
|
BaseInlineQueryResultCached.__init__(self)
|
|
self.type = 'mpeg4_gif'
|
|
self.id = id
|
|
self.mpeg4_file_id = mpeg4_file_id
|
|
self.title = title
|
|
self.description = description
|
|
self.caption = caption
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.parse_mode = parse_mode
|
|
self.payload_dic['mpeg4_file_id'] = mpeg4_file_id
|
|
|
|
|
|
class InlineQueryResultCachedSticker(BaseInlineQueryResultCached):
|
|
def __init__(self, id, sticker_file_id, reply_markup=None, input_message_content=None):
|
|
BaseInlineQueryResultCached.__init__(self)
|
|
self.type = 'sticker'
|
|
self.id = id
|
|
self.sticker_file_id = sticker_file_id
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.payload_dic['sticker_file_id'] = sticker_file_id
|
|
|
|
|
|
class InlineQueryResultCachedDocument(BaseInlineQueryResultCached):
|
|
def __init__(self, id, document_file_id, title, description=None, caption=None, parse_mode=None, reply_markup=None,
|
|
input_message_content=None):
|
|
BaseInlineQueryResultCached.__init__(self)
|
|
self.type = 'document'
|
|
self.id = id
|
|
self.document_file_id = document_file_id
|
|
self.title = title
|
|
self.description = description
|
|
self.caption = caption
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.parse_mode = parse_mode
|
|
self.payload_dic['document_file_id'] = document_file_id
|
|
|
|
|
|
class InlineQueryResultCachedVideo(BaseInlineQueryResultCached):
|
|
def __init__(self, id, video_file_id, title, description=None, caption=None, parse_mode=None, reply_markup=None,
|
|
input_message_content=None):
|
|
BaseInlineQueryResultCached.__init__(self)
|
|
self.type = 'video'
|
|
self.id = id
|
|
self.video_file_id = video_file_id
|
|
self.title = title
|
|
self.description = description
|
|
self.caption = caption
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.parse_mode = parse_mode
|
|
self.payload_dic['video_file_id'] = video_file_id
|
|
|
|
|
|
class InlineQueryResultCachedVoice(BaseInlineQueryResultCached):
|
|
def __init__(self, id, voice_file_id, title, caption=None, parse_mode=None, reply_markup=None,
|
|
input_message_content=None):
|
|
BaseInlineQueryResultCached.__init__(self)
|
|
self.type = 'voice'
|
|
self.id = id
|
|
self.voice_file_id = voice_file_id
|
|
self.title = title
|
|
self.caption = caption
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.parse_mode = parse_mode
|
|
self.payload_dic['voice_file_id'] = voice_file_id
|
|
|
|
|
|
class InlineQueryResultCachedAudio(BaseInlineQueryResultCached):
|
|
def __init__(self, id, audio_file_id, caption=None, parse_mode=None, reply_markup=None, input_message_content=None):
|
|
BaseInlineQueryResultCached.__init__(self)
|
|
self.type = 'audio'
|
|
self.id = id
|
|
self.audio_file_id = audio_file_id
|
|
self.caption = caption
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
self.parse_mode = parse_mode
|
|
self.payload_dic['audio_file_id'] = audio_file_id
|
|
|
|
|
|
# Games
|
|
|
|
class InlineQueryResultGame(JsonSerializable):
|
|
def __init__(self, id, game_short_name, reply_markup=None):
|
|
self.type = 'game'
|
|
self.id = id
|
|
self.game_short_name = game_short_name
|
|
self.reply_markup = reply_markup
|
|
|
|
def to_json(self):
|
|
json_dic = {'type': self.type, 'id': self.id, 'game_short_name': self.game_short_name}
|
|
if self.reply_markup:
|
|
json_dic['reply_markup'] = self.reply_markup.to_dict()
|
|
return json.dumps(json_dic)
|
|
|
|
|
|
class Game(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
title = obj['title']
|
|
description = obj['description']
|
|
photo = Game.parse_photo(obj['photo'])
|
|
text = obj.get('text')
|
|
if 'text_entities' in obj:
|
|
text_entities = Game.parse_entities(obj['text_entities'])
|
|
else:
|
|
text_entities = None
|
|
animation = Animation.de_json(obj.get('animation'))
|
|
return cls(title, description, photo, text, text_entities, animation)
|
|
|
|
@classmethod
|
|
def parse_photo(cls, photo_size_array):
|
|
ret = []
|
|
for ps in photo_size_array:
|
|
ret.append(PhotoSize.de_json(ps))
|
|
return ret
|
|
|
|
@classmethod
|
|
def parse_entities(cls, message_entity_array):
|
|
ret = []
|
|
for me in message_entity_array:
|
|
ret.append(MessageEntity.de_json(me))
|
|
return ret
|
|
|
|
def __init__(self, title, description, photo, text=None, text_entities=None, animation=None):
|
|
self.title = title
|
|
self.description = description
|
|
self.photo = photo
|
|
self.text = text
|
|
self.text_entities = text_entities
|
|
self.animation = animation
|
|
|
|
|
|
class Animation(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
file_id = obj['file_id']
|
|
file_unique_id = obj['file_unique_id']
|
|
thumb = PhotoSize.de_json(obj.get('thumb'))
|
|
file_name = obj.get('file_name')
|
|
mime_type = obj.get('mime_type')
|
|
file_size = obj.get('file_size')
|
|
return cls(file_id, file_unique_id, thumb, file_name, mime_type, file_size)
|
|
|
|
def __init__(self, file_id, file_unique_id, thumb=None, file_name=None, mime_type=None, file_size=None):
|
|
self.file_id = file_id
|
|
self.file_unique_id = file_unique_id
|
|
self.thumb = thumb
|
|
self.file_name = file_name
|
|
self.mime_type = mime_type
|
|
self.file_size = file_size
|
|
|
|
|
|
class GameHighScore(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
position = obj['position']
|
|
user = User.de_json(obj['user'])
|
|
score = obj['score']
|
|
return cls(position, user, score)
|
|
|
|
def __init__(self, position, user, score):
|
|
self.position = position
|
|
self.user = user
|
|
self.score = score
|
|
|
|
|
|
# Payments
|
|
|
|
class LabeledPrice(JsonSerializable):
|
|
def __init__(self, label, amount):
|
|
self.label = label
|
|
self.amount = amount
|
|
|
|
def to_json(self):
|
|
return json.dumps({
|
|
'label': self.label, 'amount': self.amount
|
|
})
|
|
|
|
|
|
class Invoice(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
title = obj['title']
|
|
description = obj['description']
|
|
start_parameter = obj['start_parameter']
|
|
currency = obj['currency']
|
|
total_amount = obj['total_amount']
|
|
return cls(title, description, start_parameter, currency, total_amount)
|
|
|
|
def __init__(self, title, description, start_parameter, currency, total_amount):
|
|
self.title = title
|
|
self.description = description
|
|
self.start_parameter = start_parameter
|
|
self.currency = currency
|
|
self.total_amount = total_amount
|
|
|
|
|
|
class ShippingAddress(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
country_code = obj['country_code']
|
|
state = obj['state']
|
|
city = obj['city']
|
|
street_line1 = obj['street_line1']
|
|
street_line2 = obj['street_line2']
|
|
post_code = obj['post_code']
|
|
return cls(country_code, state, city, street_line1, street_line2, post_code)
|
|
|
|
def __init__(self, country_code, state, city, street_line1, street_line2, post_code):
|
|
self.country_code = country_code
|
|
self.state = state
|
|
self.city = city
|
|
self.street_line1 = street_line1
|
|
self.street_line2 = street_line2
|
|
self.post_code = post_code
|
|
|
|
|
|
class OrderInfo(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
name = obj.get('name')
|
|
phone_number = obj.get('phone_number')
|
|
email = obj.get('email')
|
|
shipping_address = ShippingAddress.de_json(obj.get('shipping_address'))
|
|
return cls(name, phone_number, email, shipping_address)
|
|
|
|
def __init__(self, name, phone_number, email, shipping_address):
|
|
self.name = name
|
|
self.phone_number = phone_number
|
|
self.email = email
|
|
self.shipping_address = shipping_address
|
|
|
|
|
|
class ShippingOption(JsonSerializable):
|
|
def __init__(self, id, title):
|
|
self.id = id
|
|
self.title = title
|
|
self.prices = []
|
|
|
|
def add_price(self, *args):
|
|
"""
|
|
Add LabeledPrice to ShippingOption
|
|
:param args: LabeledPrices
|
|
"""
|
|
for price in args:
|
|
self.prices.append(price)
|
|
return self
|
|
|
|
def to_json(self):
|
|
price_list = []
|
|
for p in self.prices:
|
|
price_list.append(p.to_dict())
|
|
json_dict = json.dumps({'id': self.id, 'title': self.title, 'prices': price_list})
|
|
return json_dict
|
|
|
|
|
|
class SuccessfulPayment(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
currency = obj['currency']
|
|
total_amount = obj['total_amount']
|
|
invoice_payload = obj['invoice_payload']
|
|
shipping_option_id = obj.get('shipping_option_id')
|
|
order_info = OrderInfo.de_json(obj.get('order_info'))
|
|
telegram_payment_charge_id = obj['telegram_payment_charge_id']
|
|
provider_payment_charge_id = obj['provider_payment_charge_id']
|
|
return cls(currency, total_amount, invoice_payload, shipping_option_id, order_info,
|
|
telegram_payment_charge_id, provider_payment_charge_id)
|
|
|
|
def __init__(self, currency, total_amount, invoice_payload, shipping_option_id, order_info,
|
|
telegram_payment_charge_id, provider_payment_charge_id):
|
|
self.currency = currency
|
|
self.total_amount = total_amount
|
|
self.invoice_payload = invoice_payload
|
|
self.shipping_option_id = shipping_option_id
|
|
self.order_info = order_info
|
|
self.telegram_payment_charge_id = telegram_payment_charge_id
|
|
self.provider_payment_charge_id = provider_payment_charge_id
|
|
|
|
|
|
class ShippingQuery(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
id = obj['id']
|
|
from_user = User.de_json(obj['from'])
|
|
invoice_payload = obj['invoice_payload']
|
|
shipping_address = ShippingAddress.de_json(obj['shipping_address'])
|
|
return cls(id, from_user, invoice_payload, shipping_address)
|
|
|
|
def __init__(self, id, from_user, invoice_payload, shipping_address):
|
|
self.id = id
|
|
self.from_user = from_user
|
|
self.invoice_payload = invoice_payload
|
|
self.shipping_address = shipping_address
|
|
|
|
|
|
class PreCheckoutQuery(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
id = obj['id']
|
|
from_user = User.de_json(obj['from'])
|
|
currency = obj['currency']
|
|
total_amount = obj['total_amount']
|
|
invoice_payload = obj['invoice_payload']
|
|
shipping_option_id = obj.get('shipping_option_id')
|
|
order_info = OrderInfo.de_json(obj.get('order_info'))
|
|
return cls(id, from_user, currency, total_amount, invoice_payload, shipping_option_id, order_info)
|
|
|
|
def __init__(self, id, from_user, currency, total_amount, invoice_payload, shipping_option_id, order_info):
|
|
self.id = id
|
|
self.from_user = from_user
|
|
self.currency = currency
|
|
self.total_amount = total_amount
|
|
self.invoice_payload = invoice_payload
|
|
self.shipping_option_id = shipping_option_id
|
|
self.order_info = order_info
|
|
|
|
|
|
# Stickers
|
|
|
|
class StickerSet(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
name = obj['name']
|
|
title = obj['title']
|
|
contains_masks = obj['contains_masks']
|
|
stickers = []
|
|
for s in obj['stickers']:
|
|
stickers.append(Sticker.de_json(s))
|
|
return cls(name, title, contains_masks, stickers)
|
|
|
|
def __init__(self, name, title, contains_masks, stickers):
|
|
self.stickers = stickers
|
|
self.contains_masks = contains_masks
|
|
self.title = title
|
|
self.name = name
|
|
|
|
|
|
class Sticker(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
file_id = obj['file_id']
|
|
file_unique_id = obj['file_unique_id']
|
|
width = obj['width']
|
|
height = obj['height']
|
|
is_animated = obj['is_animated']
|
|
thumb = PhotoSize.de_json(obj.get('thumb'))
|
|
emoji = obj.get('emoji')
|
|
set_name = obj.get('set_name')
|
|
mask_position = MaskPosition.de_json(obj.get('mask_position'))
|
|
file_size = obj.get('file_size')
|
|
return cls(file_id, file_unique_id, width, height, thumb, emoji, set_name, mask_position, file_size, is_animated)
|
|
|
|
def __init__(self, file_id, file_unique_id, width, height, thumb, emoji, set_name, mask_position, file_size, is_animated):
|
|
self.file_id = file_id
|
|
self.file_unique_id = file_unique_id
|
|
self.width = width
|
|
self.height = height
|
|
self.thumb = thumb
|
|
self.emoji = emoji
|
|
self.set_name = set_name
|
|
self.mask_position = mask_position
|
|
self.file_size = file_size
|
|
self.is_animated = is_animated
|
|
|
|
|
|
class MaskPosition(Dictionaryable, JsonDeserializable, JsonSerializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
point = obj['point']
|
|
x_shift = obj['x_shift']
|
|
y_shift = obj['y_shift']
|
|
scale = obj['scale']
|
|
return cls(point, x_shift, y_shift, scale)
|
|
|
|
def __init__(self, point, x_shift, y_shift, scale):
|
|
self.point = point
|
|
self.x_shift = x_shift
|
|
self.y_shift = y_shift
|
|
self.scale = scale
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
return {'point': self.point, 'x_shift': self.x_shift, 'y_shift': self.y_shift, 'scale': self.scale}
|
|
|
|
|
|
# InputMedia
|
|
|
|
class InputMedia(Dictionaryable, JsonSerializable):
|
|
def __init__(self, type, media, caption=None, parse_mode=None):
|
|
self.type = type
|
|
self.media = media
|
|
self.caption = caption
|
|
self.parse_mode = parse_mode
|
|
|
|
if util.is_string(self.media):
|
|
self._media_name = ''
|
|
self._media_dic = self.media
|
|
else:
|
|
self._media_name = util.generate_random_token()
|
|
self._media_dic = 'attach://{0}'.format(self._media_name)
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
json_dict = {'type': self.type, 'media': self._media_dic}
|
|
if self.caption:
|
|
json_dict['caption'] = self.caption
|
|
if self.parse_mode:
|
|
json_dict['parse_mode'] = self.parse_mode
|
|
return json_dict
|
|
|
|
def convert_input_media(self):
|
|
if util.is_string(self.media):
|
|
return self.to_json(), None
|
|
|
|
return self.to_json(), {self._media_name: self.media}
|
|
|
|
|
|
class InputMediaPhoto(InputMedia):
|
|
def __init__(self, media, caption=None, parse_mode=None):
|
|
if util.is_pil_image(media):
|
|
media = util.pil_image_to_file(media)
|
|
|
|
super(InputMediaPhoto, self).__init__(type="photo", media=media, caption=caption, parse_mode=parse_mode)
|
|
|
|
def to_dict(self):
|
|
return super(InputMediaPhoto, self).to_dict()
|
|
|
|
|
|
class InputMediaVideo(InputMedia):
|
|
def __init__(self, media, thumb=None, caption=None, parse_mode=None, width=None, height=None, duration=None,
|
|
supports_streaming=None):
|
|
super(InputMediaVideo, self).__init__(type="video", media=media, caption=caption, parse_mode=parse_mode)
|
|
self.thumb = thumb
|
|
self.width = width
|
|
self.height = height
|
|
self.duration = duration
|
|
self.supports_streaming = supports_streaming
|
|
|
|
def to_dict(self):
|
|
ret = super(InputMediaVideo, self).to_dict()
|
|
if self.thumb:
|
|
ret['thumb'] = self.thumb
|
|
if self.width:
|
|
ret['width'] = self.width
|
|
if self.height:
|
|
ret['height'] = self.height
|
|
if self.duration:
|
|
ret['duration'] = self.duration
|
|
if self.supports_streaming:
|
|
ret['supports_streaming'] = self.supports_streaming
|
|
return ret
|
|
|
|
|
|
class InputMediaAnimation(InputMedia):
|
|
def __init__(self, media, thumb=None, caption=None, parse_mode=None, width=None, height=None, duration=None):
|
|
super(InputMediaAnimation, self).__init__(type="animation", media=media, caption=caption, parse_mode=parse_mode)
|
|
self.thumb = thumb
|
|
self.width = width
|
|
self.height = height
|
|
self.duration = duration
|
|
|
|
def to_dict(self):
|
|
ret = super(InputMediaAnimation, self).to_dict()
|
|
if self.thumb:
|
|
ret['thumb'] = self.thumb
|
|
if self.width:
|
|
ret['width'] = self.width
|
|
if self.height:
|
|
ret['height'] = self.height
|
|
if self.duration:
|
|
ret['duration'] = self.duration
|
|
return ret
|
|
|
|
|
|
class InputMediaAudio(InputMedia):
|
|
def __init__(self, media, thumb=None, caption=None, parse_mode=None, duration=None, performer=None, title=None):
|
|
super(InputMediaAudio, self).__init__(type="audio", media=media, caption=caption, parse_mode=parse_mode)
|
|
self.thumb = thumb
|
|
self.duration = duration
|
|
self.performer = performer
|
|
self.title = title
|
|
|
|
def to_dict(self):
|
|
ret = super(InputMediaAudio, self).to_dict()
|
|
if self.thumb:
|
|
ret['thumb'] = self.thumb
|
|
if self.duration:
|
|
ret['duration'] = self.duration
|
|
if self.performer:
|
|
ret['performer'] = self.performer
|
|
if self.title:
|
|
ret['title'] = self.title
|
|
return ret
|
|
|
|
|
|
class InputMediaDocument(InputMedia):
|
|
def __init__(self, media, thumb=None, caption=None, parse_mode=None):
|
|
super(InputMediaDocument, self).__init__(type="document", media=media, caption=caption, parse_mode=parse_mode)
|
|
self.thumb = thumb
|
|
|
|
def to_dict(self):
|
|
ret = super(InputMediaDocument, self).to_dict()
|
|
if self.thumb:
|
|
ret['thumb'] = self.thumb
|
|
return ret
|
|
|
|
|
|
class PollOption(JsonSerializable, JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
text = obj['text']
|
|
voter_count = int(obj['voter_count'])
|
|
return cls(text, voter_count)
|
|
|
|
def __init__(self, text, voter_count = 0):
|
|
self.text = text
|
|
self.voter_count = voter_count
|
|
|
|
def to_json(self):
|
|
# send_poll Option is a simple string: https://core.telegram.org/bots/api#sendpoll
|
|
return json.dumps(self.text)
|
|
|
|
|
|
class Poll(JsonDeserializable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
poll_id = obj['id']
|
|
question = obj['question']
|
|
options = []
|
|
for opt in obj['options']:
|
|
options.append(PollOption.de_json(opt))
|
|
total_voter_count = obj['total_voter_count']
|
|
is_closed = obj['is_closed']
|
|
is_anonymous = obj['is_anonymous']
|
|
poll_type = obj['type']
|
|
allows_multiple_answers = obj['allows_multiple_answers']
|
|
correct_option_id = obj.get('correct_option_id')
|
|
explanation = obj.get('explanation')
|
|
if 'explanation_entities' in obj:
|
|
explanation_entities = Message.parse_entities(obj['explanation_entities'])
|
|
else:
|
|
explanation_entities = None
|
|
open_period = obj.get('open_period')
|
|
close_date = obj.get('close_date')
|
|
return cls(
|
|
question, options,
|
|
poll_id, total_voter_count, is_closed, is_anonymous, poll_type,
|
|
allows_multiple_answers, correct_option_id, explanation, explanation_entities,
|
|
open_period, close_date)
|
|
|
|
def __init__(
|
|
self,
|
|
question, options,
|
|
poll_id=None, total_voter_count=None, is_closed=None, is_anonymous=None, poll_type=None,
|
|
allows_multiple_answers=None, correct_option_id=None, explanation=None, explanation_entities=None,
|
|
open_period=None, close_date=None):
|
|
self.id = poll_id
|
|
self.question = question
|
|
self.options = options
|
|
self.total_voter_count = total_voter_count
|
|
self.is_closed = is_closed
|
|
self.is_anonymous = is_anonymous
|
|
self.type = poll_type
|
|
self.allows_multiple_answers = allows_multiple_answers
|
|
self.correct_option_id = correct_option_id
|
|
self.explanation = explanation
|
|
self.explanation_entities = explanation_entities # Default state of entities is None. if (explanation_entities is not None) else []
|
|
self.open_period = open_period
|
|
self.close_date = close_date
|
|
|
|
def add(self, option):
|
|
if type(option) is PollOption:
|
|
self.options.append(option)
|
|
else:
|
|
self.options.append(PollOption(option))
|
|
|
|
|
|
class PollAnswer(JsonSerializable, JsonDeserializable, Dictionaryable):
|
|
@classmethod
|
|
def de_json(cls, json_string):
|
|
if (json_string is None): return None
|
|
obj = cls.check_json(json_string)
|
|
poll_id = obj['poll_id']
|
|
user = User.de_json(obj['user'])
|
|
options_ids = obj['option_ids']
|
|
return cls(poll_id, user, options_ids)
|
|
|
|
def __init__(self, poll_id, user, options_ids):
|
|
self.poll_id = poll_id
|
|
self.user = user
|
|
self.options_ids = options_ids
|
|
|
|
def to_json(self):
|
|
return json.dumps(self.to_dict())
|
|
|
|
def to_dict(self):
|
|
return {'poll_id': self.poll_id,
|
|
'user': self.user.to_dict(),
|
|
'options_ids': self.options_ids}
|