makes type simpler

Author: rishabhpoddar

supertokens_python/recipe/multifactorauth/recipe_implementation.py

@@ -147,7 +147,7 @@ async def get_mfa_requirements_for_auth(
             all_factors.add(factor)
         for factor in await required_secondary_factors_for_tenant():
             all_factors.add(factor)
-        return MFARequirementList({"oneOf": list(all_factors)})
+        return [{"oneOf": list(all_factors)}]
 
     async def assert_allowed_to_setup_factor_else_throw_invalid_claim_error(
         self,

supertokens_python/recipe/multifactorauth/types.py

@@ -22,24 +22,9 @@
     from .interfaces import RecipeInterface, APIInterface
 
 
-class MFARequirementList(List[Union[Dict[str, List[str]], str]]):
-    def __init__(
-        self,
-        *args: Union[
-            str, Dict[Union[Literal["oneOf"], Literal["allOfInAnyOrder"]], List[str]]
-        ],
-    ):
-        super().__init__()
-        for arg in args:
-            if isinstance(arg, str):
-                self.append(arg)
-            else:
-                if "oneOf" in arg:
-                    self.append({"oneOf": arg["oneOf"]})
-                elif "allOfInAnyOrder" in arg:
-                    self.append({"allOfInAnyOrder": arg["allOfInAnyOrder"]})
-                else:
-                    raise ValueError("Invalid dictionary format")
+MFARequirementList = List[
+    Union[str, Dict[Union[Literal["oneOf"], Literal["allOfInAnyOrder"]], List[str]]]
+]
 
 
 class MFAClaimValue:

tests/test-server/test_functions_mapper.py

@@ -135,11 +135,7 @@ async def get_mfa_requirements_for_auth(
                 required_secondary_factors_for_tenant: Any,
                 user_context: Dict[str, Any],
             ) -> MFARequirementList:
-                return (
-                    MFARequirementList("otp-phone")
-                    if user_context.get("requireFactor")
-                    else MFARequirementList()
-                )
+                return ["otp-phone"] if user_context.get("requireFactor") else []
 
             original_implementation.get_mfa_requirements_for_auth = (
                 get_mfa_requirements_for_auth