use pydantic for data models

Author: tomvictor

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "aws_packages"
-version = "0.0.15"
+version = "0.0.16"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
@@ -17,5 +17,6 @@ name = "aws_packages._aws_packages"
 
 dependencies = [
     "boto3==1.34.79",
-    "PyJWT==2.8.0"
+    "PyJWT==2.8.0",
+    "pydantic==2.6.4"
 ]

aws_packages/auth/models.py

@@ -1,8 +1,7 @@
-from dataclasses import dataclass
+from pydantic import BaseModel
 
 
-@dataclass
-class AuthorizationRequest:
+class AuthorizationRequest(BaseModel):
     """Authorization request model"""
 
     Authorization: str
@@ -12,26 +11,37 @@ def token(self):
         return self.Authorization
 
 
-@dataclass
-class AuthenticationRequest:
+class AuthenticationRequest(BaseModel):
     """Authentication request model"""
 
     user: str
     authcode: str
 
 
-@dataclass
-class User:
+class User(BaseModel):
     """User model."""
 
     principal: str
 
 
-@dataclass
-class LoginResponse:
+class LoginResponse(BaseModel):
     """Login response model"""
 
     access_token: str
 
     def as_dict(self):
         return {"access_token": self.access_token}
+
+
+class SuccessResponse(BaseModel):
+    """Success response for simple http request."""
+
+    message: str
+
+
+if __name__ == "__main__":
+    args = {
+        "Authorization": "123",
+    }
+    request = AuthorizationRequest(**args)
+    print(request)

requirements.txt

@@ -4,4 +4,5 @@ aws-lambda-powertools==2.36.0
 boto3==1.34.79
 aws-cdk.aws-lambda-python-alpha==2.121.1a0
 PyJWT==2.8.0
-ic-py==1.0.1
+ic-py==1.0.1
+pydantic==2.6.4