save_files fixes

null-nick · null-nick · commit 583ce6d7ce2c · 2024-09-15T23:36:31.000+02:00
diff --git a/pyrogram/methods/advanced/save_file.py b/pyrogram/methods/advanced/save_file.py
@@ -128,10 +128,7 @@ async def worker(session):
             if file_size == 0:
                 raise ValueError("File size equals to 0 B")
 
-        # TODO
-        file_size_limit_mib = 2000
-        if self.me and self.me.is_premium:
-            file_size_limit_mib = 4000
+            file_size_limit_mib = 4000 if self.me and self.me.is_premium else 2000
 
             if file_size > file_size_limit_mib * 1024 * 1024:
                 raise ValueError(f"Can't upload files bigger than {file_size_limit_mib} MiB")
