3647 lines
138 KiB
Python
3647 lines
138 KiB
Python
# Pyrogram - Telegram MTProto API Client Library for Python
|
||
# Copyright (C) 2017-2018 Dan Tès <https://github.com/delivrance>
|
||
#
|
||
# This file is part of Pyrogram.
|
||
#
|
||
# Pyrogram is free software: you can redistribute it and/or 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/>.
|
||
|
||
import base64
|
||
import binascii
|
||
import json
|
||
import logging
|
||
import math
|
||
import mimetypes
|
||
import os
|
||
import re
|
||
import shutil
|
||
import struct
|
||
import sys
|
||
import tempfile
|
||
import threading
|
||
import time
|
||
from configparser import ConfigParser
|
||
from datetime import datetime
|
||
from hashlib import sha256, md5
|
||
from queue import Queue
|
||
from signal import signal, SIGINT, SIGTERM, SIGABRT
|
||
from threading import Event, Thread
|
||
|
||
import pyrogram
|
||
from pyrogram.api import functions, types
|
||
from pyrogram.api.core import Object
|
||
from pyrogram.api.errors import (
|
||
PhoneMigrate, NetworkMigrate, PhoneNumberInvalid,
|
||
PhoneNumberUnoccupied, PhoneCodeInvalid, PhoneCodeHashEmpty,
|
||
PhoneCodeExpired, PhoneCodeEmpty, SessionPasswordNeeded,
|
||
PasswordHashInvalid, FloodWait, PeerIdInvalid, FilePartMissing,
|
||
ChatAdminRequired, FirstnameInvalid, PhoneNumberBanned,
|
||
VolumeLocNotFound, UserMigrate, FileIdInvalid)
|
||
from pyrogram.crypto import AES
|
||
from pyrogram.session import Auth, Session
|
||
from pyrogram.session.internals import MsgId
|
||
from . import types as pyrogram_types
|
||
from . import utils
|
||
from .dispatcher import Dispatcher
|
||
from .style import Markdown, HTML
|
||
from .syncer import Syncer
|
||
|
||
# Custom format for nice looking log lines
|
||
LOG_FORMAT = "[%(asctime)s.%(msecs)03d] %(filename)s:%(lineno)s %(levelname)s: %(message)s"
|
||
|
||
log = logging.getLogger(__name__)
|
||
|
||
|
||
class Client:
|
||
"""This class represents a Client, the main mean for interacting with Telegram.
|
||
It exposes bot-like methods for an easy access to the API as well as a simple way to
|
||
invoke every single Telegram API method available.
|
||
|
||
Args:
|
||
session_name (``str``):
|
||
Name to uniquely identify a session of either a User or a Bot.
|
||
For Users: pass a string of your choice, e.g.: "my_main_account".
|
||
For Bots: pass your Bot API token, e.g.: "123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11"
|
||
Note: as long as a valid User session file exists, Pyrogram won't ask you again to input your phone number.
|
||
|
||
api_id (``int``, optional):
|
||
The *api_id* part of your Telegram API Key, as integer. E.g.: 12345
|
||
This is an alternative way to pass it if you don't want to use the *config.ini* file.
|
||
|
||
api_hash (``str``, optional):
|
||
The *api_hash* part of your Telegram API Key, as string. E.g.: "0123456789abcdef0123456789abcdef"
|
||
This is an alternative way to pass it if you don't want to use the *config.ini* file.
|
||
|
||
proxy (``dict``, optional):
|
||
Your SOCKS5 Proxy settings as dict,
|
||
e.g.: *dict(hostname="11.22.33.44", port=1080, username="user", password="pass")*.
|
||
*username* and *password* can be omitted if your proxy doesn't require authorization.
|
||
This is an alternative way to setup a proxy if you don't want to use the *config.ini* file.
|
||
|
||
test_mode (``bool``, optional):
|
||
Enable or disable log-in to testing servers. Defaults to False.
|
||
Only applicable for new sessions and will be ignored in case previously
|
||
created sessions are loaded.
|
||
|
||
phone_number (``str``, optional):
|
||
Pass your phone number (with your Country Code prefix included) to avoid
|
||
entering it manually. Only applicable for new sessions.
|
||
|
||
phone_code (``str`` | ``callable``, optional):
|
||
Pass the phone code as string (for test numbers only), or pass a callback function
|
||
which must return the correct phone code as string (e.g., "12345").
|
||
Only applicable for new sessions.
|
||
|
||
password (``str``, optional):
|
||
Pass your Two-Step Verification password (if you have one) to avoid entering it
|
||
manually. Only applicable for new sessions.
|
||
|
||
force_sms (``str``, optional):
|
||
Pass True to force Telegram sending the authorization code via SMS.
|
||
Only applicable for new sessions.
|
||
|
||
first_name (``str``, optional):
|
||
Pass a First Name to avoid entering it manually. It will be used to automatically
|
||
create a new Telegram account in case the phone number you passed is not registered yet.
|
||
Only applicable for new sessions.
|
||
|
||
last_name (``str``, optional):
|
||
Same purpose as *first_name*; pass a Last Name to avoid entering it manually. It can
|
||
be an empty string: "". Only applicable for new sessions.
|
||
|
||
workers (``int``, optional):
|
||
Thread pool size for handling incoming updates. Defaults to 4.
|
||
|
||
workdir (``str``, optional):
|
||
Define a custom working directory. The working directory is the location in your filesystem
|
||
where Pyrogram will store your session files. Defaults to "." (current directory).
|
||
"""
|
||
|
||
INVITE_LINK_RE = re.compile(r"^(?:https?://)?(?:t\.me/joinchat/)([\w-]+)$")
|
||
BOT_TOKEN_RE = re.compile(r"^\d+:[\w-]+$")
|
||
DIALOGS_AT_ONCE = 100
|
||
UPDATES_WORKERS = 1
|
||
DOWNLOAD_WORKERS = 1
|
||
OFFLINE_SLEEP = 300
|
||
|
||
MEDIA_TYPE_ID = {
|
||
0: "thumbnail",
|
||
2: "photo",
|
||
3: "voice",
|
||
4: "video",
|
||
5: "document",
|
||
8: "sticker",
|
||
9: "audio",
|
||
10: "gif",
|
||
13: "video_note"
|
||
}
|
||
|
||
def __init__(self,
|
||
session_name: str,
|
||
api_id: int or str = None,
|
||
api_hash: str = None,
|
||
proxy: dict = None,
|
||
test_mode: bool = False,
|
||
phone_number: str = None,
|
||
phone_code: str or callable = None,
|
||
password: str = None,
|
||
force_sms: bool = False,
|
||
first_name: str = None,
|
||
last_name: str = None,
|
||
workers: int = 4,
|
||
workdir: str = "."):
|
||
self.session_name = session_name
|
||
self.api_id = int(api_id) if api_id else None
|
||
self.api_hash = api_hash
|
||
self.proxy = proxy
|
||
self.test_mode = test_mode
|
||
|
||
self.phone_number = phone_number
|
||
self.password = password
|
||
self.phone_code = phone_code
|
||
self.first_name = first_name
|
||
self.last_name = last_name
|
||
self.force_sms = force_sms
|
||
|
||
self.workers = workers
|
||
self.workdir = workdir
|
||
|
||
self.token = None
|
||
|
||
self.dc_id = None
|
||
self.auth_key = None
|
||
self.user_id = None
|
||
self.date = None
|
||
|
||
self.rnd_id = MsgId
|
||
|
||
self.peers_by_id = {}
|
||
self.peers_by_username = {}
|
||
self.peers_by_phone = {}
|
||
|
||
self.channels_pts = {}
|
||
|
||
self.markdown = Markdown(self.peers_by_id)
|
||
self.html = HTML(self.peers_by_id)
|
||
|
||
self.session = None
|
||
|
||
self.is_started = None
|
||
self.is_idle = None
|
||
|
||
self.updates_queue = Queue()
|
||
self.updates_workers_list = []
|
||
self.download_queue = Queue()
|
||
self.download_workers_list = []
|
||
|
||
self.dispatcher = Dispatcher(self, workers)
|
||
self.update_handler = None
|
||
|
||
def on_message(self, filters=None, group: int = 0):
|
||
"""Use this decorator to automatically register a function for handling
|
||
messages. This does the same thing as :meth:`add_handler` using the
|
||
MessageHandler.
|
||
|
||
Args:
|
||
filters (:obj:`Filters <pyrogram.Filters>`):
|
||
Pass one or more filters to allow only a subset of messages to be passed
|
||
in your function.
|
||
|
||
group (``int``, optional):
|
||
The group identifier, defaults to 0.
|
||
"""
|
||
|
||
def decorator(func):
|
||
self.add_handler(pyrogram.MessageHandler(func, filters), group)
|
||
return func
|
||
|
||
return decorator
|
||
|
||
def on_callback_query(self, filters=None, group: int = 0):
|
||
"""Use this decorator to automatically register a function for handling
|
||
callback queries. This does the same thing as :meth:`add_handler` using the
|
||
CallbackQueryHandler.
|
||
|
||
Args:
|
||
filters (:obj:`Filters <pyrogram.Filters>`):
|
||
Pass one or more filters to allow only a subset of callback queries to be passed
|
||
in your function.
|
||
|
||
group (``int``, optional):
|
||
The group identifier, defaults to 0.
|
||
"""
|
||
|
||
def decorator(func):
|
||
self.add_handler(pyrogram.CallbackQueryHandler(func, filters), group)
|
||
return func
|
||
|
||
return decorator
|
||
|
||
def on_raw_update(self, group: int = 0):
|
||
"""Use this decorator to automatically register a function for handling
|
||
raw updates. This does the same thing as :meth:`add_handler` using the
|
||
RawUpdateHandler.
|
||
|
||
Args:
|
||
group (``int``, optional):
|
||
The group identifier, defaults to 0.
|
||
"""
|
||
|
||
def decorator(func):
|
||
self.add_handler(pyrogram.RawUpdateHandler(func), group)
|
||
return func
|
||
|
||
return decorator
|
||
|
||
def add_handler(self, handler, group: int = 0):
|
||
"""Use this method to register an event handler.
|
||
|
||
You can register multiple handlers, but at most one handler within a group
|
||
will be used for a single event. To handle the same event more than once, register
|
||
your handler using a different group id (lower group id == higher priority).
|
||
|
||
Args:
|
||
handler (``Handler``):
|
||
The handler to be registered.
|
||
|
||
group (``int``, optional):
|
||
The group identifier, defaults to 0.
|
||
"""
|
||
self.dispatcher.add_handler(handler, group)
|
||
|
||
def start(self, debug: bool = False):
|
||
"""Use this method to start the Client after creating it.
|
||
Requires no parameters.
|
||
|
||
Args:
|
||
debug (``bool``, optional):
|
||
Enable or disable debug mode. When enabled, extra logging
|
||
lines will be printed out on your console.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
if self.is_started:
|
||
raise ConnectionError("Client has already been started")
|
||
|
||
logging.basicConfig(
|
||
format=LOG_FORMAT,
|
||
datefmt="%Y-%m-%d %H:%M:%S",
|
||
stream=sys.stdout
|
||
)
|
||
|
||
if debug:
|
||
logging.getLogger().setLevel(logging.INFO)
|
||
else:
|
||
logging.getLogger().setLevel(logging.WARNING)
|
||
|
||
if self.BOT_TOKEN_RE.match(self.session_name):
|
||
self.token = self.session_name
|
||
self.session_name = self.session_name.split(":")[0]
|
||
|
||
self.load_config()
|
||
self.load_session()
|
||
|
||
self.session = Session(
|
||
self.dc_id,
|
||
self.test_mode,
|
||
self.proxy,
|
||
self.auth_key,
|
||
self.api_id,
|
||
client=self
|
||
)
|
||
|
||
self.session.start()
|
||
self.is_started = True
|
||
|
||
if self.user_id is None:
|
||
if self.token is None:
|
||
self.authorize_user()
|
||
else:
|
||
self.authorize_bot()
|
||
|
||
self.save_session()
|
||
|
||
if self.token is None:
|
||
now = time.time()
|
||
|
||
if abs(now - self.date) > Client.OFFLINE_SLEEP:
|
||
self.peers_by_username = {}
|
||
self.peers_by_phone = {}
|
||
|
||
self.get_dialogs()
|
||
self.get_contacts()
|
||
else:
|
||
self.send(functions.messages.GetPinnedDialogs())
|
||
self.get_dialogs_chunk(0)
|
||
else:
|
||
self.send(functions.updates.GetState())
|
||
|
||
for i in range(self.UPDATES_WORKERS):
|
||
self.updates_workers_list.append(
|
||
Thread(
|
||
target=self.updates_worker,
|
||
name="UpdatesWorker#{}".format(i + 1)
|
||
)
|
||
)
|
||
|
||
self.updates_workers_list[-1].start()
|
||
|
||
for i in range(self.DOWNLOAD_WORKERS):
|
||
self.download_workers_list.append(
|
||
Thread(
|
||
target=self.download_worker,
|
||
name="DownloadWorker#{}".format(i + 1)
|
||
)
|
||
)
|
||
|
||
self.download_workers_list[-1].start()
|
||
|
||
self.dispatcher.start()
|
||
|
||
mimetypes.init()
|
||
Syncer.add(self)
|
||
|
||
def stop(self):
|
||
"""Use this method to manually stop the Client.
|
||
Requires no parameters.
|
||
"""
|
||
if not self.is_started:
|
||
raise ConnectionError("Client is already stopped")
|
||
|
||
for _ in range(self.UPDATES_WORKERS):
|
||
self.updates_queue.put(None)
|
||
|
||
for i in self.updates_workers_list:
|
||
i.join()
|
||
|
||
self.updates_workers_list.clear()
|
||
|
||
for _ in range(self.DOWNLOAD_WORKERS):
|
||
self.download_queue.put(None)
|
||
|
||
for i in self.download_workers_list:
|
||
i.join()
|
||
|
||
self.download_workers_list.clear()
|
||
|
||
self.dispatcher.stop()
|
||
|
||
self.is_started = False
|
||
self.session.stop()
|
||
|
||
Syncer.remove(self)
|
||
|
||
def authorize_bot(self):
|
||
try:
|
||
r = self.send(
|
||
functions.auth.ImportBotAuthorization(
|
||
flags=0,
|
||
api_id=self.api_id,
|
||
api_hash=self.api_hash,
|
||
bot_auth_token=self.token
|
||
)
|
||
)
|
||
except UserMigrate as e:
|
||
self.session.stop()
|
||
|
||
self.dc_id = e.x
|
||
self.auth_key = Auth(self.dc_id, self.test_mode, self.proxy).create()
|
||
|
||
self.session = Session(
|
||
self.dc_id,
|
||
self.test_mode,
|
||
self.proxy,
|
||
self.auth_key,
|
||
self.api_id,
|
||
client=self
|
||
)
|
||
|
||
self.session.start()
|
||
self.authorize_bot()
|
||
else:
|
||
self.user_id = r.user.id
|
||
|
||
def authorize_user(self):
|
||
phone_number_invalid_raises = self.phone_number is not None
|
||
phone_code_invalid_raises = self.phone_code is not None
|
||
password_hash_invalid_raises = self.password is not None
|
||
first_name_invalid_raises = self.first_name is not None
|
||
|
||
while True:
|
||
if self.phone_number is None:
|
||
self.phone_number = input("Enter phone number: ")
|
||
|
||
while True:
|
||
confirm = input("Is \"{}\" correct? (y/n): ".format(self.phone_number))
|
||
|
||
if confirm in ("y", "1"):
|
||
break
|
||
elif confirm in ("n", "2"):
|
||
self.phone_number = input("Enter phone number: ")
|
||
|
||
self.phone_number = self.phone_number.strip("+")
|
||
|
||
try:
|
||
r = self.send(
|
||
functions.auth.SendCode(
|
||
self.phone_number,
|
||
self.api_id,
|
||
self.api_hash
|
||
)
|
||
)
|
||
except (PhoneMigrate, NetworkMigrate) as e:
|
||
self.session.stop()
|
||
|
||
self.dc_id = e.x
|
||
self.auth_key = Auth(self.dc_id, self.test_mode, self.proxy).create()
|
||
|
||
self.session = Session(
|
||
self.dc_id,
|
||
self.test_mode,
|
||
self.proxy,
|
||
self.auth_key,
|
||
self.api_id,
|
||
client=self
|
||
)
|
||
self.session.start()
|
||
|
||
r = self.send(
|
||
functions.auth.SendCode(
|
||
self.phone_number,
|
||
self.api_id,
|
||
self.api_hash
|
||
)
|
||
)
|
||
break
|
||
except (PhoneNumberInvalid, PhoneNumberBanned) as e:
|
||
if phone_number_invalid_raises:
|
||
raise
|
||
else:
|
||
print(e.MESSAGE)
|
||
self.phone_number = None
|
||
except FloodWait as e:
|
||
print(e.MESSAGE.format(x=e.x))
|
||
time.sleep(e.x)
|
||
except Exception as e:
|
||
log.error(e, exc_info=True)
|
||
else:
|
||
break
|
||
|
||
phone_registered = r.phone_registered
|
||
phone_code_hash = r.phone_code_hash
|
||
|
||
if self.force_sms:
|
||
self.send(
|
||
functions.auth.ResendCode(
|
||
phone_number=self.phone_number,
|
||
phone_code_hash=phone_code_hash
|
||
)
|
||
)
|
||
|
||
while True:
|
||
self.phone_code = (
|
||
input("Enter phone code: ") if self.phone_code is None
|
||
else self.phone_code if type(self.phone_code) is str
|
||
else str(self.phone_code())
|
||
)
|
||
|
||
try:
|
||
if phone_registered:
|
||
r = self.send(
|
||
functions.auth.SignIn(
|
||
self.phone_number,
|
||
phone_code_hash,
|
||
self.phone_code
|
||
)
|
||
)
|
||
else:
|
||
try:
|
||
self.send(
|
||
functions.auth.SignIn(
|
||
self.phone_number,
|
||
phone_code_hash,
|
||
self.phone_code
|
||
)
|
||
)
|
||
except PhoneNumberUnoccupied:
|
||
pass
|
||
|
||
self.first_name = self.first_name if self.first_name is not None else input("First name: ")
|
||
self.last_name = self.last_name if self.last_name is not None else input("Last name: ")
|
||
|
||
r = self.send(
|
||
functions.auth.SignUp(
|
||
self.phone_number,
|
||
phone_code_hash,
|
||
self.phone_code,
|
||
self.first_name,
|
||
self.last_name
|
||
)
|
||
)
|
||
except (PhoneCodeInvalid, PhoneCodeEmpty, PhoneCodeExpired, PhoneCodeHashEmpty) as e:
|
||
if phone_code_invalid_raises:
|
||
raise
|
||
else:
|
||
print(e.MESSAGE)
|
||
self.phone_code = None
|
||
except FirstnameInvalid as e:
|
||
if first_name_invalid_raises:
|
||
raise
|
||
else:
|
||
print(e.MESSAGE)
|
||
self.first_name = None
|
||
except SessionPasswordNeeded as e:
|
||
print(e.MESSAGE)
|
||
r = self.send(functions.account.GetPassword())
|
||
|
||
while True:
|
||
try:
|
||
|
||
if self.password is None:
|
||
print("Hint: {}".format(r.hint))
|
||
self.password = input("Enter password: ") # TODO: Use getpass
|
||
|
||
if type(self.password) is str:
|
||
self.password = r.current_salt + self.password.encode() + r.current_salt
|
||
|
||
password_hash = sha256(self.password).digest()
|
||
|
||
r = self.send(functions.auth.CheckPassword(password_hash))
|
||
except PasswordHashInvalid as e:
|
||
if password_hash_invalid_raises:
|
||
raise
|
||
else:
|
||
print(e.MESSAGE)
|
||
self.password = None
|
||
except FloodWait as e:
|
||
print(e.MESSAGE.format(x=e.x))
|
||
time.sleep(e.x)
|
||
except Exception as e:
|
||
log.error(e, exc_info=True)
|
||
else:
|
||
break
|
||
break
|
||
except FloodWait as e:
|
||
print(e.MESSAGE.format(x=e.x))
|
||
time.sleep(e.x)
|
||
except Exception as e:
|
||
log.error(e, exc_info=True)
|
||
else:
|
||
break
|
||
|
||
self.password = None
|
||
self.user_id = r.user.id
|
||
|
||
def fetch_peers(self, entities: list):
|
||
for entity in entities:
|
||
if isinstance(entity, types.User):
|
||
user_id = entity.id
|
||
|
||
access_hash = entity.access_hash
|
||
|
||
if access_hash is None:
|
||
continue
|
||
|
||
username = entity.username
|
||
phone = entity.phone
|
||
|
||
input_peer = types.InputPeerUser(
|
||
user_id=user_id,
|
||
access_hash=access_hash
|
||
)
|
||
|
||
self.peers_by_id[user_id] = input_peer
|
||
|
||
if username is not None:
|
||
self.peers_by_username[username.lower()] = input_peer
|
||
|
||
if phone is not None:
|
||
self.peers_by_phone[phone] = input_peer
|
||
|
||
if isinstance(entity, types.Chat):
|
||
chat_id = entity.id
|
||
peer_id = -chat_id
|
||
|
||
input_peer = types.InputPeerChat(
|
||
chat_id=chat_id
|
||
)
|
||
|
||
self.peers_by_id[peer_id] = input_peer
|
||
|
||
if isinstance(entity, types.Channel):
|
||
channel_id = entity.id
|
||
peer_id = int("-100" + str(channel_id))
|
||
|
||
access_hash = entity.access_hash
|
||
|
||
if access_hash is None:
|
||
continue
|
||
|
||
username = entity.username
|
||
|
||
input_peer = types.InputPeerChannel(
|
||
channel_id=channel_id,
|
||
access_hash=access_hash
|
||
)
|
||
|
||
self.peers_by_id[peer_id] = input_peer
|
||
|
||
if username is not None:
|
||
self.peers_by_username[username.lower()] = input_peer
|
||
|
||
def download_worker(self):
|
||
name = threading.current_thread().name
|
||
log.debug("{} started".format(name))
|
||
|
||
while True:
|
||
media = self.download_queue.get()
|
||
|
||
if media is None:
|
||
break
|
||
|
||
temp_file_path = ""
|
||
final_file_path = ""
|
||
|
||
try:
|
||
media, file_name, done, progress, path = media
|
||
|
||
file_id = media.file_id
|
||
size = media.file_size
|
||
|
||
directory, file_name = os.path.split(file_name)
|
||
directory = directory or "downloads"
|
||
|
||
try:
|
||
decoded = utils.decode(file_id)
|
||
fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
|
||
unpacked = struct.unpack(fmt, decoded)
|
||
except (AssertionError, binascii.Error, struct.error):
|
||
raise FileIdInvalid from None
|
||
else:
|
||
media_type = unpacked[0]
|
||
dc_id = unpacked[1]
|
||
id = unpacked[2]
|
||
access_hash = unpacked[3]
|
||
volume_id = None
|
||
secret = None
|
||
local_id = None
|
||
|
||
if len(decoded) > 24:
|
||
volume_id = unpacked[4]
|
||
secret = unpacked[5]
|
||
local_id = unpacked[6]
|
||
|
||
media_type_str = Client.MEDIA_TYPE_ID.get(media_type, None)
|
||
|
||
if media_type_str:
|
||
log.info("The file_id belongs to a {}".format(media_type_str))
|
||
else:
|
||
raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
|
||
|
||
file_name = file_name or getattr(media, "file_name", None)
|
||
|
||
if not file_name:
|
||
if media_type == 3:
|
||
extension = ".ogg"
|
||
elif media_type in (4, 10, 13):
|
||
extension = mimetypes.guess_extension(media.mime_type) or ".mp4"
|
||
elif media_type == 5:
|
||
extension = mimetypes.guess_extension(media.mime_type) or ".unknown"
|
||
elif media_type == 8:
|
||
extension = ".webp"
|
||
elif media_type == 9:
|
||
extension = mimetypes.guess_extension(media.mime_type) or ".mp3"
|
||
elif media_type == 0:
|
||
extension = ".jpg"
|
||
elif media_type == 2:
|
||
extension = ".jpg"
|
||
else:
|
||
continue
|
||
|
||
file_name = "{}_{}_{}{}".format(
|
||
media_type_str,
|
||
datetime.fromtimestamp(media.date or time.time()).strftime("%Y-%m-%d_%H-%M-%S"),
|
||
self.rnd_id(),
|
||
extension
|
||
)
|
||
|
||
temp_file_path = self.get_file(
|
||
dc_id=dc_id,
|
||
id=id,
|
||
access_hash=access_hash,
|
||
volume_id=volume_id,
|
||
local_id=local_id,
|
||
secret=secret,
|
||
size=size,
|
||
progress=progress
|
||
)
|
||
|
||
if temp_file_path:
|
||
final_file_path = os.path.abspath(re.sub("\\\\", "/", os.path.join(directory, file_name)))
|
||
os.makedirs(directory, exist_ok=True)
|
||
shutil.move(temp_file_path, final_file_path)
|
||
except Exception as e:
|
||
log.error(e, exc_info=True)
|
||
|
||
try:
|
||
os.remove(temp_file_path)
|
||
except OSError:
|
||
pass
|
||
else:
|
||
# TODO: "" or None for faulty download, which is better?
|
||
# os.path methods return "" in case something does not exist, I prefer this.
|
||
# For now let's keep None
|
||
path[0] = final_file_path or None
|
||
finally:
|
||
done.set()
|
||
|
||
log.debug("{} stopped".format(name))
|
||
|
||
def updates_worker(self):
|
||
name = threading.current_thread().name
|
||
log.debug("{} started".format(name))
|
||
|
||
while True:
|
||
updates = self.updates_queue.get()
|
||
|
||
if updates is None:
|
||
break
|
||
|
||
try:
|
||
if isinstance(updates, (types.Update, types.UpdatesCombined)):
|
||
self.fetch_peers(updates.users)
|
||
self.fetch_peers(updates.chats)
|
||
|
||
for update in updates.updates:
|
||
channel_id = getattr(
|
||
getattr(
|
||
getattr(
|
||
update, "message", None
|
||
), "to_id", None
|
||
), "channel_id", None
|
||
) or getattr(update, "channel_id", None)
|
||
|
||
pts = getattr(update, "pts", None)
|
||
pts_count = getattr(update, "pts_count", None)
|
||
|
||
if isinstance(update, types.UpdateNewChannelMessage):
|
||
message = update.message
|
||
|
||
if not isinstance(message, types.MessageEmpty):
|
||
diff = self.send(
|
||
functions.updates.GetChannelDifference(
|
||
channel=self.resolve_peer(int("-100" + str(update.message.to_id.channel_id))),
|
||
filter=types.ChannelMessagesFilter(
|
||
ranges=[types.MessageRange(
|
||
min_id=update.message.id,
|
||
max_id=update.message.id
|
||
)]
|
||
),
|
||
pts=pts - pts_count,
|
||
limit=pts
|
||
)
|
||
)
|
||
|
||
if not isinstance(diff, types.updates.ChannelDifferenceEmpty):
|
||
updates.users += diff.users
|
||
updates.chats += diff.chats
|
||
|
||
if channel_id and pts:
|
||
if channel_id not in self.channels_pts:
|
||
self.channels_pts[channel_id] = []
|
||
|
||
if pts in self.channels_pts[channel_id]:
|
||
continue
|
||
|
||
self.channels_pts[channel_id].append(pts)
|
||
|
||
if len(self.channels_pts[channel_id]) > 50:
|
||
self.channels_pts[channel_id] = self.channels_pts[channel_id][25:]
|
||
|
||
self.dispatcher.updates.put((update, updates.users, updates.chats))
|
||
elif isinstance(updates, (types.UpdateShortMessage, types.UpdateShortChatMessage)):
|
||
diff = self.send(
|
||
functions.updates.GetDifference(
|
||
pts=updates.pts - updates.pts_count,
|
||
date=updates.date,
|
||
qts=-1
|
||
)
|
||
)
|
||
|
||
self.dispatcher.updates.put((
|
||
types.UpdateNewMessage(
|
||
message=diff.new_messages[0],
|
||
pts=updates.pts,
|
||
pts_count=updates.pts_count
|
||
),
|
||
diff.users,
|
||
diff.chats
|
||
))
|
||
elif isinstance(updates, types.UpdateShort):
|
||
self.dispatcher.updates.put((updates.update, [], []))
|
||
except Exception as e:
|
||
log.error(e, exc_info=True)
|
||
|
||
log.debug("{} stopped".format(name))
|
||
|
||
def signal_handler(self, *args):
|
||
self.is_idle = False
|
||
|
||
def idle(self, stop_signals: tuple = (SIGINT, SIGTERM, SIGABRT)):
|
||
"""Blocks the program execution until one of the signals are received,
|
||
then gently stop the Client by closing the underlying connection.
|
||
|
||
Args:
|
||
stop_signals (``tuple``, optional):
|
||
Iterable containing signals the signal handler will listen to.
|
||
Defaults to (SIGINT, SIGTERM, SIGABRT).
|
||
"""
|
||
for s in stop_signals:
|
||
signal(s, self.signal_handler)
|
||
|
||
self.is_idle = True
|
||
|
||
while self.is_idle:
|
||
time.sleep(1)
|
||
|
||
self.stop()
|
||
|
||
def send(self, data: Object):
|
||
"""Use this method to send Raw Function queries.
|
||
|
||
This method makes possible to manually call every single Telegram API method in a low-level manner.
|
||
Available functions are listed in the :obj:`functions <pyrogram.api.functions>` package and may accept compound
|
||
data types from :obj:`types <pyrogram.api.types>` as well as bare types such as ``int``, ``str``, etc...
|
||
|
||
Args:
|
||
data (``Object``):
|
||
The API Scheme function filled with proper arguments.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
if not self.is_started:
|
||
raise ConnectionError("Client has not been started")
|
||
|
||
r = self.session.send(data)
|
||
|
||
self.fetch_peers(getattr(r, "users", []))
|
||
self.fetch_peers(getattr(r, "chats", []))
|
||
|
||
return r
|
||
|
||
def load_config(self):
|
||
parser = ConfigParser()
|
||
parser.read("config.ini")
|
||
|
||
if self.api_id and self.api_hash:
|
||
pass
|
||
else:
|
||
if parser.has_section("pyrogram"):
|
||
self.api_id = parser.getint("pyrogram", "api_id")
|
||
self.api_hash = parser.get("pyrogram", "api_hash")
|
||
else:
|
||
raise AttributeError(
|
||
"No API Key found. "
|
||
"More info: https://docs.pyrogram.ml/start/ProjectSetup#configuration"
|
||
)
|
||
|
||
if self.proxy:
|
||
self.proxy["enabled"] = True
|
||
self.proxy["username"] = self.proxy.get("username", None)
|
||
self.proxy["password"] = self.proxy.get("password", None)
|
||
else:
|
||
self.proxy = {}
|
||
|
||
if parser.has_section("proxy"):
|
||
self.proxy["enabled"] = parser.getboolean("proxy", "enabled")
|
||
self.proxy["hostname"] = parser.get("proxy", "hostname")
|
||
self.proxy["port"] = parser.getint("proxy", "port")
|
||
self.proxy["username"] = parser.get("proxy", "username", fallback=None) or None
|
||
self.proxy["password"] = parser.get("proxy", "password", fallback=None) or None
|
||
|
||
def load_session(self):
|
||
try:
|
||
with open(os.path.join(self.workdir, "{}.session".format(self.session_name)), encoding="utf-8") as f:
|
||
s = json.load(f)
|
||
except FileNotFoundError:
|
||
self.dc_id = 1
|
||
self.date = 0
|
||
self.auth_key = Auth(self.dc_id, self.test_mode, self.proxy).create()
|
||
else:
|
||
self.dc_id = s["dc_id"]
|
||
self.test_mode = s["test_mode"]
|
||
self.auth_key = base64.b64decode("".join(s["auth_key"]))
|
||
self.user_id = s["user_id"]
|
||
self.date = s.get("date", 0)
|
||
|
||
for k, v in s.get("peers_by_id", {}).items():
|
||
self.peers_by_id[int(k)] = utils.get_input_peer(int(k), v)
|
||
|
||
for k, v in s.get("peers_by_username", {}).items():
|
||
peer = self.peers_by_id.get(v, None)
|
||
|
||
if peer:
|
||
self.peers_by_username[k] = peer
|
||
|
||
for k, v in s.get("peers_by_phone", {}).items():
|
||
peer = self.peers_by_id.get(v, None)
|
||
|
||
if peer:
|
||
self.peers_by_phone[k] = peer
|
||
|
||
def save_session(self):
|
||
auth_key = base64.b64encode(self.auth_key).decode()
|
||
auth_key = [auth_key[i: i + 43] for i in range(0, len(auth_key), 43)]
|
||
|
||
os.makedirs(self.workdir, exist_ok=True)
|
||
|
||
with open(os.path.join(self.workdir, "{}.session".format(self.session_name)), "w", encoding="utf-8") as f:
|
||
json.dump(
|
||
dict(
|
||
dc_id=self.dc_id,
|
||
test_mode=self.test_mode,
|
||
auth_key=auth_key,
|
||
user_id=self.user_id,
|
||
date=self.date
|
||
),
|
||
f,
|
||
indent=4
|
||
)
|
||
|
||
def get_dialogs_chunk(self, offset_date):
|
||
while True:
|
||
try:
|
||
r = self.send(
|
||
functions.messages.GetDialogs(
|
||
offset_date, 0, types.InputPeerEmpty(),
|
||
self.DIALOGS_AT_ONCE, True
|
||
)
|
||
)
|
||
except FloodWait as e:
|
||
log.warning("get_dialogs flood: waiting {} seconds".format(e.x))
|
||
time.sleep(e.x)
|
||
else:
|
||
log.info("Total peers: {}".format(len(self.peers_by_id)))
|
||
return r
|
||
|
||
def get_dialogs(self):
|
||
self.send(functions.messages.GetPinnedDialogs())
|
||
|
||
dialogs = self.get_dialogs_chunk(0)
|
||
offset_date = utils.get_offset_date(dialogs)
|
||
|
||
while len(dialogs.dialogs) == self.DIALOGS_AT_ONCE:
|
||
dialogs = self.get_dialogs_chunk(offset_date)
|
||
offset_date = utils.get_offset_date(dialogs)
|
||
|
||
self.get_dialogs_chunk(0)
|
||
|
||
def resolve_peer(self, peer_id: int or str):
|
||
"""Use this method to get the *InputPeer* of a known *peer_id*.
|
||
|
||
It is intended to be used when working with Raw Functions (i.e: a Telegram API method you wish to use which is
|
||
not available yet in the Client class as an easy-to-use method).
|
||
|
||
Args:
|
||
peer_id (``int`` | ``str`` | ``Peer``):
|
||
The Peer ID you want to extract the InputPeer from. Can be one of these types: ``int`` (direct ID),
|
||
``str`` (@username), :obj:`PeerUser <pyrogram.api.types.PeerUser>`,
|
||
:obj:`PeerChat <pyrogram.api.types.PeerChat>`, :obj:`PeerChannel <pyrogram.api.types.PeerChannel>`
|
||
|
||
Returns:
|
||
:obj:`InputPeerUser <pyrogram.api.types.InputPeerUser>` or
|
||
:obj:`InputPeerChat <pyrogram.api.types.InputPeerChat>` or
|
||
:obj:`InputPeerChannel <pyrogram.api.types.InputPeerChannel>` depending on the *peer_id*.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
if type(peer_id) is str:
|
||
if peer_id in ("self", "me"):
|
||
return types.InputPeerSelf()
|
||
|
||
match = self.INVITE_LINK_RE.match(peer_id)
|
||
|
||
try:
|
||
decoded = base64.b64decode(match.group(1) + "=" * (-len(match.group(1)) % 4), "-_")
|
||
return self.resolve_peer(struct.unpack(">2iq", decoded)[1])
|
||
except (AttributeError, binascii.Error, struct.error):
|
||
pass
|
||
|
||
peer_id = re.sub(r"[@+\s]", "", peer_id.lower())
|
||
|
||
try:
|
||
int(peer_id)
|
||
except ValueError:
|
||
try:
|
||
return self.peers_by_username[peer_id]
|
||
except KeyError:
|
||
self.send(functions.contacts.ResolveUsername(peer_id))
|
||
return self.peers_by_username[peer_id]
|
||
else:
|
||
try:
|
||
return self.peers_by_phone[peer_id]
|
||
except KeyError:
|
||
raise PeerIdInvalid
|
||
|
||
if type(peer_id) is not int:
|
||
if isinstance(peer_id, types.PeerUser):
|
||
peer_id = peer_id.user_id
|
||
elif isinstance(peer_id, types.PeerChat):
|
||
peer_id = -peer_id.chat_id
|
||
elif isinstance(peer_id, types.PeerChannel):
|
||
peer_id = int("-100" + str(peer_id.channel_id))
|
||
|
||
try: # User
|
||
return self.peers_by_id[peer_id]
|
||
except KeyError:
|
||
try: # Chat
|
||
return self.peers_by_id[-peer_id]
|
||
except KeyError:
|
||
try: # Channel
|
||
return self.peers_by_id[int("-100" + str(peer_id))]
|
||
except (KeyError, ValueError):
|
||
raise PeerIdInvalid
|
||
|
||
def get_me(self):
|
||
"""A simple method for testing the user authorization. Requires no parameters.
|
||
|
||
Returns:
|
||
Full information about the user in form of a :obj:`UserFull <pyrogram.api.types.UserFull>` object.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
return self.send(
|
||
functions.users.GetFullUser(
|
||
types.InputPeerSelf()
|
||
)
|
||
)
|
||
|
||
def send_message(self,
|
||
chat_id: int or str,
|
||
text: str,
|
||
parse_mode: str = "",
|
||
disable_web_page_preview: bool = None,
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None):
|
||
"""Use this method to send text messages.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
text (``str``):
|
||
Text of the message to be sent.
|
||
|
||
parse_mode (``str``):
|
||
Use :obj:`MARKDOWN <pyrogram.ParseMode.MARKDOWN>` or :obj:`HTML <pyrogram.ParseMode.HTML>`
|
||
if you want Telegram apps to show bold, italic, fixed-width text or inline URLs in your message.
|
||
Defaults to Markdown.
|
||
|
||
disable_web_page_preview (``bool``, optional):
|
||
Disables link previews for links in this message.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``bool``, optional):
|
||
If the message is a reply, ID of the original message.
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
Returns:
|
||
On success, the sent Message is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
style = self.html if parse_mode.lower() == "html" else self.markdown
|
||
|
||
return self.send(
|
||
functions.messages.SendMessage(
|
||
peer=self.resolve_peer(chat_id),
|
||
no_webpage=disable_web_page_preview or None,
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None,
|
||
**style.parse(text)
|
||
)
|
||
)
|
||
|
||
def forward_messages(self,
|
||
chat_id: int or str,
|
||
from_chat_id: int or str,
|
||
message_ids: list or int,
|
||
disable_notification: bool = None):
|
||
"""Use this method to forward messages of any kind.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
from_chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the source chat where the original message was sent.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
message_ids (``list`` | ``int``):
|
||
A list of Message identifiers in the chat specified in *from_chat_id* or a single message id.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
Returns:
|
||
On success and in case *message_ids* was a list, the returned value will be a list of the forwarded
|
||
:obj:`Messages <pyrogram.Message>` even if a list contains just one element, otherwise if
|
||
*message_ids* was an integer, the single forwarded :obj:`Message <pyrogram.Message>`
|
||
is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
is_list = isinstance(message_ids, list)
|
||
message_ids = message_ids if is_list else [message_ids]
|
||
|
||
r = self.send(
|
||
functions.messages.ForwardMessages(
|
||
to_peer=self.resolve_peer(chat_id),
|
||
from_peer=self.resolve_peer(from_chat_id),
|
||
id=message_ids,
|
||
silent=disable_notification or None,
|
||
random_id=[self.rnd_id() for _ in message_ids]
|
||
)
|
||
)
|
||
|
||
messages = []
|
||
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
messages.append(
|
||
utils.parse_message(self, i.message, users, chats)
|
||
)
|
||
|
||
return messages if is_list else messages[0]
|
||
|
||
def send_photo(self,
|
||
chat_id: int or str,
|
||
photo: str,
|
||
caption: str = "",
|
||
parse_mode: str = "",
|
||
ttl_seconds: int = None,
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None,
|
||
progress: callable = None):
|
||
"""Use this method to send photos.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
photo (``str``):
|
||
Photo to send.
|
||
Pass a file_id as string to send a photo that exists on the Telegram servers,
|
||
pass an HTTP URL as a string for Telegram to get a photo from the Internet, or
|
||
pass a file path as string to upload a new photo that exists on your local machine.
|
||
|
||
caption (``bool``, optional):
|
||
Photo caption, 0-200 characters.
|
||
|
||
parse_mode (``str``):
|
||
Use :obj:`MARKDOWN <pyrogram.ParseMode.MARKDOWN>` or :obj:`HTML <pyrogram.ParseMode.HTML>`
|
||
if you want Telegram apps to show bold, italic, fixed-width text or inline URLs in your caption.
|
||
Defaults to Markdown.
|
||
|
||
ttl_seconds (``int``, optional):
|
||
Self-Destruct Timer.
|
||
If you set a timer, the photo will self-destruct in *ttl_seconds*
|
||
seconds after it was viewed.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message.
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
progress (``callable``):
|
||
Pass a callback function to view the upload progress.
|
||
The function must accept two arguments (current, total).
|
||
|
||
Other Parameters:
|
||
current (``int``):
|
||
The amount of bytes uploaded so far.
|
||
|
||
total (``int``):
|
||
The size of the file.
|
||
|
||
Returns:
|
||
On success, the sent :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
file = None
|
||
style = self.html if parse_mode.lower() == "html" else self.markdown
|
||
|
||
if os.path.exists(photo):
|
||
file = self.save_file(photo, progress=progress)
|
||
media = types.InputMediaUploadedPhoto(
|
||
file=file,
|
||
ttl_seconds=ttl_seconds
|
||
)
|
||
elif photo.startswith("http"):
|
||
media = types.InputMediaPhotoExternal(
|
||
url=photo,
|
||
ttl_seconds=ttl_seconds
|
||
)
|
||
else:
|
||
try:
|
||
decoded = utils.decode(photo)
|
||
fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
|
||
unpacked = struct.unpack(fmt, decoded)
|
||
except (AssertionError, binascii.Error, struct.error):
|
||
raise FileIdInvalid from None
|
||
else:
|
||
if unpacked[0] != 2:
|
||
media_type = Client.MEDIA_TYPE_ID.get(unpacked[0], None)
|
||
|
||
if media_type:
|
||
raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
|
||
else:
|
||
raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
|
||
|
||
media = types.InputMediaPhoto(
|
||
id=types.InputPhoto(
|
||
id=unpacked[2],
|
||
access_hash=unpacked[3]
|
||
),
|
||
ttl_seconds=ttl_seconds
|
||
)
|
||
|
||
while True:
|
||
try:
|
||
r = self.send(
|
||
functions.messages.SendMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=media,
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None,
|
||
**style.parse(caption)
|
||
)
|
||
)
|
||
except FilePartMissing as e:
|
||
self.save_file(photo, file_id=file.id, file_part=e.x)
|
||
else:
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def send_audio(self,
|
||
chat_id: int or str,
|
||
audio: str,
|
||
caption: str = "",
|
||
parse_mode: str = "",
|
||
duration: int = 0,
|
||
performer: str = None,
|
||
title: str = None,
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None,
|
||
progress: callable = None):
|
||
"""Use this method to send audio files.
|
||
|
||
For sending voice messages, use the :obj:`send_voice` method instead.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
audio (``str``):
|
||
Audio file to send.
|
||
Pass a file_id as string to send an audio file that exists on the Telegram servers,
|
||
pass an HTTP URL as a string for Telegram to get an audio file from the Internet, or
|
||
pass a file path as string to upload a new audio file that exists on your local machine.
|
||
|
||
caption (``str``, optional):
|
||
Audio caption, 0-200 characters.
|
||
|
||
parse_mode (``str``):
|
||
Use :obj:`MARKDOWN <pyrogram.ParseMode.MARKDOWN>` or :obj:`HTML <pyrogram.ParseMode.HTML>`
|
||
if you want Telegram apps to show bold, italic, fixed-width text or inline URLs in your caption.
|
||
Defaults to Markdown.
|
||
|
||
duration (``int``, optional):
|
||
Duration of the audio in seconds.
|
||
|
||
performer (``str``, optional):
|
||
Performer.
|
||
|
||
title (``str``, optional):
|
||
Track name.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message.
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
progress (``callable``):
|
||
Pass a callback function to view the upload progress.
|
||
The function must accept two arguments (current, total).
|
||
|
||
Other Parameters:
|
||
current (``int``):
|
||
The amount of bytes uploaded so far.
|
||
|
||
total (``int``):
|
||
The size of the file.
|
||
|
||
Returns:
|
||
On success, the sent :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
file = None
|
||
style = self.html if parse_mode.lower() == "html" else self.markdown
|
||
|
||
if os.path.exists(audio):
|
||
file = self.save_file(audio, progress=progress)
|
||
media = types.InputMediaUploadedDocument(
|
||
mime_type=mimetypes.types_map.get("." + audio.split(".")[-1], "audio/mpeg"),
|
||
file=file,
|
||
attributes=[
|
||
types.DocumentAttributeAudio(
|
||
duration=duration,
|
||
performer=performer,
|
||
title=title
|
||
),
|
||
types.DocumentAttributeFilename(os.path.basename(audio))
|
||
]
|
||
)
|
||
elif audio.startswith("http"):
|
||
media = types.InputMediaDocumentExternal(
|
||
url=audio
|
||
)
|
||
else:
|
||
try:
|
||
decoded = utils.decode(audio)
|
||
fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
|
||
unpacked = struct.unpack(fmt, decoded)
|
||
except (AssertionError, binascii.Error, struct.error):
|
||
raise FileIdInvalid from None
|
||
else:
|
||
if unpacked[0] != 9:
|
||
media_type = Client.MEDIA_TYPE_ID.get(unpacked[0], None)
|
||
|
||
if media_type:
|
||
raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
|
||
else:
|
||
raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
|
||
|
||
media = types.InputMediaDocument(
|
||
id=types.InputDocument(
|
||
id=unpacked[2],
|
||
access_hash=unpacked[3]
|
||
)
|
||
)
|
||
|
||
while True:
|
||
try:
|
||
r = self.send(
|
||
functions.messages.SendMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=media,
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None,
|
||
**style.parse(caption)
|
||
)
|
||
)
|
||
except FilePartMissing as e:
|
||
self.save_file(audio, file_id=file.id, file_part=e.x)
|
||
else:
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def send_document(self,
|
||
chat_id: int or str,
|
||
document: str,
|
||
caption: str = "",
|
||
parse_mode: str = "",
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None,
|
||
progress: callable = None):
|
||
"""Use this method to send general files.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
document (``str``):
|
||
File to send.
|
||
Pass a file_id as string to send a file that exists on the Telegram servers,
|
||
pass an HTTP URL as a string for Telegram to get a file from the Internet, or
|
||
pass a file path as string to upload a new file that exists on your local machine.
|
||
|
||
caption (``str``, optional):
|
||
Document caption, 0-200 characters.
|
||
|
||
parse_mode (``str``):
|
||
Use :obj:`MARKDOWN <pyrogram.ParseMode.MARKDOWN>` or :obj:`HTML <pyrogram.ParseMode.HTML>`
|
||
if you want Telegram apps to show bold, italic, fixed-width text or inline URLs in your caption.
|
||
Defaults to Markdown.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message.
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
progress (``callable``):
|
||
Pass a callback function to view the upload progress.
|
||
The function must accept two arguments (current, total).
|
||
|
||
Other Parameters:
|
||
current (``int``):
|
||
The amount of bytes uploaded so far.
|
||
|
||
total (``int``):
|
||
The size of the file.
|
||
|
||
Returns:
|
||
On success, the sent :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
file = None
|
||
style = self.html if parse_mode.lower() == "html" else self.markdown
|
||
|
||
if os.path.exists(document):
|
||
file = self.save_file(document, progress=progress)
|
||
media = types.InputMediaUploadedDocument(
|
||
mime_type=mimetypes.types_map.get("." + document.split(".")[-1], "text/plain"),
|
||
file=file,
|
||
attributes=[
|
||
types.DocumentAttributeFilename(os.path.basename(document))
|
||
]
|
||
)
|
||
elif document.startswith("http"):
|
||
media = types.InputMediaDocumentExternal(
|
||
url=document
|
||
)
|
||
else:
|
||
try:
|
||
decoded = utils.decode(document)
|
||
fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
|
||
unpacked = struct.unpack(fmt, decoded)
|
||
except (AssertionError, binascii.Error, struct.error):
|
||
raise FileIdInvalid from None
|
||
else:
|
||
if unpacked[0] not in (5, 10):
|
||
media_type = Client.MEDIA_TYPE_ID.get(unpacked[0], None)
|
||
|
||
if media_type:
|
||
raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
|
||
else:
|
||
raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
|
||
|
||
media = types.InputMediaDocument(
|
||
id=types.InputDocument(
|
||
id=unpacked[2],
|
||
access_hash=unpacked[3]
|
||
)
|
||
)
|
||
|
||
while True:
|
||
try:
|
||
r = self.send(
|
||
functions.messages.SendMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=media,
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None,
|
||
**style.parse(caption)
|
||
)
|
||
)
|
||
except FilePartMissing as e:
|
||
self.save_file(document, file_id=file.id, file_part=e.x)
|
||
else:
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def send_sticker(self,
|
||
chat_id: int or str,
|
||
sticker: str,
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None,
|
||
progress: callable = None):
|
||
"""Use this method to send .webp stickers.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
sticker (``str``):
|
||
Sticker to send.
|
||
Pass a file_id as string to send a sticker that exists on the Telegram servers,
|
||
pass an HTTP URL as a string for Telegram to get a .webp sticker file from the Internet, or
|
||
pass a file path as string to upload a new sticker that exists on your local machine.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message.
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
progress (``callable``):
|
||
Pass a callback function to view the upload progress.
|
||
The function must accept two arguments (current, total).
|
||
|
||
Other Parameters:
|
||
current (``int``):
|
||
The amount of bytes uploaded so far.
|
||
|
||
total (``int``):
|
||
The size of the file.
|
||
|
||
Returns:
|
||
On success, the sent :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
file = None
|
||
|
||
if os.path.exists(sticker):
|
||
file = self.save_file(sticker, progress=progress)
|
||
media = types.InputMediaUploadedDocument(
|
||
mime_type="image/webp",
|
||
file=file,
|
||
attributes=[
|
||
types.DocumentAttributeFilename(os.path.basename(sticker))
|
||
]
|
||
)
|
||
elif sticker.startswith("http"):
|
||
media = types.InputMediaDocumentExternal(
|
||
url=sticker
|
||
)
|
||
else:
|
||
try:
|
||
decoded = utils.decode(sticker)
|
||
fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
|
||
unpacked = struct.unpack(fmt, decoded)
|
||
except (AssertionError, binascii.Error, struct.error):
|
||
raise FileIdInvalid from None
|
||
else:
|
||
if unpacked[0] != 8:
|
||
media_type = Client.MEDIA_TYPE_ID.get(unpacked[0], None)
|
||
|
||
if media_type:
|
||
raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
|
||
else:
|
||
raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
|
||
|
||
media = types.InputMediaDocument(
|
||
id=types.InputDocument(
|
||
id=unpacked[2],
|
||
access_hash=unpacked[3]
|
||
)
|
||
)
|
||
|
||
while True:
|
||
try:
|
||
r = self.send(
|
||
functions.messages.SendMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=media,
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None,
|
||
message=""
|
||
)
|
||
)
|
||
except FilePartMissing as e:
|
||
self.save_file(sticker, file_id=file.id, file_part=e.x)
|
||
else:
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def send_video(self,
|
||
chat_id: int or str,
|
||
video: str,
|
||
caption: str = "",
|
||
parse_mode: str = "",
|
||
duration: int = 0,
|
||
width: int = 0,
|
||
height: int = 0,
|
||
thumb: str = None,
|
||
supports_streaming: bool = True,
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None,
|
||
progress: callable = None):
|
||
"""Use this method to send video files.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
video (``str``):
|
||
Video to send.
|
||
Pass a file_id as string to send a video that exists on the Telegram servers,
|
||
pass an HTTP URL as a string for Telegram to get a video from the Internet, or
|
||
pass a file path as string to upload a new video that exists on your local machine.
|
||
|
||
caption (``str``, optional):
|
||
Video caption, 0-200 characters.
|
||
|
||
parse_mode (``str``):
|
||
Use :obj:`MARKDOWN <pyrogram.ParseMode.MARKDOWN>` or :obj:`HTML <pyrogram.ParseMode.HTML>`
|
||
if you want Telegram apps to show bold, italic, fixed-width text or inline URLs in your caption.
|
||
Defaults to Markdown.
|
||
|
||
duration (``int``, optional):
|
||
Duration of sent video in seconds.
|
||
|
||
width (``int``, optional):
|
||
Video width.
|
||
|
||
height (``int``, optional):
|
||
Video height.
|
||
|
||
thumb (``str``, optional):
|
||
Video thumbnail.
|
||
Pass a file path as string to send an image that exists on your local machine.
|
||
Thumbnail should have 90 or less pixels of width and 90 or less pixels of height.
|
||
|
||
supports_streaming (``bool``, optional):
|
||
Pass True, if the uploaded video is suitable for streaming.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message.
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
progress (``callable``):
|
||
Pass a callback function to view the upload progress.
|
||
The function must accept two arguments (current, total).
|
||
|
||
Other Parameters:
|
||
current (``int``):
|
||
The amount of bytes uploaded so far.
|
||
|
||
total (``int``):
|
||
The size of the file.
|
||
|
||
Returns:
|
||
On success, the sent :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
file = None
|
||
style = self.html if parse_mode.lower() == "html" else self.markdown
|
||
|
||
if os.path.exists(video):
|
||
thumb = None if thumb is None else self.save_file(thumb)
|
||
file = self.save_file(video, progress=progress)
|
||
media = types.InputMediaUploadedDocument(
|
||
mime_type=mimetypes.types_map[".mp4"],
|
||
file=file,
|
||
thumb=thumb,
|
||
attributes=[
|
||
types.DocumentAttributeVideo(
|
||
supports_streaming=supports_streaming or None,
|
||
duration=duration,
|
||
w=width,
|
||
h=height
|
||
),
|
||
types.DocumentAttributeFilename(os.path.basename(video))
|
||
]
|
||
)
|
||
elif video.startswith("http"):
|
||
media = types.InputMediaDocumentExternal(
|
||
url=video
|
||
)
|
||
else:
|
||
try:
|
||
decoded = utils.decode(video)
|
||
fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
|
||
unpacked = struct.unpack(fmt, decoded)
|
||
except (AssertionError, binascii.Error, struct.error):
|
||
raise FileIdInvalid from None
|
||
else:
|
||
if unpacked[0] != 4:
|
||
media_type = Client.MEDIA_TYPE_ID.get(unpacked[0], None)
|
||
|
||
if media_type:
|
||
raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
|
||
else:
|
||
raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
|
||
|
||
media = types.InputMediaDocument(
|
||
id=types.InputDocument(
|
||
id=unpacked[2],
|
||
access_hash=unpacked[3]
|
||
)
|
||
)
|
||
|
||
while True:
|
||
try:
|
||
r = self.send(
|
||
functions.messages.SendMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=media,
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None,
|
||
**style.parse(caption)
|
||
)
|
||
)
|
||
except FilePartMissing as e:
|
||
self.save_file(video, file_id=file.id, file_part=e.x)
|
||
else:
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def send_voice(self,
|
||
chat_id: int or str,
|
||
voice: str,
|
||
caption: str = "",
|
||
parse_mode: str = "",
|
||
duration: int = 0,
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None,
|
||
progress: callable = None):
|
||
"""Use this method to send audio files.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
voice (``str``):
|
||
Audio file to send.
|
||
Pass a file_id as string to send an audio that exists on the Telegram servers,
|
||
pass an HTTP URL as a string for Telegram to get an audio from the Internet, or
|
||
pass a file path as string to upload a new audio that exists on your local machine.
|
||
|
||
caption (``str``, optional):
|
||
Voice message caption, 0-200 characters.
|
||
|
||
parse_mode (``str``):
|
||
Use :obj:`MARKDOWN <pyrogram.ParseMode.MARKDOWN>` or :obj:`HTML <pyrogram.ParseMode.HTML>`
|
||
if you want Telegram apps to show bold, italic, fixed-width text or inline URLs in your caption.
|
||
Defaults to Markdown.
|
||
|
||
duration (``int``, optional):
|
||
Duration of the voice message in seconds.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
progress (``callable``):
|
||
Pass a callback function to view the upload progress.
|
||
The function must accept two arguments (current, total).
|
||
|
||
Other Parameters:
|
||
current (``int``):
|
||
The amount of bytes uploaded so far.
|
||
|
||
total (``int``):
|
||
The size of the file.
|
||
|
||
Returns:
|
||
On success, the sent :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
file = None
|
||
style = self.html if parse_mode.lower() == "html" else self.markdown
|
||
|
||
if os.path.exists(voice):
|
||
file = self.save_file(voice, progress=progress)
|
||
media = types.InputMediaUploadedDocument(
|
||
mime_type=mimetypes.types_map.get("." + voice.split(".")[-1], "audio/mpeg"),
|
||
file=file,
|
||
attributes=[
|
||
types.DocumentAttributeAudio(
|
||
voice=True,
|
||
duration=duration
|
||
)
|
||
]
|
||
)
|
||
elif voice.startswith("http"):
|
||
media = types.InputMediaDocumentExternal(
|
||
url=voice
|
||
)
|
||
else:
|
||
try:
|
||
decoded = utils.decode(voice)
|
||
fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
|
||
unpacked = struct.unpack(fmt, decoded)
|
||
except (AssertionError, binascii.Error, struct.error):
|
||
raise FileIdInvalid from None
|
||
else:
|
||
if unpacked[0] != 3:
|
||
media_type = Client.MEDIA_TYPE_ID.get(unpacked[0], None)
|
||
|
||
if media_type:
|
||
raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
|
||
else:
|
||
raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
|
||
|
||
media = types.InputMediaDocument(
|
||
id=types.InputDocument(
|
||
id=unpacked[2],
|
||
access_hash=unpacked[3]
|
||
)
|
||
)
|
||
|
||
while True:
|
||
try:
|
||
r = self.send(
|
||
functions.messages.SendMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=media,
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None,
|
||
**style.parse(caption)
|
||
)
|
||
)
|
||
except FilePartMissing as e:
|
||
self.save_file(voice, file_id=file.id, file_part=e.x)
|
||
else:
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def send_video_note(self,
|
||
chat_id: int or str,
|
||
video_note: str,
|
||
duration: int = 0,
|
||
length: int = 1,
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None,
|
||
progress: callable = None):
|
||
"""Use this method to send video messages.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
video_note (``str``):
|
||
Video note to send.
|
||
Pass a file_id as string to send a video note that exists on the Telegram servers, or
|
||
pass a file path as string to upload a new video note that exists on your local machine.
|
||
Sending video notes by a URL is currently unsupported.
|
||
|
||
duration (``int``, optional):
|
||
Duration of sent video in seconds.
|
||
|
||
length (``int``, optional):
|
||
Video width and height.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
progress (``callable``):
|
||
Pass a callback function to view the upload progress.
|
||
The function must accept two arguments (current, total).
|
||
|
||
Other Parameters:
|
||
current (``int``):
|
||
The amount of bytes uploaded so far.
|
||
|
||
total (``int``):
|
||
The size of the file.
|
||
|
||
Returns:
|
||
On success, the sent :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
file = None
|
||
|
||
if os.path.exists(video_note):
|
||
file = self.save_file(video_note, progress=progress)
|
||
media = types.InputMediaUploadedDocument(
|
||
mime_type=mimetypes.types_map[".mp4"],
|
||
file=file,
|
||
attributes=[
|
||
types.DocumentAttributeVideo(
|
||
round_message=True,
|
||
duration=duration,
|
||
w=length,
|
||
h=length
|
||
)
|
||
]
|
||
)
|
||
else:
|
||
try:
|
||
decoded = utils.decode(video_note)
|
||
fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
|
||
unpacked = struct.unpack(fmt, decoded)
|
||
except (AssertionError, binascii.Error, struct.error):
|
||
raise FileIdInvalid from None
|
||
else:
|
||
if unpacked[0] != 13:
|
||
media_type = Client.MEDIA_TYPE_ID.get(unpacked[0], None)
|
||
|
||
if media_type:
|
||
raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
|
||
else:
|
||
raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
|
||
|
||
media = types.InputMediaDocument(
|
||
id=types.InputDocument(
|
||
id=unpacked[2],
|
||
access_hash=unpacked[3]
|
||
)
|
||
)
|
||
|
||
while True:
|
||
try:
|
||
r = self.send(
|
||
functions.messages.SendMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=media,
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None,
|
||
message=""
|
||
)
|
||
)
|
||
except FilePartMissing as e:
|
||
self.save_file(video_note, file_id=file.id, file_part=e.x)
|
||
else:
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
# TODO: Add progress parameter
|
||
# TODO: Return new Message object
|
||
# TODO: Figure out how to send albums using URLs
|
||
def send_media_group(self,
|
||
chat_id: int or str,
|
||
media: list,
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None):
|
||
"""Use this method to send a group of photos or videos as an album.
|
||
On success, an Update containing the sent Messages is returned.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
media (``list``):
|
||
A list containing either :obj:`InputMediaPhoto <pyrogram.InputMediaPhoto>` or
|
||
:obj:`InputMediaVideo <pyrogram.InputMediaVideo>` objects
|
||
describing photos and videos to be sent, must include 2–10 items.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message.
|
||
"""
|
||
multi_media = []
|
||
|
||
for i in media:
|
||
style = self.html if i.parse_mode.lower() == "html" else self.markdown
|
||
|
||
if isinstance(i, pyrogram_types.InputMediaPhoto):
|
||
if os.path.exists(i.media):
|
||
media = self.send(
|
||
functions.messages.UploadMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=types.InputMediaUploadedPhoto(
|
||
file=self.save_file(i.media)
|
||
)
|
||
)
|
||
)
|
||
|
||
media = types.InputMediaPhoto(
|
||
id=types.InputPhoto(
|
||
id=media.photo.id,
|
||
access_hash=media.photo.access_hash
|
||
)
|
||
)
|
||
else:
|
||
try:
|
||
decoded = utils.decode(i.media)
|
||
fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
|
||
unpacked = struct.unpack(fmt, decoded)
|
||
except (AssertionError, binascii.Error, struct.error):
|
||
raise FileIdInvalid from None
|
||
else:
|
||
if unpacked[0] != 2:
|
||
media_type = Client.MEDIA_TYPE_ID.get(unpacked[0], None)
|
||
|
||
if media_type:
|
||
raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
|
||
else:
|
||
raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
|
||
|
||
media = types.InputMediaPhoto(
|
||
id=types.InputPhoto(
|
||
id=unpacked[2],
|
||
access_hash=unpacked[3]
|
||
)
|
||
)
|
||
elif isinstance(i, pyrogram_types.InputMediaVideo):
|
||
if os.path.exists(i.media):
|
||
media = self.send(
|
||
functions.messages.UploadMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=types.InputMediaUploadedDocument(
|
||
file=self.save_file(i.media),
|
||
mime_type=mimetypes.types_map[".mp4"],
|
||
attributes=[
|
||
types.DocumentAttributeVideo(
|
||
supports_streaming=i.supports_streaming or None,
|
||
duration=i.duration,
|
||
w=i.width,
|
||
h=i.height
|
||
),
|
||
types.DocumentAttributeFilename(os.path.basename(i.media))
|
||
]
|
||
)
|
||
)
|
||
)
|
||
|
||
media = types.InputMediaDocument(
|
||
id=types.InputDocument(
|
||
id=media.document.id,
|
||
access_hash=media.document.access_hash
|
||
)
|
||
)
|
||
else:
|
||
try:
|
||
decoded = utils.decode(i.media)
|
||
fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
|
||
unpacked = struct.unpack(fmt, decoded)
|
||
except (AssertionError, binascii.Error, struct.error):
|
||
raise FileIdInvalid from None
|
||
else:
|
||
if unpacked[0] != 4:
|
||
media_type = Client.MEDIA_TYPE_ID.get(unpacked[0], None)
|
||
|
||
if media_type:
|
||
raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
|
||
else:
|
||
raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
|
||
|
||
media = types.InputMediaDocument(
|
||
id=types.InputDocument(
|
||
id=unpacked[2],
|
||
access_hash=unpacked[3]
|
||
)
|
||
)
|
||
|
||
multi_media.append(
|
||
types.InputSingleMedia(
|
||
media=media,
|
||
random_id=self.rnd_id(),
|
||
**style.parse(i.caption)
|
||
)
|
||
)
|
||
|
||
return self.send(
|
||
functions.messages.SendMultiMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
multi_media=multi_media,
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id
|
||
)
|
||
)
|
||
|
||
def send_location(self,
|
||
chat_id: int or str,
|
||
latitude: float,
|
||
longitude: float,
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None):
|
||
"""Use this method to send points on the map.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
latitude (``float``):
|
||
Latitude of the location.
|
||
|
||
longitude (``float``):
|
||
Longitude of the location.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
Returns:
|
||
On success, the sent :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
r = self.send(
|
||
functions.messages.SendMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=types.InputMediaGeoPoint(
|
||
types.InputGeoPoint(
|
||
latitude,
|
||
longitude
|
||
)
|
||
),
|
||
message="",
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None
|
||
)
|
||
)
|
||
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def send_venue(self,
|
||
chat_id: int or str,
|
||
latitude: float,
|
||
longitude: float,
|
||
title: str,
|
||
address: str,
|
||
foursquare_id: str = "",
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None):
|
||
"""Use this method to send information about a venue.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
latitude (``float``):
|
||
Latitude of the venue.
|
||
|
||
longitude (``float``):
|
||
Longitude of the venue.
|
||
|
||
title (``str``):
|
||
Name of the venue.
|
||
|
||
address (``str``):
|
||
Address of the venue.
|
||
|
||
foursquare_id (``str``, optional):
|
||
Foursquare identifier of the venue.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
Returns:
|
||
On success, the sent :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
r = self.send(
|
||
functions.messages.SendMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=types.InputMediaVenue(
|
||
geo_point=types.InputGeoPoint(
|
||
lat=latitude,
|
||
long=longitude
|
||
),
|
||
title=title,
|
||
address=address,
|
||
provider="",
|
||
venue_id=foursquare_id,
|
||
venue_type=""
|
||
),
|
||
message="",
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None
|
||
)
|
||
)
|
||
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def send_contact(self,
|
||
chat_id: int or str,
|
||
phone_number: str,
|
||
first_name: str,
|
||
last_name: str = "",
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None,
|
||
reply_markup=None):
|
||
"""Use this method to send phone contacts.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
phone_number (``str``):
|
||
Contact's phone number.
|
||
|
||
first_name (``str``):
|
||
Contact's first name.
|
||
|
||
last_name (``str``):
|
||
Contact's last name.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``int``, optional):
|
||
If the message is a reply, ID of the original message.
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup` | :obj:`ReplyKeyboardMarkup` | :obj:`ReplyKeyboardRemove` | :obj:`ForceReply`, optional):
|
||
Additional interface options. An object for an inline keyboard, custom reply keyboard,
|
||
instructions to remove reply keyboard or to force a reply from the user.
|
||
|
||
Returns:
|
||
On success, the sent :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
r = self.send(
|
||
functions.messages.SendMedia(
|
||
peer=self.resolve_peer(chat_id),
|
||
media=types.InputMediaContact(
|
||
phone_number,
|
||
first_name,
|
||
last_name
|
||
),
|
||
message="",
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id,
|
||
random_id=self.rnd_id(),
|
||
reply_markup=reply_markup.write() if reply_markup else None
|
||
)
|
||
)
|
||
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def send_chat_action(self,
|
||
chat_id: int or str,
|
||
action: callable,
|
||
progress: int = 0):
|
||
"""Use this method when you need to tell the other party that something is happening on your side.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
action (``callable``):
|
||
Type of action to broadcast.
|
||
Choose one from the :class:`ChatAction <pyrogram.ChatAction>` class,
|
||
depending on what the user is about to receive.
|
||
|
||
progress (``int``, optional):
|
||
Progress of the upload process.
|
||
|
||
Returns:
|
||
On success, True is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
if "Upload" in action.__name__:
|
||
action = action(progress)
|
||
else:
|
||
action = action()
|
||
|
||
return self.send(
|
||
functions.messages.SetTyping(
|
||
peer=self.resolve_peer(chat_id),
|
||
action=action
|
||
)
|
||
)
|
||
|
||
def get_user_profile_photos(self,
|
||
user_id: int or str,
|
||
offset: int = 0,
|
||
limit: int = 100):
|
||
"""Use this method to get a list of profile pictures for a user.
|
||
|
||
Args:
|
||
user_id (``int`` | ``str``):
|
||
Unique identifier of the target user.
|
||
|
||
offset (``int``, optional):
|
||
Sequential number of the first photo to be returned.
|
||
By default, all photos are returned.
|
||
|
||
limit (``int``, optional):
|
||
Limits the number of photos to be retrieved.
|
||
Values between 1—100 are accepted. Defaults to 100.
|
||
|
||
Returns:
|
||
On success, a :obj:`UserProfilePhotos` object is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
return utils.parse_photos(
|
||
self.send(
|
||
functions.photos.GetUserPhotos(
|
||
user_id=self.resolve_peer(user_id),
|
||
offset=offset,
|
||
max_id=0,
|
||
limit=limit
|
||
)
|
||
)
|
||
)
|
||
|
||
def edit_message_text(self,
|
||
chat_id: int or str,
|
||
message_id: int,
|
||
text: str,
|
||
parse_mode: str = "",
|
||
disable_web_page_preview: bool = None,
|
||
reply_markup=None):
|
||
"""Use this method to edit text messages.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
message_id (``int``):
|
||
Message identifier in the chat specified in chat_id.
|
||
|
||
text (``str``):
|
||
New text of the message.
|
||
|
||
parse_mode (``str``):
|
||
Use :obj:`MARKDOWN <pyrogram.ParseMode.MARKDOWN>` or :obj:`HTML <pyrogram.ParseMode.HTML>`
|
||
if you want Telegram apps to show bold, italic, fixed-width text or inline URLs in your message.
|
||
Defaults to Markdown.
|
||
|
||
disable_web_page_preview (``bool``, optional):
|
||
Disables link previews for links in this message.
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup`):
|
||
An InlineKeyboardMarkup object.
|
||
|
||
Returns:
|
||
On success, the edited :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
style = self.html if parse_mode.lower() == "html" else self.markdown
|
||
|
||
r = self.send(
|
||
functions.messages.EditMessage(
|
||
peer=self.resolve_peer(chat_id),
|
||
id=message_id,
|
||
no_webpage=disable_web_page_preview or None,
|
||
reply_markup=reply_markup.write() if reply_markup else None,
|
||
**style.parse(text)
|
||
)
|
||
)
|
||
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateEditMessage, types.UpdateEditChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def edit_message_caption(self,
|
||
chat_id: int or str,
|
||
message_id: int,
|
||
caption: str,
|
||
parse_mode: str = "",
|
||
reply_markup=None):
|
||
"""Use this method to edit captions of messages.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
message_id (``int``):
|
||
Message identifier in the chat specified in chat_id.
|
||
|
||
caption (``str``):
|
||
New caption of the message.
|
||
|
||
parse_mode (``str``):
|
||
Use :obj:`MARKDOWN <pyrogram.ParseMode.MARKDOWN>` or :obj:`HTML <pyrogram.ParseMode.HTML>`
|
||
if you want Telegram apps to show bold, italic, fixed-width text or inline URLs in your caption.
|
||
Defaults to Markdown.
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup`):
|
||
An InlineKeyboardMarkup object.
|
||
|
||
Returns:
|
||
On success, the edited :obj:`Message <pyrogram.Message>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
style = self.html if parse_mode.lower() == "html" else self.markdown
|
||
|
||
r = self.send(
|
||
functions.messages.EditMessage(
|
||
peer=self.resolve_peer(chat_id),
|
||
id=message_id,
|
||
reply_markup=reply_markup.write() if reply_markup else None,
|
||
**style.parse(caption)
|
||
)
|
||
)
|
||
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateEditMessage, types.UpdateEditChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def edit_message_reply_markup(self,
|
||
chat_id: int or str,
|
||
message_id: int,
|
||
reply_markup=None):
|
||
"""Use this method to edit only the reply markup of messages sent by the bot or via the bot (for inline bots).
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
message_id (``int``):
|
||
Message identifier in the chat specified in chat_id.
|
||
|
||
reply_markup (:obj:`InlineKeyboardMarkup`):
|
||
An InlineKeyboardMarkup object.
|
||
|
||
Returns:
|
||
On success, if edited message is sent by the bot, the edited
|
||
:obj:`Message <pyrogram.Message>` is returned, otherwise True is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
|
||
r = self.send(
|
||
functions.messages.EditMessage(
|
||
peer=self.resolve_peer(chat_id),
|
||
id=message_id,
|
||
reply_markup=reply_markup.write() if reply_markup else None
|
||
)
|
||
)
|
||
|
||
for i in r.updates:
|
||
if isinstance(i, (types.UpdateEditMessage, types.UpdateEditChannelMessage)):
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
return utils.parse_message(self, i.message, users, chats)
|
||
|
||
def delete_messages(self,
|
||
chat_id: int or str,
|
||
message_ids: list or int,
|
||
revoke: bool = True):
|
||
"""Use this method to delete messages, including service messages, with the following limitations:
|
||
|
||
- A message can only be deleted if it was sent less than 48 hours ago.
|
||
- Users can delete outgoing messages in groups and supergroups.
|
||
- Users granted *can_post_messages* permissions can delete outgoing messages in channels.
|
||
- If the user is an administrator of a group, it can delete any message there.
|
||
- If the user has *can_delete_messages* permission in a supergroup or a channel, it can delete any message there.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
message_ids (``list`` | ``int``):
|
||
A list of Message identifiers to delete or a single message id.
|
||
|
||
revoke (``bool``, optional):
|
||
Deletes messages on both parts.
|
||
This is only for private cloud chats and normal groups, messages on
|
||
channels and supergroups are always revoked (i.e.: deleted for everyone).
|
||
Defaults to True.
|
||
|
||
Returns:
|
||
True on success.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
peer = self.resolve_peer(chat_id)
|
||
message_ids = message_ids if isinstance(message_ids, list) else [message_ids]
|
||
|
||
if isinstance(peer, types.InputPeerChannel):
|
||
self.send(
|
||
functions.channels.DeleteMessages(
|
||
channel=peer,
|
||
id=message_ids
|
||
)
|
||
)
|
||
else:
|
||
self.send(
|
||
functions.messages.DeleteMessages(
|
||
id=message_ids,
|
||
revoke=revoke or None
|
||
)
|
||
)
|
||
|
||
return True
|
||
|
||
# TODO: Improvements for the new API
|
||
def save_file(self,
|
||
path: str,
|
||
file_id: int = None,
|
||
file_part: int = 0,
|
||
progress: callable = None):
|
||
part_size = 512 * 1024
|
||
file_size = os.path.getsize(path)
|
||
file_total_parts = int(math.ceil(file_size / part_size))
|
||
is_big = True if file_size > 10 * 1024 * 1024 else False
|
||
is_missing_part = True if file_id is not None else False
|
||
file_id = file_id or self.rnd_id()
|
||
md5_sum = md5() if not is_big and not is_missing_part else None
|
||
|
||
session = Session(self.dc_id, self.test_mode, self.proxy, self.auth_key, self.api_id)
|
||
session.start()
|
||
|
||
try:
|
||
with open(path, "rb") as f:
|
||
f.seek(part_size * file_part)
|
||
|
||
while True:
|
||
chunk = f.read(part_size)
|
||
|
||
if not chunk:
|
||
if not is_big:
|
||
md5_sum = "".join([hex(i)[2:].zfill(2) for i in md5_sum.digest()])
|
||
break
|
||
|
||
if is_big:
|
||
rpc = functions.upload.SaveBigFilePart(
|
||
file_id=file_id,
|
||
file_part=file_part,
|
||
file_total_parts=file_total_parts,
|
||
bytes=chunk
|
||
)
|
||
else:
|
||
rpc = functions.upload.SaveFilePart(
|
||
file_id=file_id,
|
||
file_part=file_part,
|
||
bytes=chunk
|
||
)
|
||
|
||
assert self.send(rpc), "Couldn't upload file"
|
||
|
||
if is_missing_part:
|
||
return
|
||
|
||
if not is_big:
|
||
md5_sum.update(chunk)
|
||
|
||
file_part += 1
|
||
|
||
if progress:
|
||
progress(min(file_part * part_size, file_size), file_size)
|
||
except Exception as e:
|
||
log.error(e, exc_info=True)
|
||
else:
|
||
if is_big:
|
||
return types.InputFileBig(
|
||
id=file_id,
|
||
parts=file_total_parts,
|
||
name=os.path.basename(path),
|
||
|
||
)
|
||
else:
|
||
return types.InputFile(
|
||
id=file_id,
|
||
parts=file_total_parts,
|
||
name=os.path.basename(path),
|
||
md5_checksum=md5_sum
|
||
)
|
||
finally:
|
||
session.stop()
|
||
|
||
# TODO: Improvements for the new API
|
||
def get_file(self,
|
||
dc_id: int,
|
||
id: int = None,
|
||
access_hash: int = None,
|
||
volume_id: int = None,
|
||
local_id: int = None,
|
||
secret: int = None,
|
||
version: int = 0,
|
||
size: int = None,
|
||
progress: callable = None) -> str:
|
||
if dc_id != self.dc_id:
|
||
exported_auth = self.send(
|
||
functions.auth.ExportAuthorization(
|
||
dc_id=dc_id
|
||
)
|
||
)
|
||
|
||
session = Session(
|
||
dc_id,
|
||
self.test_mode,
|
||
self.proxy,
|
||
Auth(dc_id, self.test_mode, self.proxy).create(),
|
||
self.api_id
|
||
)
|
||
|
||
session.start()
|
||
|
||
session.send(
|
||
functions.auth.ImportAuthorization(
|
||
id=exported_auth.id,
|
||
bytes=exported_auth.bytes
|
||
)
|
||
)
|
||
else:
|
||
session = Session(
|
||
dc_id,
|
||
self.test_mode,
|
||
self.proxy,
|
||
self.auth_key,
|
||
self.api_id
|
||
)
|
||
|
||
session.start()
|
||
|
||
if volume_id: # Photos are accessed by volume_id, local_id, secret
|
||
location = types.InputFileLocation(
|
||
volume_id=volume_id,
|
||
local_id=local_id,
|
||
secret=secret
|
||
)
|
||
else: # Any other file can be more easily accessed by id and access_hash
|
||
location = types.InputDocumentFileLocation(
|
||
id=id,
|
||
access_hash=access_hash,
|
||
version=version
|
||
)
|
||
|
||
limit = 1024 * 1024
|
||
offset = 0
|
||
file_name = ""
|
||
|
||
try:
|
||
r = session.send(
|
||
functions.upload.GetFile(
|
||
location=location,
|
||
offset=offset,
|
||
limit=limit
|
||
)
|
||
)
|
||
|
||
if isinstance(r, types.upload.File):
|
||
with tempfile.NamedTemporaryFile("wb", delete=False) as f:
|
||
file_name = f.name
|
||
|
||
while True:
|
||
chunk = r.bytes
|
||
|
||
if not chunk:
|
||
break
|
||
|
||
f.write(chunk)
|
||
f.flush()
|
||
os.fsync(f.fileno())
|
||
|
||
offset += limit
|
||
|
||
if progress:
|
||
progress(min(offset, size), size)
|
||
|
||
r = session.send(
|
||
functions.upload.GetFile(
|
||
location=location,
|
||
offset=offset,
|
||
limit=limit
|
||
)
|
||
)
|
||
|
||
elif isinstance(r, types.upload.FileCdnRedirect):
|
||
cdn_session = Session(
|
||
r.dc_id,
|
||
self.test_mode,
|
||
self.proxy,
|
||
Auth(r.dc_id, self.test_mode, self.proxy).create(),
|
||
self.api_id,
|
||
is_cdn=True
|
||
)
|
||
|
||
cdn_session.start()
|
||
|
||
try:
|
||
with tempfile.NamedTemporaryFile("wb", delete=False) as f:
|
||
file_name = f.name
|
||
|
||
while True:
|
||
r2 = cdn_session.send(
|
||
functions.upload.GetCdnFile(
|
||
file_token=r.file_token,
|
||
offset=offset,
|
||
limit=limit
|
||
)
|
||
)
|
||
|
||
if isinstance(r2, types.upload.CdnFileReuploadNeeded):
|
||
try:
|
||
session.send(
|
||
functions.upload.ReuploadCdnFile(
|
||
file_token=r.file_token,
|
||
request_token=r2.request_token
|
||
)
|
||
)
|
||
except VolumeLocNotFound:
|
||
break
|
||
else:
|
||
continue
|
||
|
||
chunk = r2.bytes
|
||
|
||
# https://core.telegram.org/cdn#decrypting-files
|
||
decrypted_chunk = AES.ctr_decrypt(
|
||
chunk,
|
||
r.encryption_key,
|
||
r.encryption_iv,
|
||
offset
|
||
)
|
||
|
||
hashes = session.send(
|
||
functions.upload.GetCdnFileHashes(
|
||
r.file_token,
|
||
offset
|
||
)
|
||
)
|
||
|
||
# https://core.telegram.org/cdn#verifying-files
|
||
for i, h in enumerate(hashes):
|
||
cdn_chunk = decrypted_chunk[h.limit * i: h.limit * (i + 1)]
|
||
assert h.hash == sha256(cdn_chunk).digest(), "Invalid CDN hash part {}".format(i)
|
||
|
||
f.write(decrypted_chunk)
|
||
f.flush()
|
||
os.fsync(f.fileno())
|
||
|
||
offset += limit
|
||
|
||
if progress:
|
||
progress(min(offset, size), size)
|
||
|
||
if len(chunk) < limit:
|
||
break
|
||
except Exception as e:
|
||
raise e
|
||
finally:
|
||
cdn_session.stop()
|
||
except Exception as e:
|
||
log.error(e, exc_info=True)
|
||
|
||
try:
|
||
os.remove(file_name)
|
||
except OSError:
|
||
pass
|
||
|
||
return ""
|
||
else:
|
||
return file_name
|
||
finally:
|
||
session.stop()
|
||
|
||
def join_chat(self, chat_id: str):
|
||
"""Use this method to join a group chat or channel.
|
||
|
||
Args:
|
||
chat_id (``str``):
|
||
Unique identifier for the target chat in form of *t.me/joinchat/* links or username of the target
|
||
channel/supergroup (in the format @username).
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
match = self.INVITE_LINK_RE.match(chat_id)
|
||
|
||
if match:
|
||
return self.send(
|
||
functions.messages.ImportChatInvite(
|
||
hash=match.group(1)
|
||
)
|
||
)
|
||
else:
|
||
resolved_peer = self.send(
|
||
functions.contacts.ResolveUsername(
|
||
username=chat_id.lower().strip("@")
|
||
)
|
||
)
|
||
|
||
channel = types.InputPeerChannel(
|
||
channel_id=resolved_peer.chats[0].id,
|
||
access_hash=resolved_peer.chats[0].access_hash
|
||
)
|
||
|
||
return self.send(
|
||
functions.channels.JoinChannel(
|
||
channel=channel
|
||
)
|
||
)
|
||
|
||
def leave_chat(self, chat_id: int or str, delete: bool = False):
|
||
"""Use this method to leave a group chat or channel.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier for the target chat or username of the target channel/supergroup
|
||
(in the format @username).
|
||
|
||
delete (``bool``, optional):
|
||
Deletes the group chat dialog after leaving (for simple group chats, not supergroups).
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
peer = self.resolve_peer(chat_id)
|
||
|
||
if isinstance(peer, types.InputPeerChannel):
|
||
return self.send(
|
||
functions.channels.LeaveChannel(
|
||
channel=self.resolve_peer(chat_id)
|
||
)
|
||
)
|
||
elif isinstance(peer, types.InputPeerChat):
|
||
r = self.send(
|
||
functions.messages.DeleteChatUser(
|
||
chat_id=peer.chat_id,
|
||
user_id=types.InputPeerSelf()
|
||
)
|
||
)
|
||
|
||
if delete:
|
||
self.send(
|
||
functions.messages.DeleteHistory(
|
||
peer=peer,
|
||
max_id=0
|
||
)
|
||
)
|
||
|
||
return r
|
||
|
||
def export_chat_invite_link(self, chat_id: int or str, new: bool = False):
|
||
"""Use this method to export an invite link to a supergroup or a channel.
|
||
|
||
The user must be an administrator in the chat for this to work and must have the appropriate admin rights.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier for the target chat or username of the target channel/supergroup
|
||
(in the format @username).
|
||
|
||
new (``bool``):
|
||
The previous link will be deactivated and a new link will be generated.
|
||
This is also used to create the invite link in case it doesn't exist yet.
|
||
|
||
Returns:
|
||
On success, the exported invite link as string is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
|
||
Note:
|
||
If the returned link is a new one it may take a while for it to be activated.
|
||
"""
|
||
peer = self.resolve_peer(chat_id)
|
||
|
||
if isinstance(peer, types.InputPeerChat):
|
||
if new:
|
||
return self.send(
|
||
functions.messages.ExportChatInvite(
|
||
chat_id=peer.chat_id
|
||
)
|
||
).link
|
||
else:
|
||
chat_full = self.send(
|
||
functions.messages.GetFullChat(
|
||
chat_id=peer.chat_id
|
||
)
|
||
).full_chat # type: types.ChatFull
|
||
|
||
if isinstance(chat_full.exported_invite, types.ChatInviteExported):
|
||
return chat_full.exported_invite.link
|
||
else:
|
||
raise ChatAdminRequired
|
||
elif isinstance(peer, types.InputPeerChannel):
|
||
if new:
|
||
return self.send(
|
||
functions.channels.ExportInvite(
|
||
channel=peer
|
||
)
|
||
).link
|
||
else:
|
||
channel_full = self.send(
|
||
functions.channels.GetFullChannel(
|
||
channel=peer
|
||
)
|
||
).full_chat # type: types.ChannelFull
|
||
|
||
if isinstance(channel_full.exported_invite, types.ChatInviteExported):
|
||
return channel_full.exported_invite.link
|
||
else:
|
||
raise ChatAdminRequired
|
||
|
||
def enable_cloud_password(self, password: str, hint: str = "", email: str = ""):
|
||
"""Use this method to enable the Two-Step Verification security feature (Cloud Password) on your account.
|
||
|
||
This password will be asked when you log in on a new device in addition to the SMS code.
|
||
|
||
Args:
|
||
password (``str``):
|
||
Your password.
|
||
|
||
hint (``str``, optional):
|
||
A password hint.
|
||
|
||
email (``str``, optional):
|
||
Recovery e-mail.
|
||
|
||
Returns:
|
||
True on success, False otherwise.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
r = self.send(functions.account.GetPassword())
|
||
|
||
if isinstance(r, types.account.NoPassword):
|
||
salt = r.new_salt + os.urandom(8)
|
||
password_hash = sha256(salt + password.encode() + salt).digest()
|
||
|
||
return self.send(
|
||
functions.account.UpdatePasswordSettings(
|
||
current_password_hash=salt,
|
||
new_settings=types.account.PasswordInputSettings(
|
||
new_salt=salt,
|
||
new_password_hash=password_hash,
|
||
hint=hint,
|
||
email=email
|
||
)
|
||
)
|
||
)
|
||
else:
|
||
return False
|
||
|
||
def change_cloud_password(self, current_password: str, new_password: str, new_hint: str = ""):
|
||
"""Use this method to change your Two-Step Verification password (Cloud Password) with a new one.
|
||
|
||
Args:
|
||
current_password (``str``):
|
||
Your current password.
|
||
|
||
new_password (``str``):
|
||
Your new password.
|
||
|
||
new_hint (``str``, optional):
|
||
A new password hint.
|
||
|
||
Returns:
|
||
True on success, False otherwise.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
r = self.send(functions.account.GetPassword())
|
||
|
||
if isinstance(r, types.account.Password):
|
||
current_password_hash = sha256(r.current_salt + current_password.encode() + r.current_salt).digest()
|
||
|
||
new_salt = r.new_salt + os.urandom(8)
|
||
new_password_hash = sha256(new_salt + new_password.encode() + new_salt).digest()
|
||
|
||
return self.send(
|
||
functions.account.UpdatePasswordSettings(
|
||
current_password_hash=current_password_hash,
|
||
new_settings=types.account.PasswordInputSettings(
|
||
new_salt=new_salt,
|
||
new_password_hash=new_password_hash,
|
||
hint=new_hint
|
||
)
|
||
)
|
||
)
|
||
else:
|
||
return False
|
||
|
||
def remove_cloud_password(self, password: str):
|
||
"""Use this method to turn off the Two-Step Verification security feature (Cloud Password) on your account.
|
||
|
||
Args:
|
||
password (``str``):
|
||
Your current password.
|
||
|
||
Returns:
|
||
True on success, False otherwise.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
r = self.send(functions.account.GetPassword())
|
||
|
||
if isinstance(r, types.account.Password):
|
||
password_hash = sha256(r.current_salt + password.encode() + r.current_salt).digest()
|
||
|
||
return self.send(
|
||
functions.account.UpdatePasswordSettings(
|
||
current_password_hash=password_hash,
|
||
new_settings=types.account.PasswordInputSettings(
|
||
new_salt=b"",
|
||
new_password_hash=b"",
|
||
hint=""
|
||
)
|
||
)
|
||
)
|
||
else:
|
||
return False
|
||
|
||
def download_media(self,
|
||
message: pyrogram_types.Message,
|
||
file_name: str = "",
|
||
block: bool = True,
|
||
progress: callable = None):
|
||
"""Use this method to download the media from a Message.
|
||
|
||
Args:
|
||
message (:obj:`Message <pyrogram.Message>` | ``str``):
|
||
Pass a Message containing the media, the media itself (message.audio, message.video, ...) or
|
||
the file id as string.
|
||
|
||
file_name (``str``, optional):
|
||
A custom *file_name* to be used instead of the one provided by Telegram.
|
||
By default, all files are downloaded in the *downloads* folder in your working directory.
|
||
You can also specify a path for downloading files in a custom location: paths that end with "/"
|
||
are considered directories. All non-existent folders will be created automatically.
|
||
|
||
block (``bool``, optional):
|
||
Blocks the code execution until the file has been downloaded.
|
||
Defaults to True.
|
||
|
||
progress (``callable``):
|
||
Pass a callback function to view the download progress.
|
||
The function must accept two arguments (current, total).
|
||
Note that this will not work in case you are downloading a media using a *file_id*.
|
||
|
||
Other Parameters:
|
||
current (``int``):
|
||
The amount of bytes downloaded so far.
|
||
|
||
total (``int``):
|
||
The size of the file.
|
||
|
||
Returns:
|
||
On success, the absolute path of the downloaded file as string is returned, None otherwise.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
if isinstance(message, pyrogram_types.Message):
|
||
if message.photo:
|
||
media = message.photo[-1]
|
||
elif message.audio:
|
||
media = message.audio
|
||
elif message.document:
|
||
media = message.document
|
||
elif message.video:
|
||
media = message.video
|
||
elif message.voice:
|
||
media = message.voice
|
||
elif message.video_note:
|
||
media = message.video_note
|
||
elif message.sticker:
|
||
media = message.sticker
|
||
else:
|
||
return
|
||
elif isinstance(message, (
|
||
pyrogram_types.PhotoSize,
|
||
pyrogram_types.Audio,
|
||
pyrogram_types.Document,
|
||
pyrogram_types.Video,
|
||
pyrogram_types.Voice,
|
||
pyrogram_types.VideoNote,
|
||
pyrogram_types.Sticker
|
||
)):
|
||
media = message
|
||
elif isinstance(message, str):
|
||
media = pyrogram_types.Document(
|
||
file_id=message,
|
||
file_size=0,
|
||
mime_type=""
|
||
)
|
||
else:
|
||
return
|
||
|
||
done = Event()
|
||
path = [None]
|
||
|
||
self.download_queue.put((media, file_name, done, progress, path))
|
||
|
||
if block:
|
||
done.wait()
|
||
|
||
return path[0]
|
||
|
||
def add_contacts(self, contacts: list):
|
||
"""Use this method to add contacts to your Telegram address book.
|
||
|
||
Args:
|
||
contacts (``list``):
|
||
A list of :obj:`InputPhoneContact <pyrogram.InputPhoneContact>`
|
||
|
||
Returns:
|
||
On success, the added contacts are returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
imported_contacts = self.send(
|
||
functions.contacts.ImportContacts(
|
||
contacts=contacts
|
||
)
|
||
)
|
||
|
||
return imported_contacts
|
||
|
||
def delete_contacts(self, ids: list):
|
||
"""Use this method to delete contacts from your Telegram address book
|
||
|
||
Args:
|
||
ids (``list``):
|
||
A list of unique identifiers for the target users.
|
||
Can be an ID (int), a username (string) or phone number (string).
|
||
|
||
Returns:
|
||
True on success.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
contacts = []
|
||
|
||
for i in ids:
|
||
try:
|
||
input_user = self.resolve_peer(i)
|
||
except PeerIdInvalid:
|
||
continue
|
||
else:
|
||
if isinstance(input_user, types.InputPeerUser):
|
||
contacts.append(input_user)
|
||
|
||
return self.send(
|
||
functions.contacts.DeleteContacts(
|
||
id=contacts
|
||
)
|
||
)
|
||
|
||
def get_contacts(self, _hash: int = 0):
|
||
while True:
|
||
try:
|
||
contacts = self.send(functions.contacts.GetContacts(_hash))
|
||
except FloodWait as e:
|
||
log.warning("get_contacts flood: waiting {} seconds".format(e.x))
|
||
time.sleep(e.x)
|
||
continue
|
||
else:
|
||
if isinstance(contacts, types.contacts.Contacts):
|
||
log.info("Total contacts: {}".format(len(self.peers_by_phone)))
|
||
|
||
return contacts
|
||
|
||
def get_inline_bot_results(self,
|
||
bot: int or str,
|
||
query: str,
|
||
offset: str = "",
|
||
location: tuple = None):
|
||
"""Use this method to get bot results via inline queries.
|
||
You can then send a result using :obj:`send_inline_bot_result <pyrogram.Client.send_inline_bot_result>`
|
||
|
||
Args:
|
||
bot (``int`` | ``str``):
|
||
Unique identifier of the inline bot you want to get results from. You can specify
|
||
a @username (str) or a bot ID (int).
|
||
|
||
query (``str``):
|
||
Text of the query (up to 512 characters).
|
||
|
||
offset (``str``):
|
||
Offset of the results to be returned.
|
||
|
||
location (``tuple``, optional):
|
||
Your location in tuple format (latitude, longitude), e.g.: (51.500729, -0.124583).
|
||
Useful for location-based results only.
|
||
|
||
Returns:
|
||
On Success, :obj:`BotResults <pyrogram.api.types.messages.BotResults>` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
return self.send(
|
||
functions.messages.GetInlineBotResults(
|
||
bot=self.resolve_peer(bot),
|
||
peer=types.InputPeerSelf(),
|
||
query=query,
|
||
offset=offset,
|
||
geo_point=types.InputGeoPoint(
|
||
lat=location[0],
|
||
long=location[1]
|
||
) if location else None
|
||
)
|
||
)
|
||
|
||
def send_inline_bot_result(self,
|
||
chat_id: int or str,
|
||
query_id: int,
|
||
result_id: str,
|
||
disable_notification: bool = None,
|
||
reply_to_message_id: int = None):
|
||
"""Use this method to send an inline bot result.
|
||
Bot results can be retrieved using :obj:`get_inline_bot_results <pyrogram.Client.get_inline_bot_results>`
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
query_id (``int``):
|
||
Unique identifier for the answered query.
|
||
|
||
result_id (``str``):
|
||
Unique identifier for the result that was chosen.
|
||
|
||
disable_notification (``bool``, optional):
|
||
Sends the message silently.
|
||
Users will receive a notification with no sound.
|
||
|
||
reply_to_message_id (``bool``, optional):
|
||
If the message is a reply, ID of the original message.
|
||
|
||
Returns:
|
||
On success, the sent Message is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
return self.send(
|
||
functions.messages.SendInlineBotResult(
|
||
peer=self.resolve_peer(chat_id),
|
||
query_id=query_id,
|
||
id=result_id,
|
||
random_id=self.rnd_id(),
|
||
silent=disable_notification or None,
|
||
reply_to_msg_id=reply_to_message_id
|
||
)
|
||
)
|
||
|
||
def get_users(self, user_ids: list or int or str):
|
||
"""Use this method to get information about a user.
|
||
You can retrieve up to 200 users at once.
|
||
|
||
Args:
|
||
user_ids (``list`` | ``int`` | ``str``):
|
||
A list of User identifiers (id or username) or a single user id/username.
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
|
||
Returns:
|
||
On success and in case *user_ids* was a list, the returned value will be a list of the requested
|
||
:obj:`Users <User>` even if a list contains just one element, otherwise if
|
||
*user_ids* was an integer, the single requested :obj:`User` is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
is_list = isinstance(user_ids, list)
|
||
user_ids = user_ids if is_list else [user_ids]
|
||
user_ids = [self.resolve_peer(i) for i in user_ids]
|
||
|
||
r = self.send(
|
||
functions.users.GetUsers(
|
||
id=user_ids
|
||
)
|
||
)
|
||
|
||
users = []
|
||
|
||
for i in r:
|
||
users.append(utils.parse_user(i))
|
||
|
||
return users if is_list else users[0]
|
||
|
||
def get_messages(self,
|
||
chat_id: int or str,
|
||
message_ids: list or int,
|
||
replies: int = 1):
|
||
"""Use this method to get messages that belong to a specific chat.
|
||
You can retrieve up to 200 messages at once.
|
||
|
||
Args:
|
||
chat_id (``int`` | ``str``):
|
||
Unique identifier (int) or username (str) of the target chat.
|
||
For your personal cloud (Saved Messages) you can simply use "me" or "self".
|
||
For a contact that exists in your Telegram address book you can use his phone number (str).
|
||
For a private channel/supergroup you can use its *t.me/joinchat/* link.
|
||
|
||
message_ids (``list`` | ``int``):
|
||
A list of Message identifiers in the chat specified in *chat_id* or a single message id, as integer.
|
||
|
||
replies (``int``, optional):
|
||
The number of replies to get for each message. Defaults to 1.
|
||
|
||
Returns:
|
||
On success and in case *message_ids* was a list, the returned value will be a list of the requested
|
||
:obj:`Messages <pyrogram.Message>` even if a list contains just one element, otherwise if
|
||
*message_ids* was an integer, the single requested :obj:`Message <pyrogram.Message>`
|
||
is returned.
|
||
|
||
Raises:
|
||
:class:`Error <pyrogram.Error>`
|
||
"""
|
||
peer = self.resolve_peer(chat_id)
|
||
is_list = isinstance(message_ids, list)
|
||
message_ids = message_ids if is_list else [message_ids]
|
||
message_ids = [types.InputMessageID(i) for i in message_ids]
|
||
|
||
if isinstance(peer, types.InputPeerChannel):
|
||
rpc = functions.channels.GetMessages(
|
||
channel=peer,
|
||
id=message_ids
|
||
)
|
||
else:
|
||
rpc = functions.messages.GetMessages(
|
||
id=message_ids
|
||
)
|
||
|
||
r = self.send(rpc)
|
||
|
||
users = {i.id: i for i in r.users}
|
||
chats = {i.id: i for i in r.chats}
|
||
|
||
messages = []
|
||
|
||
for i in r.messages:
|
||
if isinstance(i, types.Message):
|
||
messages.append(
|
||
utils.parse_message(
|
||
self, i, users, chats,
|
||
replies=replies
|
||
)
|
||
)
|
||
elif isinstance(i, types.MessageService):
|
||
messages.append(
|
||
utils.parse_message_service(
|
||
self, i, users, chats
|
||
)
|
||
)
|
||
else:
|
||
messages.append(
|
||
utils.parse_message_empty(
|
||
self, i
|
||
)
|
||
)
|
||
|
||
return messages if is_list else messages[0]
|
||
|
||
def answer_callback_query(self,
|
||
callback_query_id: str,
|
||
text: str = None,
|
||
show_alert: bool = None,
|
||
url: str = None,
|
||
cache_time: int = 0):
|
||
"""Use this method to send answers to callback queries sent from inline keyboards.
|
||
The answer will be displayed to the user as a notification at the top of the chat screen or as an alert.
|
||
|
||
Args:
|
||
callback_query_id (``str``):
|
||
Unique identifier for the query to be answered.
|
||
|
||
text (``str``):
|
||
Text of the notification. If not specified, nothing will be shown to the user, 0-200 characters.
|
||
|
||
show_alert (``bool``):
|
||
If true, an alert will be shown by the client instead of a notification at the top of the chat screen.
|
||
Defaults to False.
|
||
|
||
url (``str``):
|
||
URL that will be opened by the user's client.
|
||
If you have created a Game and accepted the conditions via @Botfather, specify the URL that opens your
|
||
game – note that this will only work if the query comes from a callback_game button.
|
||
Otherwise, you may use links like t.me/your_bot?start=XXXX that open your bot with a parameter.
|
||
|
||
cache_time (``int``):
|
||
The maximum amount of time in seconds that the result of the callback query may be cached client-side.
|
||
Telegram apps will support caching starting in version 3.14. Defaults to 0.
|
||
"""
|
||
return self.send(
|
||
functions.messages.SetBotCallbackAnswer(
|
||
query_id=int(callback_query_id),
|
||
cache_time=cache_time,
|
||
alert=show_alert or None,
|
||
message=text,
|
||
url=url
|
||
)
|
||
)
|