add: pydantic type validation

pyproject.toml

@@ -16,6 +16,7 @@ bech32 = "^1.2.0"
 pycryptodome = "^3.20.0"
 pyjwt = "^2.8.0"
 eciespy = "^0.4.2"
+pydantic = "^2.8.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.2.2"

src/babble/auth.py

@@ -1,17 +1,16 @@
-from dataclasses import dataclass
 from datetime import datetime, timezone
 from typing import Optional, Tuple
 
 import jwt
 import requests
+from pydantic import BaseModel
 
 from .config import AUTH_SERVER, DEFAULT_REQUEST_TIMEOUT
 from .crypto.identity import Identity
 from .encoding import from_base64, to_base64
 
 
-@dataclass
-class TokenMetadata:
+class TokenMetadata(BaseModel):
     address: str
     public_key: str
     issued_at: datetime

src/babble/client.py

@@ -1,8 +1,8 @@
-from dataclasses import dataclass
 from datetime import datetime, timedelta, timezone
 from typing import List
 
 import bech32
+from pydantic import BaseModel
 
 from .auth import authenticate
 from .crypto.exceptions import RoutingError
@@ -24,8 +24,7 @@ def _validate_address(address: str):
         raise ValueError(f"Bad delegate address {address}")
 
 
-@dataclass
-class Message:
+class Message(BaseModel):
     id: str
     sender: str
     target: str

src/babble/mailbox.py

@@ -1,8 +1,8 @@
-from dataclasses import dataclass
 from datetime import datetime, timezone
 from typing import Any, Dict, List, Optional
 
 import requests
+from pydantic import BaseModel
 
 from .config import MEMORANDUM_SERVER
 
@@ -115,8 +115,7 @@ def dispatch_messages(token: str, messages: List[str]):
     )
 
 
-@dataclass
-class RawMessage:
+class RawMessage(BaseModel):
     id: str
     group_id: str
     sender: str