1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
import json
import sys
from importlib import import_module
from werkzeug.serving import generate_adhoc_ssl_context
from werkzeug.serving import run_simple
from werkzeug.serving import WSGIRequestHandler
from werkzeug.wrappers import Request
from werkzeug.wrappers import Response
name = sys.argv[1]
mod = import_module(f"{name}_app")
@Request.application
def app(request):
if request.path == "/ensure":
return Response()
return Response.from_app(mod.app, request.environ)
kwargs = getattr(mod, "kwargs", {})
kwargs.update(hostname="127.0.0.1", port=5000, application=app)
kwargs.update(json.loads(sys.argv[2]))
ssl_context = kwargs.get("ssl_context")
override_client_addr = kwargs.pop("override_client_addr", None)
if ssl_context == "custom":
kwargs["ssl_context"] = generate_adhoc_ssl_context()
elif isinstance(ssl_context, list):
kwargs["ssl_context"] = tuple(ssl_context)
if override_client_addr:
WSGIRequestHandler.address_string = lambda _: override_client_addr
run_simple(**kwargs)
|
