diff --git a/bot/modules/mirror.py b/bot/modules/mirror.py index 6490937..2ef9517 100644 --- a/bot/modules/mirror.py +++ b/bot/modules/mirror.py @@ -90,6 +90,8 @@ class MirrorListener(listeners.MirrorListeners): else: path = f'{DOWNLOAD_DIR}{self.uid}/{name}' up_name = pathlib.PurePath(path).name + if up_name == "None": + up_name = "".join(os.listdir(f'{DOWNLOAD_DIR}{self.uid}/')) LOGGER.info(f"Upload Name : {up_name}") drive = gdriveTools.GoogleDriveHelper(up_name, self) if size == 0: @@ -146,7 +148,7 @@ class MirrorListener(listeners.MirrorListeners): else: uname = f'{self.message.from_user.first_name}' if uname is not None: - msg += f'\n\ncc : {uname}' + msg += f'\ncc : {uname}' try: fs_utils.clean_download(download_dict[self.uid].path()) except FileNotFoundError: