2022-07-01 14:28:39 +04:00
|
|
|
"""Stores API connection session and sends requests"""
|
|
|
|
|
|
2021-10-08 19:35:20 +04:00
|
|
|
import re
|
2022-08-22 09:55:08 +04:00
|
|
|
import time
|
2022-10-05 19:24:00 +04:00
|
|
|
import secrets
|
2022-03-30 19:40:58 +04:00
|
|
|
import logging
|
2022-08-22 09:55:08 +04:00
|
|
|
from functools import partial
|
|
|
|
|
|
2022-10-05 19:24:00 +04:00
|
|
|
from typing import Optional
|
MkDocs, Readme, Files API, Automated session saving, v2.0.1
MkDocs: sphinx docstrings rewritten to google, improved config, written the major part of how-to.
Readme: centered title + logo, added badges, features list, updated changelog.
Improved Files API, added automatical session saving and restoring to Client.
Some changes in makefile and gitignore.
License Notice now refers to all contributors.
2022-08-26 16:14:07 +04:00
|
|
|
from typing import Dict, Any
|
|
|
|
|
|
|
|
|
|
import requests
|
2022-08-22 09:55:08 +04:00
|
|
|
|
2021-10-08 19:35:20 +04:00
|
|
|
from cloudscraper import CloudScraper
|
|
|
|
|
|
2021-11-01 18:04:19 +04:00
|
|
|
from . import atjsparse
|
2022-07-01 14:28:39 +04:00
|
|
|
from .aterrors import TokenError
|
|
|
|
|
from .aterrors import CloudflareError
|
|
|
|
|
from .aterrors import AternosPermissionError
|
2021-10-27 19:42:41 +03:00
|
|
|
|
2022-07-01 14:28:39 +04:00
|
|
|
REQUA = \
|
|
|
|
|
'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 ' \
|
|
|
|
|
'(KHTML, like Gecko) Chrome/99.0.4844.84 Safari/537.36 OPR/85.0.4341.47'
|
2021-10-08 19:35:20 +04:00
|
|
|
|
2022-06-23 15:13:56 +04:00
|
|
|
|
2021-10-14 17:56:01 +04:00
|
|
|
class AternosConnect:
|
2021-10-08 19:35:20 +04:00
|
|
|
|
2022-09-29 18:18:15 +04:00
|
|
|
"""Class for sending API requests,
|
|
|
|
|
bypassing Cloudflare and parsing responses"""
|
2022-06-17 12:30:58 +04:00
|
|
|
|
2022-06-23 15:13:56 +04:00
|
|
|
def __init__(self) -> None:
|
2021-11-01 18:04:19 +04:00
|
|
|
|
2022-10-05 19:44:00 +04:00
|
|
|
self.cf_init = partial(CloudScraper)
|
|
|
|
|
self.session = self.cf_init()
|
2022-07-01 14:28:39 +04:00
|
|
|
self.sec = ''
|
|
|
|
|
self.token = ''
|
2021-11-01 18:04:19 +04:00
|
|
|
|
2022-10-05 19:44:00 +04:00
|
|
|
def add_args(self, **kwargs) -> None:
|
|
|
|
|
|
2022-10-10 11:31:14 +04:00
|
|
|
"""Pass arguments to CloudScarper
|
|
|
|
|
session object __init__
|
|
|
|
|
if kwargs is not empty
|
2022-10-05 19:44:00 +04:00
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
**kwargs: Keyword arguments
|
|
|
|
|
"""
|
|
|
|
|
|
2022-10-10 11:31:14 +04:00
|
|
|
if len(kwargs) < 1:
|
|
|
|
|
logging.debug('**kwargs is empty')
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
logging.debug('New args for CloudScraper: %s', kwargs)
|
2022-10-05 19:44:00 +04:00
|
|
|
self.cf_init = partial(CloudScraper, **kwargs)
|
|
|
|
|
self.refresh_session()
|
|
|
|
|
|
2022-10-10 11:31:14 +04:00
|
|
|
def clear_args(self) -> None:
|
|
|
|
|
|
|
|
|
|
"""Clear CloudScarper object __init__ arguments
|
|
|
|
|
which was set using add_args method"""
|
|
|
|
|
|
|
|
|
|
logging.debug('Creating session object with no keywords')
|
|
|
|
|
self.cf_init = partial(CloudScraper)
|
|
|
|
|
self.refresh_session()
|
|
|
|
|
|
2022-10-05 19:44:00 +04:00
|
|
|
def refresh_session(self) -> None:
|
|
|
|
|
|
|
|
|
|
"""Creates a new CloudScraper
|
|
|
|
|
session object and copies all cookies.
|
|
|
|
|
Required for bypassing Cloudflare"""
|
|
|
|
|
|
|
|
|
|
old_cookies = self.session.cookies
|
|
|
|
|
self.session = self.cf_init()
|
|
|
|
|
self.session.cookies.update(old_cookies)
|
|
|
|
|
del old_cookies
|
|
|
|
|
|
2022-06-23 15:13:56 +04:00
|
|
|
def parse_token(self) -> str:
|
2021-11-01 18:04:19 +04:00
|
|
|
|
2022-06-23 15:13:56 +04:00
|
|
|
"""Parses Aternos ajax token that
|
|
|
|
|
is needed for most requests
|
2022-06-17 12:30:58 +04:00
|
|
|
|
MkDocs, Readme, Files API, Automated session saving, v2.0.1
MkDocs: sphinx docstrings rewritten to google, improved config, written the major part of how-to.
Readme: centered title + logo, added badges, features list, updated changelog.
Improved Files API, added automatical session saving and restoring to Client.
Some changes in makefile and gitignore.
License Notice now refers to all contributors.
2022-08-26 16:14:07 +04:00
|
|
|
Raises:
|
|
|
|
|
RuntimeWarning: If the parser can not
|
|
|
|
|
find `<head>` tag in HTML response
|
|
|
|
|
TokenError: If the parser is unable
|
|
|
|
|
to extract ajax token from HTML
|
|
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
|
Aternos ajax token
|
2022-06-23 15:13:56 +04:00
|
|
|
"""
|
2022-06-17 12:30:58 +04:00
|
|
|
|
2022-06-23 15:13:56 +04:00
|
|
|
loginpage = self.request_cloudflare(
|
2022-07-01 14:28:39 +04:00
|
|
|
'https://aternos.org/go/', 'GET'
|
2022-06-23 15:13:56 +04:00
|
|
|
).content
|
2021-11-01 18:04:19 +04:00
|
|
|
|
2022-06-23 15:13:56 +04:00
|
|
|
# Using the standard string methods
|
|
|
|
|
# instead of the expensive xml parsing
|
|
|
|
|
head = b'<head>'
|
|
|
|
|
headtag = loginpage.find(head)
|
|
|
|
|
headend = loginpage.find(b'</head>', headtag + len(head))
|
|
|
|
|
|
|
|
|
|
# Some checks
|
|
|
|
|
if headtag < 0 or headend < 0:
|
|
|
|
|
pagehead = loginpage
|
|
|
|
|
raise RuntimeWarning(
|
|
|
|
|
'Unable to find <head> tag, parsing the whole page'
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
# Extracting <head> content
|
|
|
|
|
headtag = headtag + len(head)
|
|
|
|
|
pagehead = loginpage[headtag:headend]
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
text = pagehead.decode('utf-8', 'replace')
|
|
|
|
|
js_code = re.findall(r'\(\(\)(.*?)\)\(\);', text)
|
MkDocs, Readme, Files API, Automated session saving, v2.0.1
MkDocs: sphinx docstrings rewritten to google, improved config, written the major part of how-to.
Readme: centered title + logo, added badges, features list, updated changelog.
Improved Files API, added automatical session saving and restoring to Client.
Some changes in makefile and gitignore.
License Notice now refers to all contributors.
2022-08-26 16:14:07 +04:00
|
|
|
|
|
|
|
|
token_func = js_code[0]
|
|
|
|
|
if len(js_code) > 1:
|
|
|
|
|
token_func = js_code[1]
|
2022-06-23 15:13:56 +04:00
|
|
|
|
2022-07-01 14:28:39 +04:00
|
|
|
ctx = atjsparse.exec_js(token_func)
|
2022-06-23 15:13:56 +04:00
|
|
|
self.token = ctx.window['AJAX_TOKEN']
|
|
|
|
|
|
2022-07-01 14:28:39 +04:00
|
|
|
except (IndexError, TypeError) as err:
|
2022-06-23 15:13:56 +04:00
|
|
|
raise TokenError(
|
|
|
|
|
'Unable to parse TOKEN from the page'
|
2022-07-01 14:28:39 +04:00
|
|
|
) from err
|
2022-06-23 15:13:56 +04:00
|
|
|
|
|
|
|
|
return self.token
|
|
|
|
|
|
|
|
|
|
def generate_sec(self) -> str:
|
|
|
|
|
|
|
|
|
|
"""Generates Aternos SEC token which
|
|
|
|
|
is also needed for most API requests
|
|
|
|
|
|
MkDocs, Readme, Files API, Automated session saving, v2.0.1
MkDocs: sphinx docstrings rewritten to google, improved config, written the major part of how-to.
Readme: centered title + logo, added badges, features list, updated changelog.
Improved Files API, added automatical session saving and restoring to Client.
Some changes in makefile and gitignore.
License Notice now refers to all contributors.
2022-08-26 16:14:07 +04:00
|
|
|
Returns:
|
|
|
|
|
Random SEC `key:value` string
|
2022-06-23 15:13:56 +04:00
|
|
|
"""
|
|
|
|
|
|
2022-10-05 19:24:00 +04:00
|
|
|
randkey = secrets.token_hex(8)
|
|
|
|
|
randval = secrets.token_hex(8)
|
2022-06-23 15:13:56 +04:00
|
|
|
self.sec = f'{randkey}:{randval}'
|
|
|
|
|
self.session.cookies.set(
|
|
|
|
|
f'ATERNOS_SEC_{randkey}', randval,
|
|
|
|
|
domain='aternos.org'
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
return self.sec
|
|
|
|
|
|
|
|
|
|
def request_cloudflare(
|
|
|
|
|
self, url: str, method: str,
|
MkDocs, Readme, Files API, Automated session saving, v2.0.1
MkDocs: sphinx docstrings rewritten to google, improved config, written the major part of how-to.
Readme: centered title + logo, added badges, features list, updated changelog.
Improved Files API, added automatical session saving and restoring to Client.
Some changes in makefile and gitignore.
License Notice now refers to all contributors.
2022-08-26 16:14:07 +04:00
|
|
|
params: Optional[Dict[Any, Any]] = None,
|
|
|
|
|
data: Optional[Dict[Any, Any]] = None,
|
|
|
|
|
headers: Optional[Dict[Any, Any]] = None,
|
|
|
|
|
reqcookies: Optional[Dict[Any, Any]] = None,
|
2022-06-23 15:13:56 +04:00
|
|
|
sendtoken: bool = False,
|
MkDocs, Readme, Files API, Automated session saving, v2.0.1
MkDocs: sphinx docstrings rewritten to google, improved config, written the major part of how-to.
Readme: centered title + logo, added badges, features list, updated changelog.
Improved Files API, added automatical session saving and restoring to Client.
Some changes in makefile and gitignore.
License Notice now refers to all contributors.
2022-08-26 16:14:07 +04:00
|
|
|
retry: int = 5) -> requests.Response:
|
2022-06-23 15:13:56 +04:00
|
|
|
|
|
|
|
|
"""Sends a request to Aternos API bypass Cloudflare
|
|
|
|
|
|
MkDocs, Readme, Files API, Automated session saving, v2.0.1
MkDocs: sphinx docstrings rewritten to google, improved config, written the major part of how-to.
Readme: centered title + logo, added badges, features list, updated changelog.
Improved Files API, added automatical session saving and restoring to Client.
Some changes in makefile and gitignore.
License Notice now refers to all contributors.
2022-08-26 16:14:07 +04:00
|
|
|
Args:
|
|
|
|
|
url (str): Request URL
|
|
|
|
|
method (str): Request method, must be GET or POST
|
|
|
|
|
params (Optional[Dict[Any, Any]], optional): URL parameters
|
|
|
|
|
data (Optional[Dict[Any, Any]], optional): POST request data,
|
|
|
|
|
if the method is GET, this dict will be combined with params
|
|
|
|
|
headers (Optional[Dict[Any, Any]], optional): Custom headers
|
|
|
|
|
reqcookies (Optional[Dict[Any, Any]], optional):
|
|
|
|
|
Cookies only for this request
|
|
|
|
|
sendtoken (bool, optional): If the ajax and SEC token
|
|
|
|
|
should be sent
|
|
|
|
|
retry (int, optional): How many times parser must retry
|
|
|
|
|
connection to API bypass Cloudflare
|
|
|
|
|
|
|
|
|
|
Raises:
|
|
|
|
|
CloudflareError: When the parser has exceeded retries count
|
|
|
|
|
NotImplementedError: When the specified method is not GET or POST
|
|
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
|
API response
|
2022-06-23 15:13:56 +04:00
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
if retry <= 0:
|
|
|
|
|
raise CloudflareError('Unable to bypass Cloudflare protection')
|
|
|
|
|
|
2022-10-05 19:44:00 +04:00
|
|
|
self.refresh_session()
|
2022-08-22 09:55:08 +04:00
|
|
|
|
|
|
|
|
params = params or {}
|
|
|
|
|
data = data or {}
|
|
|
|
|
headers = headers or {}
|
|
|
|
|
reqcookies = reqcookies or {}
|
|
|
|
|
|
2022-06-23 15:13:56 +04:00
|
|
|
method = method or 'GET'
|
|
|
|
|
method = method.upper().strip()
|
|
|
|
|
if method not in ('GET', 'POST'):
|
|
|
|
|
raise NotImplementedError('Only GET and POST are available')
|
|
|
|
|
|
|
|
|
|
if sendtoken:
|
|
|
|
|
params['TOKEN'] = self.token
|
|
|
|
|
params['SEC'] = self.sec
|
|
|
|
|
headers['X-Requested-With'] = 'XMLHttpRequest'
|
|
|
|
|
|
|
|
|
|
# requests.cookies.CookieConflictError bugfix
|
|
|
|
|
reqcookies['ATERNOS_SESSION'] = self.atsession
|
|
|
|
|
del self.session.cookies['ATERNOS_SESSION']
|
|
|
|
|
|
2022-09-30 14:55:24 +04:00
|
|
|
reqcookies_dbg = {
|
|
|
|
|
k: str(v or '')[:3]
|
|
|
|
|
for k, v in reqcookies.items()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
session_cookies_dbg = {
|
|
|
|
|
k: str(v or '')[:3]
|
|
|
|
|
for k, v in self.session.cookies.items()
|
|
|
|
|
}
|
|
|
|
|
|
2022-09-30 14:39:16 +04:00
|
|
|
logging.debug('Requesting(%s)%s', method, url)
|
|
|
|
|
logging.debug('headers=%s', headers)
|
|
|
|
|
logging.debug('params=%s', params)
|
|
|
|
|
logging.debug('data=%s', data)
|
2022-09-30 14:55:24 +04:00
|
|
|
logging.debug('req-cookies=%s', reqcookies_dbg)
|
|
|
|
|
logging.debug('session-cookies=%s', session_cookies_dbg)
|
2022-06-23 15:13:56 +04:00
|
|
|
|
|
|
|
|
if method == 'POST':
|
2022-08-22 09:55:08 +04:00
|
|
|
sendreq = partial(
|
|
|
|
|
self.session.post,
|
|
|
|
|
params=params,
|
|
|
|
|
data=data
|
2022-06-23 15:13:56 +04:00
|
|
|
)
|
|
|
|
|
else:
|
2022-08-22 09:55:08 +04:00
|
|
|
sendreq = partial(
|
|
|
|
|
self.session.get,
|
|
|
|
|
params={**params, **data}
|
2022-06-23 15:13:56 +04:00
|
|
|
)
|
|
|
|
|
|
2022-08-22 09:55:08 +04:00
|
|
|
req = sendreq(
|
|
|
|
|
url,
|
|
|
|
|
headers=headers,
|
|
|
|
|
cookies=reqcookies
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
resp_type = req.headers.get('content-type', '')
|
|
|
|
|
html_type = resp_type.find('text/html') != -1
|
|
|
|
|
cloudflare = req.status_code == 403
|
|
|
|
|
|
|
|
|
|
if html_type and cloudflare:
|
2022-06-23 15:13:56 +04:00
|
|
|
logging.info('Retrying to bypass Cloudflare')
|
2022-08-22 09:55:08 +04:00
|
|
|
time.sleep(0.2)
|
2022-07-01 09:36:52 +04:00
|
|
|
return self.request_cloudflare(
|
2022-06-23 15:13:56 +04:00
|
|
|
url, method,
|
|
|
|
|
params, data,
|
|
|
|
|
headers, reqcookies,
|
2022-08-22 09:55:08 +04:00
|
|
|
sendtoken, retry - 1
|
2022-06-23 15:13:56 +04:00
|
|
|
)
|
|
|
|
|
|
2022-09-30 14:39:16 +04:00
|
|
|
logging.debug('AternosConnect received: %s', req.text[:65])
|
2022-06-23 15:13:56 +04:00
|
|
|
logging.info(
|
2022-09-30 14:39:16 +04:00
|
|
|
'%s completed with %s status',
|
|
|
|
|
method, req.status_code
|
2022-06-23 15:13:56 +04:00
|
|
|
)
|
|
|
|
|
|
2022-07-01 14:28:39 +04:00
|
|
|
if req.status_code == 402:
|
|
|
|
|
raise AternosPermissionError
|
2022-07-01 10:31:23 +04:00
|
|
|
|
2022-07-01 14:28:39 +04:00
|
|
|
req.raise_for_status()
|
2022-06-23 15:13:56 +04:00
|
|
|
return req
|
MkDocs, Readme, Files API, Automated session saving, v2.0.1
MkDocs: sphinx docstrings rewritten to google, improved config, written the major part of how-to.
Readme: centered title + logo, added badges, features list, updated changelog.
Improved Files API, added automatical session saving and restoring to Client.
Some changes in makefile and gitignore.
License Notice now refers to all contributors.
2022-08-26 16:14:07 +04:00
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def atsession(self) -> str:
|
|
|
|
|
|
|
|
|
|
"""Aternos session cookie,
|
|
|
|
|
empty string if not logged in
|
|
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
|
Session cookie
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
return self.session.cookies.get(
|
|
|
|
|
'ATERNOS_SESSION', ''
|
|
|
|
|
)
|
