This repository has been archived on 2024-07-30. You can view files and clone it, but cannot push or open issues or pull requests.
python-aternos/python_aternos/atserver.py
2021-10-15 19:31:47 +04:00

199 lines
4.3 KiB
Python

import re
import json
import lxml.html
from requests import Response
from typing import Optional, Dict
from . import atconnect
from . import aterrors
from . import atfm
from . import atconf
from . import atplayers
SOFTWARE_JAVA = 0
SOFTWARE_BEDROCK = 1
PLAYERS_ALLOWED = 'whitelist'
PLAYERS_OPS = 'ops'
PLAYERS_BANNED = 'banned-players'
PLAYERS_IPS = 'banned-ips'
STATUS_OFFLINE = 0
STATUS_ONLINE = 1
STATUS_LOADING = 2
STATUS_SHUTDOWN = 3
STATUS_ERROR = 7
class AternosServer:
def __init__(self, servid:str, atconn:atconnect.AternosConnect) -> None:
self.servid = servid
self.atconn = atconn
servreq = self.atserver_request(
'https://aternos.org/server',
atconnect.REQGET
)
servtree = lxml.html.fromstring(servreq.content)
self._info = json.loads(
re.search(
r'var\s*lastStatus\s*=\s*({.*})',
servtree.head.text_content()
)[1]
)
self.atconn.parse_token(servreq.content)
self.atconn.generate_sec()
def start(self, accepteula:bool=True) -> None:
startreq = self.atserver_request(
'https://aternos.org/panel/ajax/start.php',
atconnect.REQGET, sendtoken=True
)
startresult = startreq.json()
if startresult['success']:
return
error = startresult['error']
if error == 'eula' and accepteula:
self.eula()
self.start(accepteula=False)
elif error == 'eula':
raise aterrors.AternosServerStartError(
'EULA was not accepted. Use start(accepteula=True)'
)
elif error == 'already':
raise aterrors.AternosServerStartError(
'Server is already running'
)
else:
raise aterrors.AternosServerStartError(
f'Unable to start server. Code: {error}'
)
def confirm(self) -> None:
self.atserver_request(
'https://aternos.org/panel/ajax/confirm.php',
atconnect.REQGET, sendtoken=True
)
def stop(self) -> None:
self.atserver_request(
'https://aternos.org/panel/ajax/stop.php',
atconnect.REQGET, sendtoken=True
)
def cancel(self) -> None:
self.atserver_request(
'https://aternos.org/panel/ajax/cancel.php',
atconnect.REQGET, sendtoken=True
)
def restart(self) -> None:
self.atserver_request(
'https://aternos.org/panel/ajax/restart.php',
atconnect.REQGET, sendtoken=True
)
def eula(self) -> None:
self.atserver_request(
'https://aternos.org/panel/ajax/eula.php',
atconnect.REQGET, sendtoken=True
)
def files(self) -> atfm.AternosFileManager:
return atfm.AternosFileManager(self)
def config(self) -> atconf.AternosConfig:
return atconf.AternosConfig(self)
def players(self, lst:str) -> atplayers.AternosPlayersList:
return atplayers.AternosPlayersList(lst, self)
def atserver_request(
self, url:str, method:int,
params:Optional[dict]=None,
data:Optional[dict]=None,
headers:Optional[dict]=None,
sendtoken:bool=False) -> Response:
return self.atconn.request_cloudflare(
url=url, method=method,
params=params, data=data,
headers=headers,
reqcookies={
'ATERNOS_SERVER': self.servid
},
sendtoken=sendtoken
)
@property
def subdomain(self) -> str:
atdomain = self.domain
return atdomain[:atdomain.find('.')]
@subdomain.setter
def subdomain(self, value:str) -> None:
self.atserver_request(
'https://aternos.org/panel/ajax/options/subdomain.php',
atconnect.REQGET, params={'subdomain': value}
)
@property
def motd(self) -> str:
return self._info['motd']
@motd.setter
def motd(self, value:str) -> None:
self.atserver_request(
'https://aternos.org/panel/ajax/options/motd.php',
atconnect.REQPOST, data={'motd': value}
)
@property
def address(self) -> str:
return f'{self.domain}:{self.port}'
@property
def domain(self) -> str:
return self._info['ip']
@property
def port(self) -> int:
return self._info['port']
@property
def edition(self) -> int:
soft_type = self._info['bedrock']
if soft_type == True:
return SOFTWARE_BEDROCK
else:
return SOFTWARE_JAVA
@property
def software(self) -> str:
return self._info['software']
@property
def version(self) -> str:
return self._info['version']
@property
def status(self) -> int:
return int(self._info['status'])
@property
def ram(self) -> int:
return int(self._info['ram'])