Fix None FileName in mirroring some TG File (#6)

This commit is contained in:
Hafitz Setya 2021-02-24 19:55:10 +07:00 committed by GitHub
parent 78a3f13f92
commit 37b30cd10f
Signed by untrusted user: GitHub
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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'<a href="tg://user?id={self.message.from_user.id}">{self.message.from_user.first_name}</a>'
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: