147 lines
5.5 KiB
Python
147 lines
5.5 KiB
Python
# Pyrogram - Telegram MTProto API Client Library for Python
|
|
# Copyright (C) 2017-present Dan httpsgithub.comdelivrance
|
|
#
|
|
# This file is part of Pyrogram.
|
|
#
|
|
# Pyrogram is free software you can redistribute it andor modify
|
|
# it under the terms of the GNU Lesser General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Pyrogram is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from typing import Optional, List, Union
|
|
|
|
import pyrogram
|
|
from pyrogram import raw, types, utils
|
|
from pyrogram.types import InlineQueryResult
|
|
from pyrogram.file_id import FileId, FileType, PHOTO_TYPES, DOCUMENT_TYPES
|
|
|
|
|
|
def get_input_file_from_file_id(
|
|
file_id: str,
|
|
expected_file_type: FileType = None
|
|
) -> Union["raw.types.InputPhoto", "raw.types.InputDocument"]:
|
|
try:
|
|
decoded = FileId.decode(file_id)
|
|
except Exception:
|
|
raise ValueError(f'Failed to decode "{file_id}". The value does not represent an existing local file, '
|
|
f'HTTP URL, or valid file id.')
|
|
|
|
file_type = decoded.file_type
|
|
|
|
if expected_file_type is not None and file_type != expected_file_type:
|
|
raise ValueError(f'Expected: "{expected_file_type}", got "{file_type}" file_id instead')
|
|
|
|
if file_type in (FileType.THUMBNAIL, FileType.CHAT_PHOTO):
|
|
raise ValueError(f"This file_id can only be used for download: {file_id}")
|
|
|
|
if file_type in PHOTO_TYPES:
|
|
return raw.types.InputPhoto(
|
|
id=decoded.media_id,
|
|
access_hash=decoded.access_hash,
|
|
file_reference=decoded.file_reference
|
|
)
|
|
|
|
if file_type in DOCUMENT_TYPES:
|
|
return raw.types.InputDocument(
|
|
id=decoded.media_id,
|
|
access_hash=decoded.access_hash,
|
|
file_reference=decoded.file_reference
|
|
)
|
|
|
|
raise ValueError(f"Unknown file id: {file_id}")
|
|
|
|
|
|
class InlineQueryResultCachedDocument(InlineQueryResult):
|
|
"""Link to a file stored on the Telegram servers.
|
|
|
|
By default, this file will be sent by the user with an optional caption.
|
|
Alternatively, you can use input_message_content to send a message with the specified content instead of the file.
|
|
|
|
Parameters:
|
|
title (``str``):
|
|
Title for the result.
|
|
|
|
file_id (``str``):
|
|
Pass a file_id as string to send a media that exists on the Telegram servers.
|
|
|
|
id (``str``, *optional*):
|
|
Unique identifier for this result, 1-64 bytes.
|
|
Defaults to a randomly generated UUID4.
|
|
|
|
description (``str``, *optional*):
|
|
Short description of the result.
|
|
|
|
caption (``str``, *optional*):
|
|
Caption of the document to be sent, 0-1024 characters.
|
|
|
|
parse_mode (``str``, *optional*):
|
|
By default, texts are parsed using both Markdown and HTML styles.
|
|
You can combine both syntaxes together.
|
|
Pass "markdown" or "md" to enable Markdown-style parsing only.
|
|
Pass "html" to enable HTML-style parsing only.
|
|
Pass None to completely disable style parsing.
|
|
caption_entities (List of :obj:`~pyrogram.types.MessageEntity`):
|
|
List of special entities that appear in the caption, which can be specified instead of *parse_mode*.
|
|
|
|
reply_markup (:obj:`~pyrogram.types.InlineKeyboardMarkup`, *optional*):
|
|
Inline keyboard attached to the message.
|
|
|
|
input_message_content (:obj:`~pyrogram.types.InputMessageContent`):
|
|
Content of the message to be sent.
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
title: str,
|
|
file_id: str,
|
|
id: str = None,
|
|
description: str = None,
|
|
caption: str = "",
|
|
parse_mode: Optional[str] = object,
|
|
caption_entities: List["types.MessageEntity"] = None,
|
|
reply_markup: "types.InlineKeyboardMarkup" = None,
|
|
input_message_content: "types.InputMessageContent" = None
|
|
):
|
|
super().__init__("file", id, input_message_content, reply_markup)
|
|
|
|
self.file_id = file_id
|
|
self.title = title
|
|
self.description = description
|
|
self.caption = caption
|
|
self.caption_entities = caption_entities
|
|
self.parse_mode = parse_mode
|
|
self.reply_markup = reply_markup
|
|
self.input_message_content = input_message_content
|
|
|
|
async def write(self, client: "pyrogram.Client"):
|
|
document = get_input_file_from_file_id(self.file_id)
|
|
|
|
message, entities = (await utils.parse_text_entities(
|
|
client, self.caption, self.parse_mode, self.caption_entities
|
|
)).values()
|
|
|
|
return raw.types.InputBotInlineResultDocument(
|
|
id=self.id,
|
|
type=self.type,
|
|
title=self.title,
|
|
description=self.description,
|
|
document=document,
|
|
send_message=(
|
|
await self.input_message_content.write(client, self.reply_markup)
|
|
if self.input_message_content
|
|
else raw.types.InputBotInlineMessageMediaAuto(
|
|
reply_markup=await self.reply_markup.write(client) if self.reply_markup else None,
|
|
message=message,
|
|
entities=entities
|
|
)
|
|
)
|
|
)
|