Merge pull request #4 from reflex-dev/user-info-total-false

Mark AzureUserInfo as total=False

Author: masenf

reflex_azure_auth/types.py

@@ -1,23 +1,23 @@
 """Shared types and helpers for Azure user info."""
 
-from typing import TypedDict
+from typing import Required, TypedDict
 
 
-class AzureUserInfo(TypedDict):
+class AzureUserInfo(TypedDict, total=False):
     """TypedDict representing user information from Azure / Microsoft identity platform.
 
     Contains user profile data returned by the /userinfo endpoint following
     successful authentication (OpenID Connect standard claims).
     """
 
-    sub: str
-    email: str | None
-    name: str | None
-    given_name: str | None
-    middle_name: str | None
-    family_name: str | None
-    picture: str | None
-    locale: str | None
+    sub: Required[str]
+    email: str
+    name: str
+    given_name: str
+    middle_name: str
+    family_name: str
+    picture: str
+    locale: str
 
 
 # Microsoft API returns these without the underscore... not sure why
@@ -33,13 +33,4 @@ def user_info_from_dict(data: dict) -> AzureUserInfo:
     for mapped_name, real_name in user_info_mapping.items():
         if mapped_name in data and real_name not in data:
             data[real_name] = data.pop(mapped_name)
-    return AzureUserInfo(
-        sub=data["sub"],
-        email=data.get("email"),
-        name=data.get("name"),
-        given_name=data.get("given_name"),
-        middle_name=data.get("middle_name"),
-        family_name=data.get("family_name"),
-        picture=data.get("picture"),
-        locale=data.get("locale"),
-    )
+    return AzureUserInfo(**data)