refactor: Enum models also inherit from 'str'

So they can be compared with strings.
python-scim · May 24, 2024 · 61399c0 · 61399c0
1 parent a505ad5
commit 61399c0
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 11 deletions.
diff --git a/pydantic_scim2/responses.py b/pydantic_scim2/responses.py
@@ -33,7 +33,7 @@ def unprocessable(cls, detail: str = "Unprocessable Entity") -> "SCIMError":
         return cls(detail=detail, status=422)
 
 
-class PatchOp(Enum):
+class PatchOp(str, Enum):
     replace = "replace"
     remove = "remove"
     add = "add"

diff --git a/pydantic_scim2/schema.py b/pydantic_scim2/schema.py
@@ -7,7 +7,7 @@
 from .resource import Meta
 
 
-class AttributeKind(Enum):
+class AttributeKind(str, Enum):
     string = "string"
     boolean = "boolean"
     decimal = "decimal"
@@ -18,21 +18,21 @@ class AttributeKind(Enum):
     complex = "complex"
 
 
-class Mutability(Enum):
+class Mutability(str, Enum):
     readOnly = "readOnly"
     readWrite = "readWrite"
     immutable = "immutable"
     writeOnly = "writeOnly"
 
 
-class Returned(Enum):
+class Returned(str, Enum):
     always = "always"
     never = "never"
     default = "default"
     request = "request"
 
 
-class Uniqueness(Enum):
+class Uniqueness(str, Enum):
     none = "none"
     server = "server"
     global_ = "global"

diff --git a/pydantic_scim2/service_provider.py b/pydantic_scim2/service_provider.py
@@ -59,7 +59,7 @@ class ETag(BaseModel):
     """A Boolean value specifying whether or not the operation is supported."""
 
 
-class AuthenticationSchemeKind(Enum):
+class AuthenticationSchemeKind(str, Enum):
     oauth = "oauth"
     oauth2 = "oauth2"
     oauthbearertoken = "oauthbearertoken"

diff --git a/pydantic_scim2/user.py b/pydantic_scim2/user.py
@@ -38,7 +38,7 @@ class Name(BaseModel):
     )
 
 
-class EmailKind(Enum):
+class EmailKind(str, Enum):
     work = "work"
     home = "home"
     other = "other"
@@ -63,7 +63,7 @@ class Email(BaseModel):
     )
 
 
-class PhoneNumberKind(Enum):
+class PhoneNumberKind(str, Enum):
     work = "work"
     home = "home"
     mobile = "mobile"
@@ -88,7 +88,7 @@ class PhoneNumber(BaseModel):
     )
 
 
-class ImKind(Enum):
+class ImKind(str, Enum):
     aim = "aim"
     gtalk = "gtalk"
     icq = "icq"
@@ -117,7 +117,7 @@ class Im(BaseModel):
     )
 
 
-class PhotoKind(Enum):
+class PhotoKind(str, Enum):
     photo = "photo"
     thumbnail = "thumbnail"
 
@@ -138,7 +138,7 @@ class Photo(BaseModel):
     )
 
 
-class AddressKind(Enum):
+class AddressKind(str, Enum):
     work = "work"
     home = "home"
     other = "other"