Pyrofork: Move duplicated codes to separate method

Signed-off-by: wulan17 <wulan17@nusantararom.org>
Mayuri-Chan · Jan 4, 2024 · 04cafe3 · 04cafe3
1 parent a882a99
commit 04cafe3
Showing 1 changed file with 21 additions and 26 deletions.
diff --git a/pyrogram/methods/users/send_story.py b/pyrogram/methods/users/send_story.py
@@ -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 @@ async def send_story(
                     )
                 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!")