Updated Post models to include like data

postly/common/models.py

@@ -1,5 +1,5 @@
-from dataclasses import dataclass
-from typing import List
+from dataclasses import dataclass, field
+from typing import List, Set
 
 from pydantic import BaseModel
 
@@ -8,6 +8,8 @@ class StrictPost(BaseModel):
     content: str
     timestamp: int
     topics: List[str]
+    likes: int = 0
+    liked_by: Set[str] = set()
 
 
 @dataclass
@@ -15,11 +17,25 @@ class Post:
     content: str
     timestamp: int
     topics: List[str]
+    likes: int = 0
+    liked_by: Set[str] = field(default_factory=set)
 
 
 if __name__ == "__main__":
     # this should be OK, as not strictly typed
-    Post(content=1, timestamp=1, topics=["1"])
+    post = Post(content="1", timestamp=1, topics=["1"])
+    post.likes += 1
+    post.liked_by.add("user1")
+    print(post)
 
     # this should result in a validation error, as pydantic enforces strict typing on runtime
-    StrictPost(content=1, timestamp=1, topics=["1"])
+    try:
+        strict_post = StrictPost(content=1, timestamp=1, topics=["1"])
+    except Exception as e:
+        print(f"Validation error: {e}")
+
+    # this should be OK, as strictly typed
+    strict_post = StrictPost(content="1", timestamp=1, topics=["1"])
+    strict_post.likes += 1
+    strict_post.liked_by.add("user1")
+    print(strict_post)