You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.4 KiB
85 lines
2.4 KiB
from io import BytesIO
|
|
from pathlib import Path
|
|
from zipfile import ZipFile
|
|
|
|
FS_ROOT = '/srv/www/atr0phy.net/acks/quests'
|
|
URL_ROOT = 'https://atr0phy.net/acks/quests'
|
|
IMG_EXTENSIONS = ('png', 'jpg', 'jpeg')
|
|
|
|
def load_quests():
|
|
quest_list = {}
|
|
|
|
fs = Path(FS_ROOT)
|
|
for e in fs.iterdir():
|
|
if not e.is_dir():
|
|
continue
|
|
|
|
# Each directory correlates to a level group
|
|
quest_level_url_name = e.name.replace('Level', '').replace('-', '').replace(' ', '')
|
|
|
|
# Gather quests under each level
|
|
quest_list[quest_level_url_name] = load_quest_level_directory(e)
|
|
return quest_list
|
|
|
|
def load_quest_level_directory(qdir):
|
|
this_level = []
|
|
for e in qdir.iterdir():
|
|
if not e.is_dir():
|
|
continue
|
|
|
|
# Each directory is a quest for this level
|
|
this_level.append(e.name)
|
|
return this_level
|
|
|
|
def get_quest_details(level, quest_name):
|
|
level_range = str(level)
|
|
folder_name = "Level {0} - {1}".format(level_range[0], level_range[1])
|
|
|
|
quest_path = "{}/{}/{}".format(FS_ROOT, folder_name, quest_name)
|
|
fs = Path(quest_path)
|
|
|
|
if not fs.exists() or not fs.is_dir():
|
|
print('EXITING {}'.format(fs))
|
|
return None
|
|
|
|
quest = {}
|
|
quest['name'] = fs.name
|
|
quest['download'] = (level, quest_name)
|
|
quest['assets'] = []
|
|
|
|
def urlify(path):
|
|
return {'display': path.name, 'url': path.as_posix().replace(FS_ROOT, URL_ROOT)}
|
|
|
|
for e in fs.iterdir():
|
|
if e.name.endswith(IMG_EXTENSIONS):
|
|
quest['assets'].append(urlify(e))
|
|
elif e.name == 'info.html':
|
|
quest['info'] = urlify(e)
|
|
elif e.name == 'tsv.txt':
|
|
quest['tsv'] = urlify(e)
|
|
|
|
print("Quest: {}".format(quest))
|
|
return quest
|
|
|
|
def get_quest_archive(level, quest_name):
|
|
level_range = str(level)
|
|
folder_name = "Level {0} - {1}".format(level_range[0], level_range[1])
|
|
|
|
quest_path = "{}/{}/{}".format(FS_ROOT, folder_name, quest_name)
|
|
fs = Path(quest_path)
|
|
|
|
if not fs.exists() or not fs.is_dir():
|
|
print('EXITING {}'.format(fs))
|
|
return None
|
|
|
|
file_path_list = []
|
|
for e in fs.iterdir():
|
|
file_path_list.append((e.as_posix(), e.name))
|
|
|
|
archive_mem = BytesIO()
|
|
with ZipFile(archive_mem, 'w') as archive:
|
|
for f in file_path_list:
|
|
archive.write(f[0], f[1])
|
|
archive_mem.seek(0)
|
|
|
|
return archive_mem
|