211 lines
7.2 KiB
Python
211 lines
7.2 KiB
Python
# Ultroid - UserBot
|
|
# Copyright (C) 2020 TeamUltroid
|
|
#
|
|
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
|
|
# PLease read the GNU Affero General Public License in
|
|
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>.
|
|
|
|
|
|
"""
|
|
✘ Commands Available
|
|
|
|
• `{i}ulgdrive <reply/file name>`
|
|
Reply to file to upload on Google Drive.
|
|
Add file name to upload on Google Drive.
|
|
|
|
• `{i}drivesearch <file name>`
|
|
Search file name on Google Drive and get link.
|
|
|
|
• `{i}uldir <directory name>`
|
|
Upload a directory on Google Drive.
|
|
|
|
• `{i}gfolder`
|
|
Link to your Google Drive Folder.
|
|
If added then all uploaded files will be placed here.
|
|
"""
|
|
|
|
|
|
import asyncio
|
|
import os
|
|
import time
|
|
from datetime import datetime
|
|
|
|
from telethon import events
|
|
|
|
from . import *
|
|
|
|
TOKEN_FILE = "resources/downloads/auth_token.txt"
|
|
|
|
|
|
@asst_cmd("auth")
|
|
async def aut(event):
|
|
if event.is_group:
|
|
return
|
|
if os.path.exists(TOKEN_FILE):
|
|
return await event.reply("`You have already authorised with Google Drive`")
|
|
if Redis("GDRIVE_CLIENT_ID") == None or Redis("GDRIVE_CLIENT_SECRET") == None:
|
|
await event.reply(
|
|
"Go [here](https://console.developers.google.com/flows/enableapi?apiid=drive) and get your GDRIVE_CLIENT_ID and GDRIVE_CLIENT_SECRET\n\n"
|
|
+ "Send your GDRIVE_CLIENT_ID and GDRIVE_CLIENT_SECRET as this.\n`GDRIVE_CLIENT_ID GDRIVE_CLIENT_SECRET` separated by space.",
|
|
)
|
|
async with asst.conversation(ultroid_bot.uid) as conv:
|
|
reply = conv.wait_event(events.NewMessage(from_users=ultroid_bot.uid))
|
|
repl = await reply
|
|
try:
|
|
creds = repl.text.split(" ")
|
|
id = creds[0]
|
|
if not id.endswith("com"):
|
|
return await event.reply("`Wrong Client Id`")
|
|
try:
|
|
secret = creds[1]
|
|
except IndexError:
|
|
return await event.reply("`No Client Secret Found`")
|
|
udB.set("GDRIVE_CLIENT_ID", id)
|
|
udB.set("GDRIVE_CLIENT_SECRET", secret)
|
|
return await repl.reply("`Success!` Now send /auth again")
|
|
except Exception as exx:
|
|
return await repl.reply(
|
|
"`Something went wrong! Send `/auth` again.\nIf same happens contact `@TheUltroid"
|
|
)
|
|
else:
|
|
storage = await create_token_file(TOKEN_FILE, event)
|
|
http = authorize(TOKEN_FILE, storage)
|
|
f = open(TOKEN_FILE, "r")
|
|
token_file_data = f.read()
|
|
udB.set("GDRIVE_TOKEN", token_file_data)
|
|
await event.reply(
|
|
"`Success!\nYou are all set to use Google Drive with Ultroid Userbot.`"
|
|
)
|
|
|
|
|
|
@ultroid_cmd(
|
|
pattern="ulgdrive ?(.*)",
|
|
)
|
|
async def _(event):
|
|
mone = await eor(event, "Processing ...")
|
|
if not os.path.exists(TOKEN_FILE):
|
|
return await eod(mone, f"`Go to `{Var.BOT_USERNAME}` and send ``/auth.`")
|
|
input_str = event.pattern_match.group(1)
|
|
required_file_name = None
|
|
start = datetime.now()
|
|
dddd = time.time()
|
|
if event.reply_to_msg_id and not input_str:
|
|
reply_message = await event.get_reply_message()
|
|
try:
|
|
downloaded_file_name = await event.client.download_media(
|
|
reply_message,
|
|
"resources/downloads",
|
|
progress_callback=lambda d, t: asyncio.get_event_loop().create_task(
|
|
progress(
|
|
d,
|
|
t,
|
|
mone,
|
|
dddd,
|
|
"Downloading...",
|
|
),
|
|
),
|
|
)
|
|
except Exception as e:
|
|
return await eod(mone, str(e), time=10)
|
|
end = datetime.now()
|
|
ms = (end - start).seconds
|
|
required_file_name = downloaded_file_name
|
|
await mone.edit(
|
|
"Downloaded to `{}` in {} seconds.".format(downloaded_file_name, ms)
|
|
)
|
|
elif input_str:
|
|
input_str = input_str.strip()
|
|
if os.path.exists(input_str):
|
|
end = datetime.now()
|
|
ms = (end - start).seconds
|
|
required_file_name = input_str
|
|
await mone.edit("Found `{}` in {} seconds.".format(input_str, ms))
|
|
else:
|
|
return await eod(
|
|
mone, "File Not found in local server. Give me a file path :((", time=5
|
|
)
|
|
if required_file_name:
|
|
http = authorize(TOKEN_FILE, None)
|
|
file_name, mime_type = file_ops(required_file_name)
|
|
try:
|
|
g_drive_link = await upload_file(
|
|
http,
|
|
required_file_name,
|
|
file_name,
|
|
mime_type,
|
|
mone,
|
|
Redis("GDRIVE_FOLDER_ID"),
|
|
)
|
|
await mone.edit(
|
|
"**Successfully Uploaded File on G-Drive :**\n\n[{}]({})".format(
|
|
file_name, g_drive_link
|
|
)
|
|
)
|
|
except Exception as e:
|
|
await mone.edit(f"Exception occurred while uploading to gDrive {e}")
|
|
else:
|
|
return await eod(mone, "`File Not found in local server.`", time=10)
|
|
|
|
|
|
@ultroid_cmd(
|
|
pattern="drivesearch ?(.*)",
|
|
)
|
|
async def sch(event):
|
|
if not os.path.exists(TOKEN_FILE):
|
|
return await eod(mone, f"`Go to `{Var.BOT_USERNAME}` and send ``/auth.`")
|
|
http = authorize(TOKEN_FILE, None)
|
|
input_str = event.pattern_match.group(1).strip()
|
|
a = await eor(event, "Searching for {} in G-Drive.".format(input_str))
|
|
if Redis("GDRIVE_FOLDER_ID") is not None:
|
|
query = "'{}' in parents and (title contains '{}')".format(
|
|
Redis("GDRIVE_FOLDER_ID"), input_str
|
|
)
|
|
else:
|
|
query = "title contains '{}'".format(input_str)
|
|
try:
|
|
msg = await gsearch(http, query, input_str)
|
|
return await a.edit(str(msg))
|
|
except Exception as ex:
|
|
return await a.edit(str(ex))
|
|
|
|
|
|
@ultroid_cmd(
|
|
pattern="uldir ?(.*)",
|
|
)
|
|
async def _(event):
|
|
if not os.path.exists(TOKEN_FILE):
|
|
return await eod(mone, f"`Go to `{Var.BOT_USERNAME}` and send ``/auth.`")
|
|
input_str = event.pattern_match.group(1)
|
|
if os.path.isdir(input_str):
|
|
http = authorize(TOKEN_FILE, None)
|
|
a = await eor(event, "Uploading `{}` to G-Drive...".format(input_str))
|
|
dir_id = await create_directory(
|
|
http,
|
|
os.path.basename(os.path.abspath(input_str)),
|
|
Redis("GDRIVE_FOLDER_ID"),
|
|
)
|
|
await DoTeskWithDir(http, input_str, event, dir_id)
|
|
dir_link = "https://drive.google.com/folderview?id={}".format(dir_id)
|
|
await eod(
|
|
a,
|
|
f"**Successfully Uploaded Folder To G-Drive...**\n[{input_str}]({dir_link})",
|
|
)
|
|
else:
|
|
return await eod(event, f"Directory {input_str} does not seem to exist", time=5)
|
|
|
|
|
|
@ultroid_cmd(
|
|
pattern="gfolder$",
|
|
)
|
|
async def _(event):
|
|
if Redis("GDRIVE_FOLDER_ID"):
|
|
folder_link = "https://drive.google.com/folderview?id=" + Redis(
|
|
"GDRIVE_FOLDER_ID"
|
|
)
|
|
await eod(event, "`Here is Your G-Drive Folder link : `\n" + folder_link)
|
|
else:
|
|
await eod(event, "Set GDRIVE_FOLDER_ID with value of your folder id")
|
|
|
|
|
|
HELP.update({f"{__name__.split('.')[1]}": f"{__doc__.format(i=HNDLR)}"})
|