9
0
mirror of https://github.com/donlon/cloudflare-error-page.git synced 2025-12-19 14:59:28 +00:00
Files
cloudflare-error-page/cloudflare_error_page/__init__.py
2025-11-21 09:30:51 +08:00

44 lines
1.3 KiB
Python

import html
import os
import secrets
from datetime import datetime, timezone
from jinja2 import Environment, PackageLoader, Template, select_autoescape
env = Environment(
loader=PackageLoader("cloudflare_error_page"),
autoescape=select_autoescape(),
trim_blocks=True,
lstrip_blocks=True,
)
default_template = env.get_template("error.html")
def get_resources_folder() -> str:
"""
Get resources folder that contains stylesheet and icons for the error page.
If you pass resources_use_cdn=True to render(), local resources are not used.
"""
return os.path.join(os.path.dirname(os.path.abspath(__file__)), 'resources')
def fill_params(params: dict):
if not params.get('time'):
utc_now = datetime.now(timezone.utc)
params['time'] = utc_now.strftime("%Y-%m-%d %H:%M:%S UTC")
if not params.get('ray_id'):
params['ray_id'] = secrets.token_hex(8)
def render(params: dict, allow_html: bool=True, use_cdn: bool=True) -> str:
"""
Render a customized Cloudflare error page.
"""
params = {**params}
fill_params(params)
if not allow_html:
params['what_happened'] = html.escape(params.get('what_happened', ''))
params['what_can_i_do'] = html.escape(params.get('what_can_i_do', ''))
return default_template.render(params=params, resources_use_cdn=use_cdn)