58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
import hashlib
|
|
import lxml.html
|
|
|
|
from atserver import AternosServer
|
|
from atconnect import AternosConnect
|
|
from aterrors import AternosCredentialsError
|
|
|
|
class Client:
|
|
|
|
def __init__(self, username, md5=None, password=None):
|
|
|
|
if (password == None) and (md5 == None):
|
|
raise AttributeError('Password was not specified')
|
|
|
|
if (password != None):
|
|
self.__init__(
|
|
username,
|
|
md5=hashlib.md5(password.encode('utf-8'))\
|
|
.hexdigest().lower()
|
|
)
|
|
return
|
|
|
|
self.atconn = AternosConnect()
|
|
|
|
self.token = self.atconn.get_token()
|
|
self.sec = self.atconn.generate_sec()
|
|
|
|
self.credentials = {
|
|
'user': username,
|
|
'password': md5
|
|
}
|
|
|
|
loginreq = self.atconn.request_cloudflare(
|
|
f'https://aternos.org/panel/ajax/account/login.php?' + \
|
|
f'SEC={self.sec}&TOKEN={self.token}',
|
|
self.atconn.REQPOST, data=self.credentials
|
|
)
|
|
|
|
if loginreq.cookies.get('ATERNOS_SESSION', None) == None:
|
|
raise AternosCredentialsError(
|
|
'Check your username and password'
|
|
)
|
|
|
|
def get_servers(self):
|
|
|
|
serverspage = self.atconn.request_cloudflare(
|
|
'https://aternos.org/servers/',
|
|
self.atconn.REQGET
|
|
)
|
|
serverstree = lxml.html.fromstring(serverspage.content)
|
|
serverslist = serverstree.xpath('//div[@class="servers"]/div')
|
|
|
|
servers = []
|
|
for server in serverslist:
|
|
servid = server.xpath('./div[@class="server-body"]/@data-id')[0]
|
|
servers.append(AternosServer(servid, self.atconn))
|
|
|
|
return servers
|