Initial
This commit is contained in:
jaskaranSM
commit
e826385571
|
|
@ -0,0 +1,97 @@
|
|||
import aria2p
|
||||
import os
|
||||
from time import sleep
|
||||
import logging
|
||||
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||
level=logging.INFO)
|
||||
|
||||
|
||||
|
||||
cmd = "aria2c --enable-rpc --rpc-listen-all=false --rpc-listen-port 6800 --max-connection-per-server=10 --rpc-max-request-size=1024M --seed-time=0.01 --min-split-size=10M --follow-torrent=mem --split=10 --daemon=true --allow-overwrite=true"
|
||||
EDIT_SLEEP_TIME_OUT = 5
|
||||
aria2_is_running = os.system(cmd)
|
||||
|
||||
aria2 = aria2p.API(
|
||||
aria2p.Client(
|
||||
host="http://localhost",
|
||||
port=6800,
|
||||
secret=""
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
allDls = {}
|
||||
|
||||
def check_metadata(gid):
|
||||
file = aria2.get_download(gid)
|
||||
if file.followed_by_ids[0] != None:
|
||||
new_gid = file.followed_by_ids[0]
|
||||
logging.info("Changing GID "+gid+" to "+new_gid)
|
||||
return new_gid
|
||||
else:
|
||||
return False
|
||||
|
||||
|
||||
def add_download(link,message):
|
||||
if "magnet" in link:
|
||||
download = aria2.add_magnet(link)
|
||||
allDls[message[0]] = [download,message[1]]
|
||||
logging.info("Adding: "+link)
|
||||
return download
|
||||
else:
|
||||
download = aria2.add_uris([link])
|
||||
allDls[message[0]] = [download,message[1]]
|
||||
logging.info("Adding: "+link)
|
||||
return download
|
||||
|
||||
|
||||
def remove_download(message):
|
||||
del allDls[message]
|
||||
aria2.remove_download(allDls[message].gid)
|
||||
return True
|
||||
|
||||
|
||||
|
||||
def get_download_by_message(message):
|
||||
if allDls[message] == None:
|
||||
return None
|
||||
else:
|
||||
download = allDls[message]
|
||||
return download[0]
|
||||
|
||||
|
||||
|
||||
def progress_status(context,update,previous):
|
||||
download = get_download_by_message(update)
|
||||
file = aria2.get_download(download.gid)
|
||||
if not file.is_complete:
|
||||
if not file.error_message:
|
||||
msg = "<i>"+str(file.name) +"</i>:- " +str(file.progress_string())+" of "+str(file.total_length_string())+" at "+str(file.download_speed_string())+" ,ETA: "+str(file.eta_string())
|
||||
if previous != msg:
|
||||
print("editing message")
|
||||
try:
|
||||
context.bot.edit_message_text(text=msg,message_id=update.message_id,chat_id=update.chat.id,parse_mode='HTMl')
|
||||
except:
|
||||
pass
|
||||
previous = msg
|
||||
sleep(5)
|
||||
progress_status(context,update,previous)
|
||||
else:
|
||||
logging.error(file.error_message)
|
||||
return
|
||||
else:
|
||||
try:
|
||||
new_gid = check_metadata(file.gid)
|
||||
except:
|
||||
new_gid = None
|
||||
pass
|
||||
if new_gid:
|
||||
download = aria2.get_download(new_gid)
|
||||
allDls[update][0] = download
|
||||
progress_status(context,update,previous=None)
|
||||
else:
|
||||
logging.info(file.name+" Completed.")
|
||||
msg = "<i>"+str(file.name) +"</i>:- Uploading."
|
||||
context.bot.edit_message_text(text=msg,message_id=update.message_id,chat_id=update.chat.id,parse_mode='HTMl')
|
||||
with open('data','w') as f:
|
||||
f.write(file.name)
|
||||
|
|
@ -0,0 +1,50 @@
|
|||
from telegram.ext import Updater,CommandHandler,run_async
|
||||
import ariaTools
|
||||
import logging
|
||||
import gdriveTools
|
||||
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||
level=logging.INFO)
|
||||
|
||||
|
||||
|
||||
|
||||
@run_async
|
||||
def start(update, context):
|
||||
print(update)
|
||||
context.bot.send_message(chat_id=update.message.chat_id, text="I'm a bot, please talk to me!")
|
||||
|
||||
@run_async
|
||||
def mirror(update,context):
|
||||
message = update.message.text
|
||||
link = message.replace('/mirror','')[1:]
|
||||
reply_msg = context.bot.send_message(chat_id=update.message.chat_id, text="Starting Download")
|
||||
download = ariaTools.add_download(link,[reply_msg,update])
|
||||
ariaTools.progress_status(context,reply_msg,previous=None)
|
||||
with open('data','r') as f:
|
||||
file_name = f.read()
|
||||
print("File-Name: "+file_name)
|
||||
link = gdriveTools.upload(file_name)
|
||||
# with open('data','r') as f:
|
||||
# link = f.read()
|
||||
msg = "<i>"+str(file_name) +"</i>\n"+link
|
||||
context.bot.edit_message_text(text=msg,message_id=reply_msg.message_id,chat_id=reply_msg.chat.id,parse_mode='HTMl')
|
||||
|
||||
|
||||
def main():
|
||||
BOT_TOKEN = "976868081:AAEO--j0dqomyy0ZOYD0sSIGim4UHTYQg5E"
|
||||
updater = Updater(token=BOT_TOKEN, use_context=True)
|
||||
|
||||
start_handler = CommandHandler('start', start)
|
||||
mirror_handler = CommandHandler('mirror',mirror)
|
||||
dispatcher = updater.dispatcher
|
||||
dispatcher.add_handler(start_handler)
|
||||
dispatcher.add_handler(mirror_handler)
|
||||
logging.info("Bot Started")
|
||||
|
||||
|
||||
|
||||
|
||||
updater.start_polling()
|
||||
|
||||
|
||||
main()
|
||||
|
|
@ -0,0 +1,5 @@
|
|||
class Config(object):
|
||||
"""docstring for Config"""
|
||||
G_DRIVE_CLIENT_ID = ""
|
||||
G_DRIVE_CLIENT_SECRET = ""
|
||||
GDRIVE_FOLDER_ID = ""
|
||||
|
|
@ -0,0 +1,182 @@
|
|||
from apiclient.discovery import build
|
||||
from apiclient.http import MediaFileUpload
|
||||
from apiclient.errors import ResumableUploadError
|
||||
from oauth2client.client import OAuth2WebServerFlow
|
||||
from oauth2client.file import Storage
|
||||
from oauth2client import file, client, tools
|
||||
from mimetypes import guess_type
|
||||
import httplib2
|
||||
import os
|
||||
from config import Config
|
||||
import logging
|
||||
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||
level=logging.INFO)
|
||||
|
||||
|
||||
G_DRIVE_TOKEN_FILE = "auth_token.txt"
|
||||
# Copy your credentials from the APIs Console
|
||||
CLIENT_ID = Config.G_DRIVE_CLIENT_ID
|
||||
CLIENT_SECRET = Config.G_DRIVE_CLIENT_SECRET
|
||||
# Check https://developers.google.com/drive/scopes for all available scopes
|
||||
OAUTH_SCOPE = "https://www.googleapis.com/auth/drive.file"
|
||||
# Redirect URI for installed apps, can be left as is
|
||||
REDIRECT_URI = "urn:ietf:wg:oauth:2.0:oob"
|
||||
parent_id = Config.GDRIVE_FOLDER_ID
|
||||
G_DRIVE_DIR_MIME_TYPE = "application/vnd.google-apps.folder"
|
||||
|
||||
|
||||
if CLIENT_ID is None or CLIENT_SECRET is None or parent_id is None:
|
||||
logging.error("Please Setup Config Properly.")
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
def upload(fileName):
|
||||
try:
|
||||
with open(G_DRIVE_TOKEN_FILE) as f:
|
||||
pass
|
||||
except IOError:
|
||||
storage = create_token_file(G_DRIVE_TOKEN_FILE)
|
||||
http = authorize(G_DRIVE_TOKEN_FILE, storage)
|
||||
print("Uploading File: "+fileName)
|
||||
if os.path.isfile(fileName):
|
||||
http = authorize(G_DRIVE_TOKEN_FILE, None)
|
||||
file_name, mime_type = file_ops(fileName)
|
||||
try:
|
||||
g_drive_link = upload_file(http, file_name,file_name, mime_type,parent_id)
|
||||
logging.info("Uploaded To G-Drive: "+fileName)
|
||||
link = g_drive_link
|
||||
except Exception as e:
|
||||
logging.error(str(e))
|
||||
pass
|
||||
else:
|
||||
http = authorize(G_DRIVE_TOKEN_FILE, None)
|
||||
file_name, mime_type = file_ops(fileName)
|
||||
try:
|
||||
dir_id = create_directory(http, os.path.basename(os.path.abspath(fileName)), parent_id)
|
||||
DoTeskWithDir(http,fileName, dir_id)
|
||||
logging.info("Uploaded To G-Drive: "+fileName)
|
||||
dir_link = "https://drive.google.com/folderview?id={}".format(dir_id)
|
||||
link = dir_link
|
||||
except Exception as e:
|
||||
logging.error(str(e))
|
||||
pass
|
||||
return link
|
||||
# with open('data','w') as f:
|
||||
# f.write(link)
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
def create_directory(http, directory_name, parent_id):
|
||||
drive_service = build("drive", "v2", http=http, cache_discovery=False)
|
||||
permissions = {
|
||||
"role": "reader",
|
||||
"type": "anyone",
|
||||
"value": None,
|
||||
"withLink": True
|
||||
}
|
||||
file_metadata = {
|
||||
"title": directory_name,
|
||||
"mimeType": G_DRIVE_DIR_MIME_TYPE
|
||||
}
|
||||
if parent_id is not None:
|
||||
file_metadata["parents"] = [{"id": parent_id}]
|
||||
file = drive_service.files().insert(body=file_metadata).execute()
|
||||
file_id = file.get("id")
|
||||
drive_service.permissions().insert(fileId=file_id, body=permissions).execute()
|
||||
logging.info("Created Gdrive Folder:\nName: {}\nID: {} ".format(file.get("title"), file_id))
|
||||
return file_id
|
||||
|
||||
|
||||
def DoTeskWithDir(http, input_directory, parent_id):
|
||||
list_dirs = os.listdir(input_directory)
|
||||
if len(list_dirs) == 0:
|
||||
return parent_id
|
||||
r_p_id = None
|
||||
for a_c_f_name in list_dirs:
|
||||
current_file_name = os.path.join(input_directory, a_c_f_name)
|
||||
if os.path.isdir(current_file_name):
|
||||
current_dir_id = create_directory(http, a_c_f_name, parent_id)
|
||||
r_p_id = DoTeskWithDir(http, current_file_name,current_dir_id)
|
||||
else:
|
||||
file_name, mime_type = file_ops(current_file_name)
|
||||
# current_file_name will have the full path
|
||||
g_drive_link = upload_file(http, current_file_name, file_name, mime_type, parent_id)
|
||||
r_p_id = parent_id
|
||||
# TODO: there is a #bug here :(
|
||||
return r_p_id
|
||||
|
||||
def file_ops(file_path):
|
||||
mime_type = guess_type(file_path)[0]
|
||||
mime_type = mime_type if mime_type else "text/plain"
|
||||
file_name = file_path.split("/")[-1]
|
||||
return file_name, mime_type
|
||||
|
||||
|
||||
def create_token_file(token_file):
|
||||
# Run through the OAuth flow and retrieve credentials
|
||||
flow = OAuth2WebServerFlow(
|
||||
CLIENT_ID,
|
||||
CLIENT_SECRET,
|
||||
OAUTH_SCOPE,
|
||||
redirect_uri=REDIRECT_URI
|
||||
)
|
||||
authorize_url = flow.step1_get_authorize_url()
|
||||
print('Go to the following link in your browser: ' + authorize_url)
|
||||
code = input('Enter verification code: ').strip()
|
||||
credentials = flow.step2_exchange(code)
|
||||
storage = Storage(token_file)
|
||||
storage.put(credentials)
|
||||
return storage
|
||||
|
||||
def authorize(token_file, storage):
|
||||
# Get credentials
|
||||
if storage is None:
|
||||
storage = Storage(token_file)
|
||||
credentials = storage.get()
|
||||
# Create an httplib2.Http object and authorize it with our credentials
|
||||
http = httplib2.Http()
|
||||
credentials.refresh(http)
|
||||
http = credentials.authorize(http)
|
||||
return http
|
||||
|
||||
|
||||
|
||||
|
||||
def upload_file(http, file_path,file_name, mime_type, parent_id):
|
||||
# Create Google Drive service instance
|
||||
drive_service = build('drive', 'v2', http=http, cache_discovery=False)
|
||||
# File body description
|
||||
media_body = MediaFileUpload(file_path,
|
||||
mimetype=mime_type,
|
||||
resumable=True)
|
||||
body = {
|
||||
'title': file_name,
|
||||
'description': 'backup',
|
||||
'mimeType': mime_type,
|
||||
}
|
||||
if parent_id is not None:
|
||||
body["parents"] = [{"id": parent_id}]
|
||||
# Permissions body description: anyone who has link can upload
|
||||
# Other permissions can be found at https://developers.google.com/drive/v2/reference/permissions
|
||||
permissions = {
|
||||
'role': 'reader',
|
||||
'type': 'anyone',
|
||||
'value': None,
|
||||
'withLink': True
|
||||
}
|
||||
# Insert a file
|
||||
file = drive_service.files().insert(body=body, media_body=media_body).execute()
|
||||
# Insert new permissions
|
||||
drive_service.permissions().insert(fileId=file['id'], body=permissions).execute()
|
||||
# Define file instance and get url for download
|
||||
file = drive_service.files().get(fileId=file['id']).execute()
|
||||
download_url = file.get('webContentLink')
|
||||
return download_url
|
||||
|
||||
|
||||
|
||||
|
|
@ -0,0 +1,13 @@
|
|||
google-api-python-client>=1.7.3
|
||||
google-auth>=1.5.0
|
||||
google-auth-httplib2<=0.0.3
|
||||
httplib2>=0.11.3
|
||||
more-itertools>=4.2.0
|
||||
ndg-httpsclient>=0.4.0
|
||||
oauth>=1.0.1
|
||||
pyasn1>=0.4.3
|
||||
pyasn1-modules>=0.2.1
|
||||
pycurl>=7.43.0
|
||||
uritemplate>=3.0.0
|
||||
urllib3>=1.23
|
||||
|
||||
Loading…
Reference in New Issue