add type hints for query() method

co0lc0der · co0lc0der · commit 584d2657fe3f · 2023-03-18T17:10:56.000+05:00
diff --git a/simple_query_builder/querybuilder.py b/simple_query_builder/querybuilder.py
@@ -98,7 +98,7 @@ def _set_row_factory(self, result_dict: bool = True):
                 [(col[0], r[idx]) for idx, col in enumerate(c.description)]
             )
 
-    def query(self, sql: str = "", params=(), fetch=2, column=0):
+    def query(self, sql: str = "", params: tuple = (), fetch: int = 2, column: Union[str, int] = 0):
         if fetch == 2:
             fetch = self._FETCH_ALL
 

Original file line number	Diff line number	Diff line change
`@@ -98,7 +98,7 @@ def _set_row_factory(self, result_dict: bool = True):`
`98`	`98`	`[(col[0], r[idx]) for idx, col in enumerate(c.description)]`
`99`	`99`	`)`
`100`	`100`
`101`		`- def query(self, sql: str = "", params=(), fetch=2, column=0):`
	`101`	`+ def query(self, sql: str = "", params: tuple = (), fetch: int = 2, column: Union[str, int] = 0):`
`102`	`102`	`if fetch == 2:`
`103`	`103`	`fetch = self._FETCH_ALL`
`104`	`104`