Skip to content

Commit dc7cfd3

Browse files
committed
add pluck() method
1 parent b2540fc commit dc7cfd3

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

simple_query_builder/querybuilder.py

+4
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,10 @@ def column(self, column: int = 0) -> Union[tuple, list, dict, None]:
163163
self.query('', (), self._FETCH_COLUMN, column)
164164
return self._result
165165

166+
def pluck(self, key: int = 0, column: int = 1) -> Union[tuple, list, dict, None]:
167+
self.query()
168+
return [(x[key], x[column]) for x in self._result]
169+
166170
def count(self, table: Union[str, dict], field: str = ''):
167171
if table == '' or table == {}:
168172
self.set_error(f"Empty table in {inspect.stack()[0][3]} method")

0 commit comments

Comments
 (0)