fix: handle with missing param about RelayState

python-social-auth · Aug 1, 2023 · 5808d6d · 5808d6d
1 parent cf95380
commit 5808d6d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/social_core/backends/saml.py b/social_core/backends/saml.py
@@ -302,7 +302,11 @@ def auth_complete(self, *args, **kwargs):
         The user has been redirected back from the IdP and we should
         now log them in, if everything checks out.
         """
-        idp_name = self.strategy.request_data()["RelayState"]
+        try:
+            idp_name = self.strategy.request_data()["RelayState"]
+        except KeyError:
+            raise AuthMissingParameter(self, "RelayState")
+
         idp = self.get_idp(idp_name)
         auth = self._create_saml_auth(idp)
         auth.process_response()