Skip to content

Commit 7d547f7

Browse files
committed
fix pluck() method
1 parent 458f639 commit 7d547f7

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

simple_query_builder/querybuilder.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -212,7 +212,15 @@ def column(self, column: int = 0) -> Union[tuple, list, dict, None]:
212212
self.query("", (), self._FETCH_COLUMN, column)
213213
return self._result
214214

215-
def pluck(self, key: int = 0, column: int = 1) -> Union[tuple, list, dict, None]:
215+
def pluck(self, key: Union[str, int] = 0, column: Union[str, int] = 1):
216+
if (
217+
self._result_dict and (isinstance(key, int) or isinstance(column, int))
218+
) or (
219+
not self._result_dict and (isinstance(key, str) or isinstance(column, str))
220+
):
221+
self.set_error(f"Incorrect type of key or column in {inspect.stack()[0][3]} method. Result dict is {self._result_dict}")
222+
return self
223+
216224
self.query()
217225
return [(x[key], x[column]) for x in self._result]
218226

0 commit comments

Comments
 (0)