Skip to content

Commit b2540fc

Browse files
committed
small types fixes
1 parent cc826f0 commit b2540fc

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

simple_query_builder/querybuilder.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -138,27 +138,28 @@ def get_result(self) -> Union[tuple, list]:
138138
def get_count(self) -> int:
139139
return self._count
140140

141-
def reset(self) -> None:
141+
def reset(self) -> bool:
142142
self._sql = ''
143143
self._params = ()
144144
self._query = None
145145
self._result = []
146146
self._count = -1
147147
self.set_error()
148+
return True
148149

149-
def all(self) -> Union[list, dict]:
150+
def all(self) -> Union[tuple, list, dict, None]:
150151
self.query()
151152
return self._result
152153

153-
def one(self) -> Union[list, dict]:
154+
def one(self) -> Union[tuple, list, dict, None]:
154155
self.query(self._sql, self._params, self._FETCH_ONE)
155156
return self._result
156157

157158
def go(self) -> Union[int, None]:
158159
self.query(self._sql, self._params, self._NO_FETCH)
159160
return self._cur.lastrowid
160161

161-
def column(self, column=0):
162+
def column(self, column: int = 0) -> Union[tuple, list, dict, None]:
162163
self.query('', (), self._FETCH_COLUMN, column)
163164
return self._result
164165

0 commit comments

Comments
 (0)