Get primary key from schema instance rather

than schema class
rbw · Sep 22, 2020 · f76648e · f76648e
1 parent 5e05818
commit f76648e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/aiosnow/models/common/base.py b/aiosnow/models/common/base.py
@@ -73,12 +73,12 @@ class BaseModel(metaclass=BaseModelMeta):
 
     def __init__(self, client: Client):
         self._client = client
-        self._primary_key = getattr(self.schema_cls, "_primary_key")
         self.fields = dict(self.schema_cls.fields)
         self.nested_fields = {
             n: f for n, f in self.fields.items() if isinstance(f, Nested)
         }
         self.schema = self.schema_cls(unknown=marshmallow.EXCLUDE)
+        self._primary_key = getattr(self.schema, "_primary_key")
 
     @property
     @abstractmethod

diff --git a/aiosnow/models/table/model.py b/aiosnow/models/table/model.py
@@ -143,8 +143,8 @@ async def get_object_id(self, value: Union[Condition, str]) -> str:
         """
 
         if isinstance(value, Condition):
-            record = await self.get_one(value, return_only=[self._primary_key])
-            return record.__dict__[self._primary_key]
+            response = await self.get_one(value, return_only=[self._primary_key])
+            return response.data[self._primary_key]
         elif isinstance(value, str):
             return value
         else: