Add attachment model

Author: lcd1232

.pre-commit-config.yaml

@@ -1,5 +1,5 @@
 exclude: '.git'
-default_stages: [ commit ]
+default_stages: [ pre-commit ]
 
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks

README.md

@@ -201,19 +201,19 @@ try:
     email = client.create_email()
     print(f"Created email: {email.email}")
     print(f"TTL: {email.ttl} seconds")
-    
+
     # List available domains
     domains = client.list_domains()
     print(f"Available domains: {len(domains)}")
-    
+
     # Check for messages (would be empty initially)
     messages = client.list_email_messages(email.email)
     print(f"Messages: {len(messages)}")
-    
+
     # Check rate limit
     rate_limit = client.get_rate_limit()
     print(f"Requests remaining: {rate_limit.remaining}/{rate_limit.limit}")
-    
+
 except AuthenticationError:
     print("Invalid API key")
 except RateLimitError:

tempmail/__init__.py

@@ -10,6 +10,7 @@
     Domain,
     EmailAddress,
     EmailMessage,
+    Attachment,
 )
 from .exceptions import (
     TempMailError,
@@ -23,6 +24,7 @@
     "RateLimit",
     "Domain",
     "EmailAddress",
+    "Attachment",
     "EmailMessage",
     "TempMailError",
     "AuthenticationError",

tempmail/client.py

@@ -148,7 +148,7 @@ def list_email_messages(
         data = self._make_request("GET", f"/v1/emails/{email}/messages")
 
         messages = []
-        for msg_data in data.get("messages", []):
+        for msg_data in data["messages"]:
             messages.append(EmailMessage.from_json(msg_data))
 
         return messages

tempmail/models.py

@@ -3,7 +3,7 @@
 import enum
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Optional, List, Dict, Any
+from typing import List, Dict, Any
 
 
 @dataclass
@@ -59,10 +59,18 @@ def from_json(cls, data: Dict[str, Any]) -> "EmailAddress":
 class Attachment:
     """Attachment information for an email message."""
 
-    filename: str
-    content_type: str
+    id: str
+    name: str
     size: int  # Size in bytes
 
+    @classmethod
+    def from_json(cls, data: Dict[str, Any]) -> "Attachment":
+        return cls(
+            id=data["id"],
+            name=data["name"],
+            size=data["size"],
+        )
+
 
 @dataclass
 class EmailMessage:
@@ -74,9 +82,9 @@ class EmailMessage:
     subject: str
     body_text: str
     created_at: datetime
-    cc: Optional[List[str]] = None
-    body_html: Optional[str] = None
-    attachments: Optional[List[Dict[str, Any]]] = None
+    cc: List[str]
+    body_html: str
+    attachments: List[Attachment]
 
     @classmethod
     def from_json(cls, data: Dict[str, Any]) -> "EmailMessage":
@@ -89,9 +97,9 @@ def from_json(cls, data: Dict[str, Any]) -> "EmailMessage":
             created_at=datetime.fromisoformat(
                 data["created_at"].replace("Z", "+00:00")
             ),
-            cc=data.get("cc", []),
-            body_html=data.get("body_html"),
-            attachments=data.get("attachments", []),
+            cc=data["cc"],
+            body_html=data["body_html"],
+            attachments=[Attachment.from_json(v) for v in data["attachments"]],
         )
 
 

tests/test_client.py

@@ -14,7 +14,7 @@
     TempMailError,
 )
 from requests.exceptions import ConnectionError
-from tempmail.models import DomainType, RateLimit
+from tempmail.models import DomainType, RateLimit, Attachment
 
 
 class TestTempMailClient:
@@ -116,7 +116,18 @@ def test_list_email_messages_success(self, mock_request):
                     "body_text": "Test body",
                     "body_html": "<p>Test body</p>",
                     "created_at": "2023-01-01T00:00:00Z",
-                    "attachments": [],
+                    "attachments": [
+                        {
+                            "id": "att1",
+                            "name": "file.txt",
+                            "size": 1234,
+                        },
+                        {
+                            "id": "att2",
+                            "name": "image.png",
+                            "size": 4567,
+                        },
+                    ],
                 }
             ]
         }
@@ -138,7 +149,10 @@ def test_list_email_messages_success(self, mock_request):
             created_at=datetime.datetime(
                 2023, 1, 1, 0, 0, tzinfo=datetime.timezone.utc
             ),
-            attachments=[],
+            attachments=[
+                Attachment(id="att1", name="file.txt", size=1234),
+                Attachment(id="att2", name="image.png", size=4567),
+            ],
         )
 
     @patch("tempmail.client.requests.Session.request")