Common: Add CredentialComponentSchema

Extract common _strip_credential_type() post_load function from
PasswordSchema and UsernameSchema into a parent class.

monkey/common/credentials/credential_component_schema.py

@@ -0,0 +1,10 @@
+from marshmallow import Schema, post_load
+
+from common.utils.code_utils import del_key
+
+
+class CredentialComponentSchema(Schema):
+    @post_load
+    def _strip_credential_type(self, data, **kwargs):
+        del_key(data, "credential_type")
+        return data

monkey/common/credentials/password.py

@@ -1,24 +1,18 @@
 from dataclasses import dataclass, field
 
-from marshmallow import Schema, fields, post_load, validate
+from marshmallow import fields, validate
 from marshmallow_enum import EnumField
 
-from common.utils.code_utils import del_key
-
 from . import CredentialComponentType, ICredentialComponent
+from .credential_component_schema import CredentialComponentSchema
 
 
-class PasswordSchema(Schema):
+class PasswordSchema(CredentialComponentSchema):
     credential_type = EnumField(
         CredentialComponentType, validate=validate.Equal(CredentialComponentType.PASSWORD)
     )
     password = fields.Str()
 
-    @post_load
-    def _strip_credential_type(self, data, **kwargs):
-        del_key(data, "credential_type")
-        return data
-
 
 @dataclass(frozen=True)
 class Password(ICredentialComponent):

monkey/common/credentials/username.py

@@ -1,24 +1,18 @@
 from dataclasses import dataclass, field
 
-from marshmallow import Schema, fields, post_load, validate
+from marshmallow import fields, validate
 from marshmallow_enum import EnumField
 
-from common.utils.code_utils import del_key
-
 from . import CredentialComponentType, ICredentialComponent
+from .credential_component_schema import CredentialComponentSchema
 
 
-class UsernameSchema(Schema):
+class UsernameSchema(CredentialComponentSchema):
     credential_type = EnumField(
         CredentialComponentType, validate=validate.Equal(CredentialComponentType.USERNAME)
     )
     username = fields.Str()
 
-    @post_load
-    def _strip_credential_type(self, data, **kwargs):
-        del_key(data, "credential_type")
-        return data
-
 
 @dataclass(frozen=True)
 class Username(ICredentialComponent):