112 lines
2.6 KiB
Python
112 lines
2.6 KiB
Python
import lxml.html
|
|
from typing import Optional, Union, List
|
|
from typing import TYPE_CHECKING
|
|
|
|
from . import atconnect
|
|
from . import atfile
|
|
|
|
if TYPE_CHECKING:
|
|
from atserver import AternosServer
|
|
|
|
class AternosFileManager:
|
|
|
|
def __init__(self, atserv:'AternosServer') -> None:
|
|
|
|
self.atserv = atserv
|
|
|
|
def listdir(self, path:str='') -> List[atfile.AternosFile]:
|
|
|
|
filesreq = self.atserv.atserver_request(
|
|
f'https://aternos.org/files/{path}',
|
|
atconnect.REQGET
|
|
)
|
|
filestree = lxml.html.fromstring(filesreq.content)
|
|
fileslist = filestree.xpath(
|
|
'//div[@class="files"]/div[@class="directory dropzone"]' + \
|
|
'/div[@class="file clickable"]'
|
|
)
|
|
|
|
files = []
|
|
for f in fileslist:
|
|
|
|
ftype_raw = f.xpath('/@data-type')
|
|
ftype = atfile.FTYPE_FILE \
|
|
if ftype_raw == 'file' \
|
|
else atfile.FTYPE_DIR
|
|
|
|
fsize_raw = f.xpath('/div[@class="filesize"]')
|
|
fsize = 0
|
|
if len(fsize_raw) > 0:
|
|
|
|
fsize_text = fsize_raw[0].text.strip()
|
|
fsize_num = fsize_text[:fsize_text.rfind(' ')]
|
|
fsize_msr = fsize_text[fsize_text.rfind(' ')+1:]
|
|
|
|
try:
|
|
fsize = convert_size(float(fsize_num), fsize_msr)
|
|
except ValueError:
|
|
fsize = -1
|
|
|
|
dlbutton = f.xpath('/div[@class="js-download-file btn btn-main btn-small btn-notext btn-no-margin"]')
|
|
dlallowed = False
|
|
if len(dlbutton) > 0:
|
|
dlallowed = True
|
|
|
|
fullpath = f.xpath('/@data-path')
|
|
filepath = fullpath[:fullpath.rfind('/')]
|
|
filename = fullpath[fullpath.rfind('/'):]
|
|
files.append(
|
|
atfile.AternosFile(
|
|
self.atserv,
|
|
filepath, filename,
|
|
ftype, fsize, dlallowed
|
|
)
|
|
)
|
|
|
|
return files
|
|
|
|
def convert_size(self, num:Union[int,float], measure:str) -> float:
|
|
|
|
measure_match = {
|
|
'B': 1,
|
|
'kB': 1000,
|
|
'MB': 1000000,
|
|
'GB': 1000000000
|
|
}
|
|
try:
|
|
result = num * measure_match[measure]
|
|
except KeyError:
|
|
result = -1
|
|
return result
|
|
|
|
def get_file(self, path:str) -> Union[atfile.AternosFile,None]:
|
|
|
|
filepath = path[:path.rfind('/')]
|
|
filename = path[path.rfind('/'):]
|
|
|
|
filedir = listdir(filepath)
|
|
for file in filedir:
|
|
if file.name == filename:
|
|
return file
|
|
|
|
return None
|
|
|
|
def dl_file(self, path:str) -> bytes:
|
|
|
|
file = self.atserv.atserver_request(
|
|
f'https://aternos.org/panel/ajax/files/download.php?' + \
|
|
f'file={path.replace("/","%2F")}',
|
|
atconnect.REQGET
|
|
)
|
|
|
|
return file.content
|
|
|
|
def dl_world(self, world:str='world') -> bytes:
|
|
|
|
world = self.atserv.atserver_request(
|
|
f'https://aternos.org/panel/ajax/worlds/download.php?' + \
|
|
f'world={world.replace("/","%2F")}',
|
|
atconnect.REQGET
|
|
)
|
|
|
|
return world.content
|