处理protocol与host信息

HenryJi529 · Jun 5, 2024 · 218c90b · 218c90b
1 parent b11c2fc
commit 218c90b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 14 deletions.
diff --git a/Morningstar/views/base.py b/Morningstar/views/base.py
@@ -42,13 +42,7 @@ def create_activate_message(username, host):
         except:
             pass
         conn.set(f"{username}-activate", code, ex=60 * 10)
-
-        protocol = (
-            "https://"
-            if os.environ.get("DJANGO_SETTINGS_MODULE", "Morningstar.settings.dev")
-            == "Morningstar.settings.production"
-            else "http://"
-        )
+        protocol = "https://" if request.is_secure() else "http://"
         link = (
             protocol
             + host

diff --git a/Morningstar/views/tool.py b/Morningstar/views/tool.py
@@ -139,12 +139,7 @@ def get_login_token(request: HttpRequest, identity):
         username = user.username
         password = user.password
         token = html.escape(password[-20:])
-        protocol = (
-            "https://"
-            if os.environ.get("DJANGO_SETTINGS_MODULE", "Morningstar.settings.dev")
-            == "Morningstar.settings.production"
-            else "http://"
-        )
+        protocol = "https://" if request.is_secure() else "http://"
         link = (
             protocol
             + request.get_host()

diff --git a/apps/share/views.py b/apps/share/views.py
@@ -67,5 +67,8 @@ def submit(request: HttpRequest):
         item.save()
         id = item.id
         link = f"redirect/{id}/"
-        request.session["share-qrcode"] = "https://morningstar369.com/share/" + link
+        protocol = "https://" if request.is_secure() else "http://"
+        request.session["share-qrcode"] = (
+            protocol + request.META["HTTP_HOST"] + "/share/" + link
+        )
         return Response({"link": link})