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-20 07:24:19 +08:00

38 lines
1.3 KiB
Python

import html
import os
import secrets
from datetime import datetime, timezone
from jinja2 import Environment, PackageLoader, select_autoescape
env = Environment(
loader=PackageLoader("cloudflare_error_page"),
autoescape=select_autoescape()
)
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 render(params: dict, allow_html: bool=True, use_cdn: bool=True, show_creator: bool=False) -> str:
"""
Render a customized Cloudflare error page.
"""
params = {**params}
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)
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', ''))
template = env.get_template("error.html")
return template.render(params=params, resources_use_cdn=use_cdn, show_creator=show_creator)