adding dataclass to record model (#33)

Co-authored-by: Dan Sikes <[email protected]>
vaphes · Apr 9, 2023 · c0afb20 · c0afb20
1 parent 90bd223
commit c0afb20
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pocketbase/models/record.py b/pocketbase/models/record.py
@@ -2,12 +2,13 @@
 
 from pocketbase.models.utils.base_model import BaseModel
 from pocketbase.utils import camel_to_snake
+from dataclasses import dataclass, field
 
-
+@dataclass
 class Record(BaseModel):
     collection_id: str
-    collection_name: str
-    expand: dict
+    collection_name: str = ""
+    expand: dict = field(default_factory=dict)
 
     def load(self, data: dict) -> None:
         super().load(data)