239 lines
7.1 KiB
Python
239 lines
7.1 KiB
Python
import logging
|
|
import os
|
|
import threading
|
|
import time
|
|
import random
|
|
import string
|
|
import aria2p
|
|
import telegram.ext as tg
|
|
from dotenv import load_dotenv
|
|
from pyrogram import Client
|
|
from telegraph import Telegraph
|
|
import socket
|
|
from megasdkrestclient import MegaSdkRestClient, errors as mega_err
|
|
import subprocess
|
|
|
|
socket.setdefaulttimeout(600)
|
|
|
|
botStartTime = time.time()
|
|
if os.path.exists('log.txt'):
|
|
with open('log.txt', 'r+') as f:
|
|
f.truncate(0)
|
|
|
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
handlers=[logging.FileHandler('log.txt'), logging.StreamHandler()],
|
|
level=logging.INFO)
|
|
|
|
load_dotenv('config.env')
|
|
|
|
Interval = []
|
|
|
|
|
|
def getConfig(name: str):
|
|
return os.environ[name]
|
|
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
try:
|
|
if bool(getConfig('_____REMOVE_THIS_LINE_____')):
|
|
logging.error('The README.md file there to be read! Exiting now!')
|
|
exit()
|
|
except KeyError:
|
|
pass
|
|
|
|
aria2 = aria2p.API(
|
|
aria2p.Client(
|
|
host="http://localhost",
|
|
port=6800,
|
|
secret="",
|
|
)
|
|
)
|
|
|
|
DOWNLOAD_DIR = None
|
|
BOT_TOKEN = None
|
|
|
|
download_dict_lock = threading.Lock()
|
|
status_reply_dict_lock = threading.Lock()
|
|
# Key: update.effective_chat.id
|
|
# Value: telegram.Message
|
|
status_reply_dict = {}
|
|
# Key: update.message.message_id
|
|
# Value: An object of Status
|
|
download_dict = {}
|
|
# Stores list of users and chats the bot is authorized to use in
|
|
AUTHORIZED_CHATS = set()
|
|
if os.path.exists('authorized_chats.txt'):
|
|
with open('authorized_chats.txt', 'r+') as f:
|
|
lines = f.readlines()
|
|
for line in lines:
|
|
# LOGGER.info(line.split())
|
|
AUTHORIZED_CHATS.add(int(line.split()[0]))
|
|
try:
|
|
achats = getConfig('AUTHORIZED_CHATS')
|
|
achats = achats.split(" ")
|
|
for chats in achats:
|
|
AUTHORIZED_CHATS.add(int(chats))
|
|
except:
|
|
pass
|
|
|
|
try:
|
|
BOT_TOKEN = getConfig('BOT_TOKEN')
|
|
parent_id = getConfig('GDRIVE_FOLDER_ID')
|
|
DOWNLOAD_DIR = getConfig('DOWNLOAD_DIR')
|
|
if DOWNLOAD_DIR[-1] != '/' or DOWNLOAD_DIR[-1] != '\\':
|
|
DOWNLOAD_DIR = DOWNLOAD_DIR + '/'
|
|
DOWNLOAD_STATUS_UPDATE_INTERVAL = int(getConfig('DOWNLOAD_STATUS_UPDATE_INTERVAL'))
|
|
OWNER_ID = int(getConfig('OWNER_ID'))
|
|
AUTO_DELETE_MESSAGE_DURATION = int(getConfig('AUTO_DELETE_MESSAGE_DURATION'))
|
|
TELEGRAM_API = getConfig('TELEGRAM_API')
|
|
TELEGRAM_HASH = getConfig('TELEGRAM_HASH')
|
|
except KeyError as e:
|
|
LOGGER.error("One or more env variables missing! Exiting now")
|
|
exit(1)
|
|
|
|
LOGGER.info("Generating USER_SESSION_STRING")
|
|
with Client(':memory:', api_id=int(TELEGRAM_API), api_hash=TELEGRAM_HASH, bot_token=BOT_TOKEN) as app:
|
|
USER_SESSION_STRING = app.export_session_string()
|
|
|
|
#Generate Telegraph Token
|
|
sname = ''.join(random.SystemRandom().choices(string.ascii_letters, k=8))
|
|
LOGGER.info("Generating Telegraph Token using '" + sname + "' name")
|
|
telegraph = Telegraph()
|
|
telegraph.create_account(short_name=sname)
|
|
telegraph_token = telegraph.get_access_token()
|
|
LOGGER.info("Telegraph Token Generated: '" + telegraph_token + "'")
|
|
|
|
try:
|
|
MEGA_KEY = getConfig('MEGA_KEY')
|
|
|
|
except KeyError:
|
|
MEGA_KEY = None
|
|
LOGGER.info('MEGA API KEY NOT AVAILABLE')
|
|
if MEGA_KEY is not None:
|
|
# Start megasdkrest binary
|
|
subprocess.Popen(["megasdkrest", "--apikey", MEGA_KEY])
|
|
time.sleep(3) # Wait for the mega server to start listening
|
|
mega_client = MegaSdkRestClient('http://localhost:6090')
|
|
try:
|
|
MEGA_USERNAME = getConfig('MEGA_USERNAME')
|
|
MEGA_PASSWORD = getConfig('MEGA_PASSWORD')
|
|
if len(MEGA_USERNAME) > 0 and len(MEGA_PASSWORD) > 0:
|
|
try:
|
|
mega_client.login(MEGA_USERNAME, MEGA_PASSWORD)
|
|
except mega_err.MegaSdkRestClientException as e:
|
|
logging.error(e.message['message'])
|
|
exit(0)
|
|
else:
|
|
LOGGER.info("Mega API KEY provided but credentials not provided. Starting mega in anonymous mode!")
|
|
MEGA_USERNAME = None
|
|
MEGA_PASSWORD = None
|
|
except KeyError:
|
|
LOGGER.info("Mega API KEY provided but credentials not provided. Starting mega in anonymous mode!")
|
|
MEGA_USERNAME = None
|
|
MEGA_PASSWORD = None
|
|
else:
|
|
MEGA_USERNAME = None
|
|
MEGA_PASSWORD = None
|
|
try:
|
|
HEROKU_API_KEY = getConfig('HEROKU_API_KEY')
|
|
except KeyError:
|
|
logging.warning('HEROKU API KEY not provided!')
|
|
HEROKU_API_KEY = None
|
|
try:
|
|
HEROKU_APP_NAME = getConfig('HEROKU_APP_NAME')
|
|
except KeyError:
|
|
logging.warning('HEROKU APP NAME not provided!')
|
|
HEROKU_APP_NAME = None
|
|
try:
|
|
MAX_TORRENT_SIZE = int(getConfig("MAX_TORRENT_SIZE"))
|
|
except KeyError:
|
|
MAX_TORRENT_SIZE = None
|
|
try:
|
|
ENABLE_FILESIZE_LIMIT = getConfig('ENABLE_FILESIZE_LIMIT')
|
|
if ENABLE_FILESIZE_LIMIT.lower() == 'true':
|
|
ENABLE_FILESIZE_LIMIT = True
|
|
else:
|
|
ENABLE_FILESIZE_LIMIT = False
|
|
except KeyError:
|
|
ENABLE_FILESIZE_LIMIT = False
|
|
try:
|
|
UPTOBOX_TOKEN = getConfig('UPTOBOX_TOKEN')
|
|
except KeyError:
|
|
logging.warning('UPTOBOX_TOKEN not provided!')
|
|
UPTOBOX_TOKEN = None
|
|
try:
|
|
INDEX_URL = getConfig('INDEX_URL')
|
|
if len(INDEX_URL) == 0:
|
|
INDEX_URL = None
|
|
except KeyError:
|
|
INDEX_URL = None
|
|
try:
|
|
BUTTON_THREE_NAME = getConfig('BUTTON_THREE_NAME')
|
|
BUTTON_THREE_URL = getConfig('BUTTON_THREE_URL')
|
|
if len(BUTTON_THREE_NAME) == 0 or len(BUTTON_THREE_URL) == 0:
|
|
raise KeyError
|
|
except KeyError:
|
|
BUTTON_THREE_NAME = None
|
|
BUTTON_THREE_URL = None
|
|
try:
|
|
BUTTON_FOUR_NAME = getConfig('BUTTON_FOUR_NAME')
|
|
BUTTON_FOUR_URL = getConfig('BUTTON_FOUR_URL')
|
|
if len(BUTTON_FOUR_NAME) == 0 or len(BUTTON_FOUR_URL) == 0:
|
|
raise KeyError
|
|
except KeyError:
|
|
BUTTON_FOUR_NAME = None
|
|
BUTTON_FOUR_URL = None
|
|
try:
|
|
BUTTON_FIVE_NAME = getConfig('BUTTON_FIVE_NAME')
|
|
BUTTON_FIVE_URL = getConfig('BUTTON_FIVE_URL')
|
|
if len(BUTTON_FIVE_NAME) == 0 or len(BUTTON_FIVE_URL) == 0:
|
|
raise KeyError
|
|
except KeyError:
|
|
BUTTON_FIVE_NAME = None
|
|
BUTTON_FIVE_URL = None
|
|
try:
|
|
STOP_DUPLICATE_MIRROR = getConfig('STOP_DUPLICATE_MIRROR')
|
|
if STOP_DUPLICATE_MIRROR.lower() == 'true':
|
|
STOP_DUPLICATE_MIRROR = True
|
|
else:
|
|
STOP_DUPLICATE_MIRROR = False
|
|
except KeyError:
|
|
STOP_DUPLICATE_MIRROR = False
|
|
try:
|
|
IS_TEAM_DRIVE = getConfig('IS_TEAM_DRIVE')
|
|
if IS_TEAM_DRIVE.lower() == 'true':
|
|
IS_TEAM_DRIVE = True
|
|
else:
|
|
IS_TEAM_DRIVE = False
|
|
except KeyError:
|
|
IS_TEAM_DRIVE = False
|
|
try:
|
|
USE_SERVICE_ACCOUNTS = getConfig('USE_SERVICE_ACCOUNTS')
|
|
if USE_SERVICE_ACCOUNTS.lower() == 'true':
|
|
USE_SERVICE_ACCOUNTS = True
|
|
else:
|
|
USE_SERVICE_ACCOUNTS = False
|
|
except KeyError:
|
|
USE_SERVICE_ACCOUNTS = False
|
|
try:
|
|
BLOCK_MEGA_LINKS = getConfig('BLOCK_MEGA_LINKS')
|
|
if BLOCK_MEGA_LINKS.lower() == 'true':
|
|
BLOCK_MEGA_LINKS = True
|
|
else:
|
|
BLOCK_MEGA_LINKS = False
|
|
except KeyError:
|
|
BLOCK_MEGA_LINKS = False
|
|
try:
|
|
SHORTENER = getConfig('SHORTENER')
|
|
SHORTENER_API = getConfig('SHORTENER_API')
|
|
if len(SHORTENER) == 0 or len(SHORTENER_API) == 0:
|
|
raise KeyError
|
|
except KeyError:
|
|
SHORTENER = None
|
|
SHORTENER_API = None
|
|
|
|
updater = tg.Updater(token=BOT_TOKEN,use_context=True)
|
|
bot = updater.bot
|
|
dispatcher = updater.dispatcher
|