more robust handling of list like objects values

lolipopshock · Feb 22, 2022 · a6a9506 · a6a9506
1 parent fd11fea
commit a6a9506
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/notion_df/values.py b/src/notion_df/values.py
@@ -19,7 +19,10 @@
     FileObject,
     FormulaObject
 )
-from notion_df.utils import flatten_dict
+from notion_df.utils import (
+    flatten_dict,
+    is_list_like
+)
 
 
 class BasePropertyValues(BaseModel):
@@ -106,7 +109,7 @@ def value(self) -> List[str]:
 
     @classmethod
     def from_value(cls, values: Union[List[str], str]):
-        if isinstance(values, list):
+        if is_list_like(values):
             return cls(
                 multi_select=[SelectOption.from_value(value) for value in values]
             )
@@ -143,7 +146,7 @@ def value(self) -> List[str]:
 
     @classmethod
     def from_value(cls, values: Union[List[str], str]):
-        if isinstance(values, list):
+        if is_list_like(values):
             return cls(relation=[RelationObject.from_value(value) for value in values])
         else:
             return cls(relation=[RelationObject.from_value(values)])
@@ -158,7 +161,7 @@ def value(self) -> List[str]:
 
     @classmethod
     def from_value(cls, values: Union[List[str], str]):
-        if isinstance(values, list):
+        if is_list_like(values):
             return cls(people=[UserObject.from_value(value) for value in values])
         else:
             return cls(people=[UserObject.from_value(values)])