Plugins: gdrive: don't use custom file name if it's None (adekmaulana#31

)
originalnobody2745 · Sep 2, 2021 · 8f20a6b · 8f20a6b
1 parent 007bf3c
commit 8f20a6b
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/bot/plugins/gdrive.py b/bot/plugins/gdrive.py
@@ -383,10 +383,13 @@ def prog_func(current: int, total: int) -> None:
 
                 last_update_time = now
 
-        file_path = download_path / file_name
-        file_path = await ctx.bot.client.download_media(msg,
-                                                        file_name=file_path,
-                                                        progress=prog_func)  # type: ignore
+        if file_name is None:
+            file_path = await ctx.bot.client.download_media(msg, progress=prog_func)
+        else:
+            file_path = f"{download_path}/{file_name}"
+            file_path = await ctx.bot.client.download_media(msg,
+                                                            file_name=file_path,
+                                                            progress=prog_func)  # type: ignore
 
         if file_path is not None:
             return AsyncPath(file_path)