Adding fix for localhost domain not being set

sennetconsortium · Oct 2, 2023 · 14088d0 · 14088d0
1 parent f62730f
commit 14088d0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/routes/auth/__init__.py b/src/routes/auth/__init__.py
@@ -110,7 +110,10 @@ def _login(redirect_uri, key = 'tokens'):
 
         # create a response for the user
         response = make_response(redirect(redirect_uri))
-        response.set_cookie('info', base64_json_str, expires=2**31 - 1, domain=current_app.config['COOKIE_DOMAIN'])
+        if current_app.config['COOKIE_DOMAIN'] == 'localhost':
+            response.set_cookie('info', base64_json_str, expires=2**31 - 1)
+        else:
+            response.set_cookie('info', base64_json_str, expires=2**31 - 1, domain=current_app.config['COOKIE_DOMAIN'])
         return response