GH-9: Each client config can have its claim mappings

Author: ArtyomVancyan

src/fastapi_oauth2/client.py

@@ -1,9 +1,12 @@
 from typing import Optional
 from typing import Sequence
 from typing import Type
+from typing import Union
 
 from social_core.backends.oauth import BaseOAuth2
 
+from .claims import Claims
+
 
 class OAuth2Client:
     """OAuth2 client configuration for a single provider."""
@@ -13,6 +16,7 @@ class OAuth2Client:
     client_secret: str
     redirect_uri: Optional[str]
     scope: Optional[Sequence[str]]
+    claims: Optional[Union[Claims, dict]]
 
     def __init__(
             self,
@@ -22,9 +26,11 @@ def __init__(
             client_secret: str,
             redirect_uri: Optional[str] = None,
             scope: Optional[Sequence[str]] = None,
+            claims: Optional[Union[Claims, dict]] = None,
     ) -> None:
         self.backend = backend
         self.client_id = client_id
         self.client_secret = client_secret
         self.redirect_uri = redirect_uri
         self.scope = scope or []
+        self.claims = Claims(claims)