evanferrao
/
dump-mirrorbot
mirror of https://github.com/Box-boi/dump-mirrorbot.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Add Index URL in /list cmd 

Also simplified code a bit

Signed-off-by: lzzy12 <jhashivam2020@gmail.com>
This commit is contained in:
lzzy12 2019-12-25 15:35:48 +05:30
parent 53b6829235
commit 8e886a3030
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
bot/helper/mirror_utils/upload_utils/gdriveTools.py
View File

@ -4,7 +4,7 @@ from google.auth.transport.requests import Request
from googleapiclient.http import MediaFileUpload
import pickle
import os
from bot import LOGGER, parent_id, DOWNLOAD_DIR, DOWNLOAD_STATUS_UPDATE_INTERVAL, IS_TEAM_DRIVE
from bot import LOGGER, parent_id, DOWNLOAD_DIR, DOWNLOAD_STATUS_UPDATE_INTERVAL, IS_TEAM_DRIVE, INDEX_URL
from bot.helper.ext_utils.fs_utils import get_mime_type
from bot.helper.ext_utils.bot_utils import *
@ -234,19 +234,23 @@ class GoogleDriveHelper:
pageToken=page_token,
orderBy='modifiedTime desc').execute()
for file in response.get('files', []):
if len(results) >= 20:
break
if file.get(
'mimeType') == "application/vnd.google-apps.folder": # Detect Whether Current Entity is a Folder or File.
if len(results) >= 20:
break
msg += f"⁍ <a href='https://drive.google.com/drive/folders/{file.get('id')}'>{file.get('name')}" \
f"</a> (folder)" + "\n"
results.append(file)
f"</a> (folder)"
if INDEX_URL is not None:
url = f'{INDEX_URL}/{file.get("name")}/'
msg += f'| <a href="{url}"> Index URL</a>'
else:
if len(results) >= 20:
break
msg += f"⁍ <a href='https://drive.google.com/uc?id={file.get('id')}" \
f"&export=download'>{file.get('name')}</a> ({get_readable_file_size(int(file.get('size')))})" + "\n"
results.append(file)
f"&export=download'>{file.get('name')}</a> ({get_readable_file_size(int(file.get('size')))})"
if INDEX_URL is not None:
url = f'{INDEX_URL}/{file.get("name")}'
msg += f' | <a href="{url}"> Index URL</a>'
msg += '\n'
results.append(file)
page_token = response.get('nextPageToken', None)
if page_token is None:
break