diff --git a/bot/helper/mirror_utils/upload_utils/gdriveTools.py b/bot/helper/mirror_utils/upload_utils/gdriveTools.py index 0731bf6..3adcb7d 100644 --- a/bot/helper/mirror_utils/upload_utils/gdriveTools.py +++ b/bot/helper/mirror_utils/upload_utils/gdriveTools.py @@ -7,11 +7,10 @@ from googleapiclient.discovery import build from googleapiclient.errors import HttpError from googleapiclient.http import MediaFileUpload from tenacity import * - from bot import LOGGER, parent_id, DOWNLOAD_DIR, IS_TEAM_DRIVE, INDEX_URL, DOWNLOAD_STATUS_UPDATE_INTERVAL from bot.helper.ext_utils.bot_utils import * from bot.helper.ext_utils.fs_utils import get_mime_type - +import requests logging.getLogger('googleapiclient.discovery').setLevel(logging.ERROR) @@ -239,13 +238,13 @@ class GoogleDriveHelper: msg += f"⁍ {file.get('name')}" \ f" (folder)" if INDEX_URL is not None: - url = f'{INDEX_URL}/{file.get("name")}/' + url = requests.utils.requote_uri(f'{INDEX_URL}/{file.get("name")}/') msg += f' | Index URL' else: msg += f"⁍ {file.get('name')} ({get_readable_file_size(int(file.get('size')))})" if INDEX_URL is not None: - url = f'{INDEX_URL}/{file.get("name")}' + url = requests.utils.requote_uri(f'{INDEX_URL}/{file.get("name")}') msg += f' | Index URL' msg += '\n' results.append(file) diff --git a/bot/modules/mirror.py b/bot/modules/mirror.py index 32d03e5..0dea2e1 100644 --- a/bot/modules/mirror.py +++ b/bot/modules/mirror.py @@ -15,6 +15,7 @@ import pathlib import os from bot.helper.mirror_utils.download_utils.direct_link_generator import direct_link_generator from bot.helper.ext_utils.exceptions import DirectDownloadLinkException +import requests class MirrorListener(listeners.MirrorListeners): @@ -96,8 +97,7 @@ class MirrorListener(listeners.MirrorListeners): msg = f'{download_dict[self.uid].name()} ({download_dict[self.uid].size()})' LOGGER.info(f'Done Uploading {download_dict[self.uid].name()}') if INDEX_URL is not None: - share_url = f'{INDEX_URL}/{download_dict[self.uid].name()}' - share_url = share_url.replace(' ', '%20') + share_url = requests.utils.requote_uri(f'{INDEX_URL}/{download_dict[self.uid].name()}') if os.path.isdir(f'{DOWNLOAD_DIR}/{self.uid}/{download_dict[self.uid].name()}'): share_url += '/' msg += f'\n\n Shareable link: here'