mirror of
https://github.com/donlon/cloudflare-error-page.git
synced 2025-12-19 14:59:28 +00:00
56 lines
1.2 KiB
Python
56 lines
1.2 KiB
Python
# SPDX-License-Identifier: MIT
|
|
|
|
import copy
|
|
import os
|
|
import re
|
|
|
|
from flask import (
|
|
Blueprint,
|
|
json,
|
|
abort,
|
|
redirect,
|
|
)
|
|
|
|
from cloudflare_error_page import ErrorPageParams
|
|
from .utils import (
|
|
render_extended_template,
|
|
)
|
|
|
|
root_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../../')
|
|
examples_dir = os.path.join(root_dir, 'examples')
|
|
|
|
bp = Blueprint('examples', __name__, url_prefix='/')
|
|
|
|
|
|
param_cache: dict[str, dict] = {}
|
|
|
|
def get_page_params(name: str) -> ErrorPageParams:
|
|
name = re.sub(r'[^\w]', '', name)
|
|
params = param_cache.get(name)
|
|
if params is not None:
|
|
return copy.deepcopy(params)
|
|
try:
|
|
with open(os.path.join(examples_dir, f'{name}.json')) as f:
|
|
params = json.load(f)
|
|
param_cache[name] = params
|
|
return copy.deepcopy(params)
|
|
except Exception as _:
|
|
return None
|
|
|
|
|
|
@bp.route('/', defaults={'name': 'default'})
|
|
@bp.route('/<path:name>')
|
|
def index(name: str):
|
|
lower_name = name.lower()
|
|
if name != lower_name:
|
|
return redirect(lower_name)
|
|
else:
|
|
name = lower_name
|
|
|
|
params = get_page_params(name)
|
|
if params is None:
|
|
abort(404)
|
|
|
|
# Render the error page
|
|
return render_extended_template(params=params)
|