locust-cloud 1.11.2__py3-none-any.whl → 1.11.4__py3-none-any.whl

locust_cloud/auth.py

@@ -96,11 +96,14 @@ def register_auth(environment: locust.env.Environment):
                 session["challenge_session"] = credentials.get("challenge_session")
                 session["username"] = username
                 return redirect(url_for("locust_cloud_auth.password_reset"))
-
-            if os.getenv("CUSTOMER_ID", "") and credentials["customer_id"] != os.getenv("CUSTOMER_ID", ""):
+            if os.getenv("CUSTOMER_ID", "") and credentials.get("customer_id") != os.getenv("CUSTOMER_ID", ""):
                 session["auth_error"] = "Invalid login for this deployment"
                 return redirect(url_for("locust.login"))
 
+            if not credentials.get("user_sub_id"):
+                session["auth_error"] = "Unknown error during authentication, check logs and/or contact support"
+                return redirect(url_for("locust.login"))
+
             response = redirect(url_for("locust.index"))
             response = set_credentials(username, credentials, response)
             login_user(AuthUser(credentials["user_sub_id"]))
@@ -221,10 +224,14 @@ def register_auth(environment: locust.env.Environment):
 
     @auth_blueprint.route("/resend-code")
     def resend_code():
+        if not session.get("username"):
+            session["auth_sign_up_error"] = "An unexpected error occured. Please try again."
+            return redirect(url_for("locust_cloud_auth.signup"))
+
         try:
             auth_response = requests.post(
                 f"{environment.parsed_options.deployer_url}/auth/resend-confirmation",
-                json={"username": session["username"]},
+                json={"username": session.get("username")},
             )
 
             auth_response.raise_for_status()
@@ -244,6 +251,9 @@ def register_auth(environment: locust.env.Environment):
     def confirm_signup():
         if not environment.parsed_options.allow_signup:
             return redirect(url_for("locust.login"))
+        if not session.get("user_sub_id"):
+            session["auth_sign_up_error"] = "An unexpected error occured. Please try again."
+            return redirect(url_for("locust_cloud_auth.signup"))
 
         session["auth_sign_up_error"] = ""
         confirmation_code = request.form.get("confirmation_code")

{locust_cloud-1.11.2.dist-info → locust_cloud-1.11.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: locust-cloud
-Version: 1.11.2
+Version: 1.11.4
 Summary: Locust Cloud
 Project-URL: Homepage, https://locust.cloud
 Requires-Python: >=3.11

{locust_cloud-1.11.2.dist-info → locust_cloud-1.11.4.dist-info}/RECORD

@@ -1,5 +1,5 @@
 locust_cloud/__init__.py,sha256=Mibsyg96q15lEs7pmQ-zQ-MTLp0k2oYgseORKjsDYAI,3620
-locust_cloud/auth.py,sha256=lypbT4_KL42JzQl_51YuPlhTjJuwhNiD6GSYG0L9c74,16154
+locust_cloud/auth.py,sha256=76pucGMYyn2G1Gj3mpEQwwGcJ-pMrHT5iD44_QWDLok,16783
 locust_cloud/cloud.py,sha256=lecZ5SRHuSfYQ15Heo5qxVA-kLjD-JEbfbSdvsdB4ro,15267
 locust_cloud/credential_manager.py,sha256=MiESqS8IOxncBON04irK1-seF-VNAHHTcELcyxp1ZR8,5571
 locust_cloud/idle_exit.py,sha256=bYD6LeUa0d8Ev2b1f-tiRXVnfyolCoACcDWC8zizNPE,1189
@@ -17,7 +17,7 @@ locust_cloud/webui/vite.config.ts,sha256=cqxPMkbwEA3H9mGGbuPulQUhIHCosUqm_1usxzs
 locust_cloud/webui/yarn.lock,sha256=52aat3Nl-O3OVBRrKO4RMpNJFCIpodvop0rxp0iWo-E,220316
 locust_cloud/webui/dist/index.html,sha256=uvHqVmQMDdDm0n8RbDoNeLMl1Y_uLtstxRf__hz5idc,664
 locust_cloud/webui/dist/assets/index-BpUoQtjh.js,sha256=Q99aUgCZb4a_zeXtPG3XC29TRCxYbuWJVKjKMC56WOg,2865980
-locust_cloud-1.11.2.dist-info/METADATA,sha256=6DMIDl1Cg05xHvK7iMmkC08NfUFm5PPHtwt5wWac1-s,452
-locust_cloud-1.11.2.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-locust_cloud-1.11.2.dist-info/entry_points.txt,sha256=PGyAb4e3aTsGS3N3VGShDl6VzJaXy7QwsEgsLOC7V00,57
-locust_cloud-1.11.2.dist-info/RECORD,,
+locust_cloud-1.11.4.dist-info/METADATA,sha256=GjPk5zgfKQZPS3GLWu5lF64Ld4CZ3HEHMuIP_EH_PvU,452
+locust_cloud-1.11.4.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+locust_cloud-1.11.4.dist-info/entry_points.txt,sha256=PGyAb4e3aTsGS3N3VGShDl6VzJaXy7QwsEgsLOC7V00,57
+locust_cloud-1.11.4.dist-info/RECORD,,