From 67f0096f7f9981a527fc4ffc65cdf16360cac100 Mon Sep 17 00:00:00 2001 From: Henrik Jess Nielsen Date: Wed, 11 Dec 2024 20:39:22 +0100 Subject: [PATCH] Lets test --- app.py | 22 ++-------------------- 1 file changed, 2 insertions(+), 20 deletions(-) diff --git a/app.py b/app.py index 34ad408..aa78e0e 100644 --- a/app.py +++ b/app.py @@ -3,31 +3,14 @@ from fastapi.staticfiles import StaticFiles from fastapi.templating import Jinja2Templates import json import os +from fastapi import FastAPI from fastapi.responses import HTMLResponse from contextlib import asynccontextmanager from markdown_render import render_markdown_with_jinja from starlette.middleware.httpsredirect import HTTPSRedirectMiddleware -from fastapi import FastAPI, Request -from fastapi.responses import RedirectResponse -from starlette.middleware.base import BaseHTTPMiddleware -class ConditionalHTTPSRedirectMiddleware(BaseHTTPMiddleware): - async def dispatch(self, request: Request, call_next): - # Extract the host or remote address - host = request.headers.get("host", "") - client_ip = request.client.host # Get the client's IP address - # Log for debugging - print(f"Client IP: {client_ip}, Host: {host}") - # Check condition: Redirect only if not on 127.0.0.1 - if client_ip != "127.0.0.1" and request.url.scheme == "http": - https_url = request.url.replace(scheme="https") - return RedirectResponse(url=https_url) - - # Continue processing other requests - response = await call_next(request) - return response # Context manager for app lifespan @@ -45,8 +28,7 @@ app.mount("/data", StaticFiles(directory="data"), name="data") # Mount static files app.mount("/static", StaticFiles(directory="static"), name="static") -#app.add_middleware( HTTPSRedirectMiddleware ) -app.add_middleware(ConditionalHTTPSRedirectMiddleware) +app.add_middleware( HTTPSRedirectMiddleware ) # Templates directory templates = Jinja2Templates(directory="templates")