TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 04:35:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix message.download in case of Story

Browse Source
This commit is contained in:
KurimuzonAkuma 2023-12-24 10:14:02 +03:00
parent f144c999ef
commit d8a760e9a8
2 changed files with 9 additions and 77 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pyrogram/methods/messages/download_media.py
View File

@ -120,12 +120,19 @@ class DownloadMedia:
file_bytes = bytes(file.getbuffer())
"""
available_media = ("audio", "document", "photo", "sticker", "animation", "video", "voice", "video_note",
"new_chat_photo")
"new_chat_photo", "story")
if isinstance(message, types.Message):
for kind in available_media:
media = getattr(message, kind, None)
if kind == "story":
for kind in available_media:
media = getattr(message.story, kind, None)
if media is not None:
break
if media is not None:
break
else:

77
pyrogram/types/messages_and_media/story.py
View File

@ -63,7 +63,7 @@ class Story(Object, Update):
media (:obj:`~pyrogram.enums.MessageMediaType`, *optional*):
The media type of the Story.
This field will contain the enumeration type of the media message.
You can use ``media = getattr(message, message.media.value)`` to access the media message.
You can use ``media = getattr(message, message.story.media.value)`` to access the media message.
has_protected_content (``bool``, *optional*):
True, if the story can't be forwarded.
@ -1733,81 +1733,6 @@ class Story(Object, Update):
schedule_date=schedule_date
)
async def download(
self,
file_name: str = "",
in_memory: bool = False,
block: bool = True,
progress: Callable = None,
progress_args: tuple = ()
) -> str:
"""Bound method *download* of :obj:`~pyrogram.types.Story`.
Use as a shortcut for:
.. code-block:: python
await client.download_media(story)
Example:
.. code-block:: python
await story.download()
Parameters:
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.
in_memory (``bool``, *optional*):
Pass True to download the media in-memory.
A binary file-like object with its attribute ".name" set will be returned.
Defaults to False.
block (``bool``, *optional*):
Blocks the code execution until the file has been downloaded.
Defaults to True.
progress (``Callable``, *optional*):
Pass a callback function to view the file transmission progress.
The function must take *(current, total)* as positional arguments (look at Other Parameters below for a
detailed description) and will be called back each time a new file chunk has been successfully
transmitted.
progress_args (``tuple``, *optional*):
Extra custom arguments for the progress callback function.
You can pass anything you need to be available in the progress callback scope; for example, a Message
object or a Client instance in order to edit the message with the updated progress status.
Other Parameters:
current (``int``):
The amount of bytes transmitted so far.
total (``int``):
The total size of the file.
*args (``tuple``, *optional*):
Extra custom arguments as defined in the ``progress_args`` parameter.
You can either keep ``*args`` or add every single extra argument in your function signature.
Returns:
On success, the absolute path of the downloaded file as string is returned, None otherwise.
Raises:
RPCError: In case of a Telegram RPC error.
``ValueError``: If the message doesn't contain any downloadable media
"""
return await self._client.download_media(
message=getattr(self, self.media.value),
file_name=file_name,
in_memory=in_memory,
block=block,
progress=progress,
progress_args=progress_args,
)
async def read(self) -> List[int]:
"""Bound method *read* of :obj:`~pyrogram.types.Story`.