mm/noaa
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

proxy-specific changes

Browse Source
This commit is contained in:
Michael Pilosov 2024-05-31 12:48:10 -06:00
parent 3c2d665bfd
commit c617323375
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
app/app.py
View File

@ -9,6 +9,9 @@ from flask import Flask, Response, render_template, request, send_from_directory
from prefect.deployments import run_deployment from prefect.deployments import run_deployment
PORT = 9021 PORT = 9021
PROTO = "https"
BARE = True
app = Flask(__name__) app = Flask(__name__)
logging.basicConfig(level=logging.DEBUG) logging.basicConfig(level=logging.DEBUG)
@ -22,20 +25,20 @@ def get_host():
host = os.environ.get("LIGHTNING_CLOUDSPACE_HOST") host = os.environ.get("LIGHTNING_CLOUDSPACE_HOST")
if host is None: if host is None:
default_host = os.environ.get("HOST_NAME", "0.0.0.0") default_host = os.environ.get("HOST_NAME", "0.0.0.0")
return f"{default_host}:{PORT}" return f"{default_host}:{PORT}" if not BARE else f"{default_host}"
else: else:
return f"{PORT}-{host}" return f"{PORT}-{host}" if not BARE else f"{host}"
@app.route("/iframe") @app.route("/iframe")
@app.route("/iframe/") @app.route("/iframe/")
@app.route("/iframe/<path:subpath>") @app.route("/iframe/<path:subpath>")
def home(subpath="images/animations/"): def home(subpath="images/animations/", proto=PROTO):
host = get_host() host = get_host()
initial_url = f"http://{host}/{subpath}" initial_url = f"{proto}://{host}/{subpath}"
api_url = f"http://{host}/api" api_url = f"{proto}://{host}/api"
return render_template( return render_template(
"index.html", initial_url=initial_url, host=f"http://{host}", api_url=api_url "index.html", initial_url=initial_url, host=f"{proto}://{host}", api_url=api_url
) )
@ -70,10 +73,10 @@ def custom_static(filename):
@app.route("/", methods=["GET"]) @app.route("/", methods=["GET"])
@app.route("/<path:url>", methods=["GET"]) @app.route("/<path:url>", methods=["GET"])
def proxy(url=""): def proxy(url="", proto=PROTO):
original_base_url = "https://services.swpc.noaa.gov" original_base_url = "https://services.swpc.noaa.gov"
host = get_host() host = get_host()
proxy_base_url = f"http://{host}/" proxy_base_url = f"{proto}://{host}/"
target_url = f"{original_base_url}/{url}" target_url = f"{original_base_url}/{url}"
logging.debug(f"Fetching URL: {target_url}") logging.debug(f"Fetching URL: {target_url}")