Skip to content

Commit

Permalink
Use Login inplace of cookies (#249)
Browse files Browse the repository at this point in the history 
* Use Login in place of cookies

* some fixes

* blank line contains white space

Co-authored-by: Krishna-singhal <65902764+Krishna-Singhal@users.noreply.github.com>
@ZekXtreme @Krishna-Singhal
ZekXtreme and Krishna-Singhal authored Feb 7, 2022
1 parent 18ce483 commit 3d32e4f
Showing 2 changed files with 59 additions and 56 deletions.
3 changes: 2 additions & 1 deletion config.env.sample
View file
Original file line number Diff line number Diff line change
@@ -28,7 +28,8 @@ G_PHOTOS_AUTH_TOKEN_ID = ""
# parser.py
# get these by reading https://t.me/UnofficialPluginsHelp/129
CRYPT = ""
APPDRIVE_MD = ""
APPDRIVE_EMAIL = "" # Website User Account (NOT GOOGLE ACCOUNT)
APPDRIVE_PASS = "" # Website User Account (NOT GOOGLE ACCOUNT)


# time.py
112 changes: 57 additions & 55 deletions plugins/parser.py
View file
Original file line number Diff line number Diff line change
@@ -5,34 +5,39 @@
import re
import json
import base64
from urllib.parse import urlparse

import requests
from bs4 import BeautifulSoup

from userge import Message, userge, pool

CRYPT = os.environ.get("CRYPT")
MD = os.environ.get("APPDRIVE_MD")
# Website User Account (NOT GOOGLE ACCOUNT)
APPDRIVE_EMAIL = os.environ.get("APPDRIVE_EMAIL")
APPDRIVE_PASS = os.environ.get("APPDRIVE_PASS")


async def _init():
global CRYPT, MD # pylint: disable=global-statement
global CRYPT # pylint: disable=global-statement
if CRYPT:
try:
crypt = json.loads(CRYPT)
except Exception:
pass # user entered only crypt value from dict
else:
CRYPT = crypt.get("cookie").split('=')[-1]
if MD:
try:
md = json.loads(MD)
except Exception:
pass # user entered only crypt value from dict
else:
MD = md.get("cookie").split(';')[0].split('=')[-1]


def gen_data_string(data, boundary=f'{"-"*6}_'):
async def account_login(client, url):
data = {
'email': APPDRIVE_EMAIL,
'password': APPDRIVE_PASS
}
await pool.run_in_thread(client.post)(
f'https://{urlparse(url).netloc}/login', data=data)


def gen_payload(data, boundary=f'{"-"*6}_'):
data_string = ''
for item in data:
data_string += f'{boundary}\r\n'
@@ -52,63 +57,62 @@ def parse_info(data):


async def appdrive_dl(url):

client = requests.Session()
client.cookies.update({'MD': MD})

client.headers.update({
"user-agent": ("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) "
"AppleWebKit/537.36 (KHTML, like Gecko) "
"Chrome/97.0.4692.99 Safari/537.36")
})
await account_login(client, url)
res = await pool.run_in_thread(client.get)(url)
key = re.findall(r'"key",\s+"(.*?)"', res.text)[0]
soup = BeautifulSoup(res.content, 'html.parser')
ddl_btn = soup.find('button', {'id': 'drc'})
info_parsed = parse_info(res.text)
info_parsed['error'] = False
info_parsed['link_type'] = 'login' # direct/login

headers = {
"Content-Type": f"multipart/form-data; boundary={'-'*4}_",
"user-agent": ("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) "
"AppleWebKit/537.36 (KHTML, like Gecko) "
"Chrome/97.0.4692.99 Safari/537.36")
}

data = {
'key': key,
'action': 'original'
'action': 'original',
'type': 1
}

if ddl_btn:
info_parsed['link_type'] = 'direct'
data['action'] = 'direct'
info_parsed['link_type'] = 'direct'

for i in range(1, 4):
data['type'] = i
try:
response = (await pool.run_in_thread(client.post)(
url,
data=gen_data_string(data),
headers=headers
)).json()
break
except Exception as e:
if i == 3:
response = {
'error': True,
'message': str(e)
}
if 'url' in response:
info_parsed['gdrive_link'] = response['url']

elif response.get('error'):
info_parsed['error'] = True
info_parsed['error_message'] = response['message']

while data['type'] <= 3:
try:
response = (await pool.run_in_thread(client.post)(
url,
data=gen_payload(data),
headers=headers
)).json()
break
except Exception as e:
if data['type'] == 3:
response = {
'error': True,
'message': str(e)
}
data['type'] += 1

if 'url' in response:
info_parsed['gdrive_link'] = response['url']
else:
info_parsed['error'] = True
info_parsed['error_message'] = response.get('message', 'Something went wrong :(')
return info_parsed


@userge.on_cmd("gdtot", about={
'header': "parse gdtot links",
'description': "you have to set <code>CRYPT</code>.\nget it by reading "
"<a href='https://t.me/UnofficialPluginsHelp/129'>help</a>.",
"<a href='https://t.me/UnofficialPluginsHelp/129'>Help</a>",
'usage': "{tr}gdtot gdtot_link"})
async def gdtot(message: Message):
""" Gets gdrive link """
@@ -145,28 +149,26 @@ async def gdtot(message: Message):

@userge.on_cmd("appdrive", about={
'header': "parse appdrive links",
'description': "you have to set <code>APPDRIVE_MD</code>.\nget it by reading "
"<a href='https://t.me/UnofficialPluginsHelp/129'>help</a>.",
'description': "you have to set <code>Required Vars</code>.\nget it by reading"
"<a href='https://t.me/UnofficialPluginsHelp/129'>Help</a>",
'usage': "{tr}appdrive appdrive_link"})
async def appdrive(message: Message):
if not MD:
if not (APPDRIVE_EMAIL or APPDRIVE_PASS):
return await message.err("read .help appdrive")
url = message.input_or_reply_str
url = message.input_str
if not url:
await message.err("Send a link along with command")
else:
try:
await message.edit("Parsing.....")
res = await appdrive_dl(url)
if res.get('error') and res.get('error_message'):
raise Exception(res.get('error_message'))
await message.edit("Parsing.....")
res = await appdrive_dl(url)
if res.get('error') and res.get('error_message'):
await message.err(res.get('error_message'))
else:
output = (
f'Title: {res.get("name")}\n'
f'Format: {res.get("format")}\n'
f'Size: {res.get("size")}\n'
'Drive_Link: '
f'{res.get("gdrive_link", "this link requires login to get drive_link")}'
f'{res.get("gdrive_link", "Something Went Wrong")}'
)
await message.edit(output, disable_web_page_preview=True)
except Exception as e:
await message.err(e)

0 comments on commit 3d32e4f

Please sign in to comment.