Fix count after upload while index link is none (#131)

breakdowns · Jun 5, 2021 · d23bcbd · d23bcbd
1 parent b709b8e
commit d23bcbd
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/bot/modules/mirror.py b/bot/modules/mirror.py
@@ -149,6 +149,12 @@ def onUploadProgress(self):
     def onUploadComplete(self, link: str, size, files, folders, typ):
         with download_dict_lock:
             msg = f'<b>Filename: </b><code>{download_dict[self.uid].name()}</code>\n<b>Size: </b><code>{size}</code>'
+            if os.path.isdir(f'{DOWNLOAD_DIR}/{self.uid}/{download_dict[self.uid].name()}'):
+                msg += '\n\n<b>Type: </b>Folder'
+                msg += f'\n<b>SubFolders: </b>{folders}'
+                msg += f'\n<b>Files: </b>{files}'
+            else:
+                msg += f'\n\n<b>Type: </b>{typ}'
             buttons = button_build.ButtonMaker()
             if SHORTENER is not None and SHORTENER_API is not None:
                 surl = requests.get('https://{}/api?api={}&url={}&format=text'.format(SHORTENER, SHORTENER_API, link)).text
@@ -161,11 +167,6 @@ def onUploadComplete(self, link: str, size, files, folders, typ):
                 share_url = f'{INDEX_URL}/{url_path}'
                 if os.path.isdir(f'{DOWNLOAD_DIR}/{self.uid}/{download_dict[self.uid].name()}'):
                     share_url += '/'
-                    msg += '\n\n<b>Type: </b>Folder'
-                    msg += f'\n<b>SubFolders: </b>{folders}'
-                    msg += f'\n<b>Files: </b>{files}'
-                else:
-                    msg += f'\n\n<b>Type: </b>{typ}'
                 if SHORTENER is not None and SHORTENER_API is not None:
                     siurl = requests.get('https://{}/api?api={}&url={}&format=text'.format(SHORTENER, SHORTENER_API, share_url)).text
                     buttons.buildbutton("⚡Index Link⚡", siurl)