Add proper scheme (not default to https)

PythonistaGuild · Jul 15, 2024 · 68c40e7 · 68c40e7
1 parent 29e207c
commit 68c40e7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/views/api.py b/views/api.py
@@ -302,8 +302,8 @@ async def security_info(self, request: starlette_plus.Request) -> starlette_plus
                 status_code=404,
             )
 
-        delete: str = f"https://{request.url.hostname}/api/security/delete/{token}"
-        info: str = f"https://{request.url.hostname}/api/security/info/{token}"
+        delete: str = f"{request.url.scheme}://{request.url.hostname}/api/security/delete/{token}"
+        info: str = f"{request.url.scheme}://{request.url.hostname}/api/security/info/{token}"
         data: dict[str, str] = {
             "token": paste.safety,
             "delete": delete,

diff --git a/views/htmx.py b/views/htmx.py
@@ -225,7 +225,7 @@ async def paste_raw(self, request: starlette_plus.Request) -> starlette_plus.Res
 
         htmx_url: str | None = request.headers.get("HX-Current-URL", None)
         if identifier == "0" and htmx_url:
-            identifier = htmx_url.removeprefix(f"https://{request.url.hostname}/")
+            identifier = htmx_url.removeprefix(f"{request.url.scheme}://{request.url.hostname}/")
 
         headers: dict[str, str] = {"HX-Redirect": f"/raw/{identifier}"}
         paste = await self.app.database.fetch_paste(identifier, password=password)