in server/app/__init__.py [0:0]
def main():
asfquart.construct(__name__, oauth="/api/auth")
asfquart.APP.secret_key = secrets.token_hex() # For session management
asfquart.APP.config[
"MAX_CONTENT_LENGTH"
] = config.server.max_content_length # Ensure upload limits match expectations
asfquart.APP.url_map.converters[
"filename"
] = middleware.FilenameConverter # Special converter for filename-style vars
# Static files (or index.html if requesting a dir listing)
@asfquart.APP.route("/<path:path>")
@asfquart.APP.route("/")
async def static_files(path="index.html"):
if path.endswith("/"):
path += "index.html"
if path.endswith(".html"): # Serve HTML from the compiled output dir
return await quart.send_from_directory(COMPILED_DIR, path)
return await quart.send_from_directory(STATIC_DIR, path)
@asfquart.APP.before_serving
async def compile_html():
"""Compiles HTML files in htdocs/ using a master template"""
master_template = open(os.path.join(TEMPLATES_DIR, "master.html")).read()
# Add sub-resource integrity to all scripts
for script_src in re.finditer(r'(src="(.+?\.js)")', master_template):
script_name = script_src.group(2).lstrip("/")
script_path = os.path.join(STATIC_DIR, script_name)
if os.path.isfile(script_path):
sri = file_to_sri(script_path)
orig_src = script_src.group(1)
new_src = f'{orig_src} integrity="{sri}"'
master_template = master_template.replace(orig_src, new_src)
if not os.path.isdir(COMPILED_DIR):
log.log(
f"Compiled HTML directory {COMPILED_DIR} does not exist, will attempt to create it"
)
os.makedirs(COMPILED_DIR, exist_ok=True, mode=0o700)
for htmlfile in [
filename
for filename in os.listdir(STATIC_DIR)
if filename.endswith(".html")
]:
print(f"Compiling {htmlfile} into output/{htmlfile}")
htmldata = open(os.path.join(STATIC_DIR, htmlfile)).read()
output = master_template.replace("{contents}", htmldata)
open(os.path.join(COMPILED_DIR, htmlfile), "w").write(output)
@asfquart.APP.before_serving
async def load_endpoints():
"""Load all API end points. This is run before Quart starts serving requests"""
async with asfquart.APP.app_context():
from . import endpoints
from .lib import tokens
# Regularly update the list of projects from LDAP
asfquart.APP.add_background_task(config.get_projects_from_ldap)
# Reset rate limits daily
asfquart.APP.add_background_task(middleware.reset_rate_limits)
# Fetch mailing lists hourly
asfquart.APP.add_background_task(config.fetch_valid_lists)
@asfquart.APP.after_serving
async def shutdown():
"""Ensure a clean shutdown of the portal by stopping background tasks"""
log.log("Shutting down selfserve portal...")
asfquart.APP.background_tasks.clear() # Clear repo polling etc
return asfquart.APP