9
0
mirror of https://github.com/donlon/cloudflare-error-page.git synced 2025-12-19 23:09:29 +00:00

examples: update flask demo

This commit is contained in:
Anthony Donlon
2025-11-21 23:55:01 +08:00
parent 22677adda2
commit 28b5ee7e69

View File

@@ -6,21 +6,19 @@ import sys
from flask import ( from flask import (
Flask, Flask,
json,
request, request,
abort,
redirect,
send_from_directory send_from_directory
) )
# Append this directory to sys.path is not required if the package is already installed
examples_dir = os.path.dirname(os.path.abspath(__file__)) examples_dir = os.path.dirname(os.path.abspath(__file__))
sys.path.append(os.path.dirname(examples_dir)) sys.path.append(os.path.dirname(examples_dir))
from cloudflare_error_page import get_resources_folder, render as render_cf_error_page from cloudflare_error_page import get_resources_folder, render as render_cf_error_page
# TODO: use blueprint
app = Flask(__name__) app = Flask(__name__)
# You can use resources from Cloudflare CDN. But in case of changing, you can set use_cdn = False to use bundled resources. # You can **use resources from Cloudflare CDN. But in case of changing, you can set use_cdn = False to use bundled resources.
use_cdn = True use_cdn = True
if use_cdn: if use_cdn:
@@ -29,65 +27,48 @@ if use_cdn:
# This handler is used to provide stylesheet and icon resources for the error page. If you pass use_cdn=True to render_cf_error_page # This handler is used to provide stylesheet and icon resources for the error page. If you pass use_cdn=True to render_cf_error_page
# or if your site is under proxy of Cloudflare (the cdn-cgi folder is already provided by Cloudflare), this handler can be removed. # or if your site is under proxy of Cloudflare (the cdn-cgi folder is already provided by Cloudflare), this handler can be removed.
@app.route('/cdn-cgi/<path:path>') @app.route('/cdn-cgi/<path:path>')
def cdn_cgi(path): def cdn_cgi(path: str):
return send_from_directory(res_folder, path) return send_from_directory(res_folder, path)
param_cache: dict[str, dict] = {} @app.route('/')
def index():
def get_page_params(name: str) -> dict: params = {
name = re.sub(r'[^\w]', '', name) "title": "Internal server error",
params = param_cache.get(name) "error_code": 500,
if params is not None: "browser_status": {
return params "status": "ok"
try: },
with open(os.path.join(examples_dir, f'{name}.json')) as f: "cloudflare_status": {
params = json.load(f) "status": "error",
param_cache[name] = params "status_text": "Error"
return params },
except Exception as _: "host_status": {
return None "status": "ok",
"location": "example.com"
},
@app.route('/', defaults={'name': 'default'}) "error_source": "cloudflare",
@app.route('/<path:name>') "what_happened": "<p>There is an internal server error on Cloudflare\"s network.</p>",
def index(name: str): "what_can_i_do": "<p>Please try again in a few minutes.</p>"
name = os.path.basename(name) # keep only the base name }
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)
# Get real Ray ID from Cloudflare header # Get real Ray ID from Cloudflare header
ray_id = request.headers.get('Cf-Ray') ray_id = request.headers.get('Cf-Ray', '')[:16]
if ray_id:
ray_id = ray_id[:16]
# Get real client ip from Cloudflare header or request.remote_addr # Get real client ip from Cloudflare header or request.remote_addr
client_ip = request.headers.get('X-Forwarded-For') client_ip = request.headers.get('X-Forwarded-For')
if not client_ip: if not client_ip:
client_ip = request.remote_addr client_ip = request.remote_addr
params = { params.update({
**params,
'ray_id': ray_id, 'ray_id': ray_id,
'client_ip': client_ip, 'client_ip': client_ip,
} })
# Render the error page # Render the error page
return render_cf_error_page(params, use_cdn=use_cdn), 500 return render_cf_error_page(params, use_cdn=use_cdn), 500
@app.route('/favicon.ico')
def favicon_ico():
abort(404)
if __name__ == '__main__': if __name__ == '__main__':
if len(sys.argv) > 1: if len(sys.argv) > 1:
host = sys.argv[1] host = sys.argv[1]