yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-01 05:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Pyrofork: Move duplicated codes to separate method

Browse source 
Signed-off-by: wulan17 <wulan17@nusantararom.org>
This commit is contained in:
wulan17 2024-01-04 20:15:51 +07:00
parent 41e13454e7
commit f2b6182584
No known key found for this signature in database
GPG key ID: 318CD6CD3A6AC0A5
1 changed files with 21 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
pyrogram/methods/users/send_story.py
View file

@ -28,6 +28,25 @@ class SendStory:
def _split(self, message, entities, *args, **kwargs):
return message, entities
async def _upload_video(
self: "pyrogram.Client",
file_name: str,
video: Union[str, BinaryIO]
):
file = await self.save_file(video)
return raw.types.InputMediaUploadedDocument(
mime_type=self.guess_mime_type(file_name or video.name) or "video/mp4",
file=file,
attributes=[
raw.types.DocumentAttributeVideo(
supports_streaming=True,
duration=0,
w=0,
h=0
)
]
)
async def send_story(
self: "pyrogram.Client",
chat_id: Union[int,str] = None,
@ -185,33 +204,9 @@ class SendStory:
)
else:
video = await self.download_media(video, in_memory=True)
file = await self.save_file(video)
media = raw.types.InputMediaUploadedDocument(
mime_type=self.guess_mime_type(file_name or video.name) or "video/mp4",
file=file,
attributes=[
raw.types.DocumentAttributeVideo(
supports_streaming=True,
duration=0,
w=0,
h=0
)
]
)
media = await self._upload_video(file_name,video)
else:
file = await self.save_file(video)
media = raw.types.InputMediaUploadedDocument(
mime_type=self.guess_mime_type(file_name or video.name) or "video/mp4",
file=file,
attributes=[
raw.types.DocumentAttributeVideo(
supports_streaming=True,
duration=0,
w=0,
h=0
)
]
)
media = await self._upload_video(file_name,video)
else:
if forward_from_chat_id is None:
raise ValueError("You need to pass one of the following parameter photo/video/forward_from_chat_id!")