better stubs

django-stubs/db/backends/utils.pyi

@@ -1,54 +1,70 @@
-# Stubs for django.db.backends.utils (Python 3.6)
-#
-# NOTE: This dynamically typed stub was automatically generated by stubgen.
-
-from typing import Any, Optional
-
-from datetime import date, time
+from datetime import date, datetime, time
 from decimal import Decimal
-from django.db.backends.sqlite3.base import DatabaseWrapper, SQLiteCursorWrapper
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+from typing import Any, Callable, Dict, Iterator, List, Optional, Tuple, Union
+
+from django.db.backends.sqlite3.base import (DatabaseWrapper,
+                                             SQLiteCursorWrapper)
 
 logger: Any
 
 class CursorWrapper:
-    cursor: Any = ...
-    db: Any = ...
-    def __init__(self, cursor: SQLiteCursorWrapper, db: DatabaseWrapper) -> None: ...
+    cursor: django.db.backends.sqlite3.base.SQLiteCursorWrapper = ...
+    db: django.db.backends.sqlite3.base.DatabaseWrapper = ...
+    def __init__(
+        self, cursor: SQLiteCursorWrapper, db: DatabaseWrapper
+    ) -> None: ...
     WRAP_ERROR_ATTRS: Any = ...
-    def __getattr__(self, attr: str) -> Any: ...
+    def __getattr__(
+        self, attr: str
+    ) -> Union[
+        Callable, int, Tuple[Tuple[str, None, None, None, None, None, None]]
+    ]: ...
     def __iter__(self) -> None: ...
     def __enter__(self) -> CursorWrapper: ...
     def __exit__(self, type: None, value: None, traceback: None) -> None: ...
     def callproc(
-        self, procname: str, params: List[Any] = ..., kparams: Dict[str, int] = ...
-    ): ...
-    def execute(self, sql: str, params: Any = ...) -> SQLiteCursorWrapper: ...
-    def executemany(self, sql: Any, param_list: Any): ...
-    def _execute_with_wrappers(
-        self, sql: str, params: Any, many: bool, executor: Callable
-    ) -> Optional[SQLiteCursorWrapper]: ...
-    def _execute(self, sql: str, params: Any, *ignored_wrapper_args: Any): ...
-    def _executemany(
+        self,
+        procname: str,
+        params: List[Any] = ...,
+        kparams: Dict[str, int] = ...,
+    ) -> Any: ...
+    def execute(
         self,
         sql: str,
-        param_list: Union[List[Tuple[int]], List[Tuple[int, int, int]]],
-        *ignored_wrapper_args: Any,
-    ): ...
+        params: Optional[
+            Union[
+                List[Union[int, str, None, memoryview]],
+                List[Union[str, float, None]],
+                List[datetime],
+                Tuple,
+            ]
+        ] = ...,
+    ) -> Optional[SQLiteCursorWrapper]: ...
+    def executemany(
+        self,
+        sql: str,
+        param_list: Union[List[Tuple[str]], List[Tuple[int]], Iterator[Any]],
+    ) -> Optional[SQLiteCursorWrapper]: ...
 
 class CursorDebugWrapper(CursorWrapper):
-    def execute(self, sql: str, params: Optional[Union[List[str], Tuple]] = ...): ...
-    def executemany(self, sql: Any, param_list: Any): ...
+    cursor: django.db.backends.sqlite3.base.SQLiteCursorWrapper
+    db: django.db.backends.sqlite3.base.DatabaseWrapper
+    def execute(
+        self, sql: str, params: Optional[Union[Tuple, List[str]]] = ...
+    ) -> Any: ...
+    def executemany(self, sql: str, param_list: Iterator[Any]) -> Any: ...
 
-def typecast_date(s: str) -> date: ...
-def typecast_time(s: str) -> Optional[time]: ...
-def typecast_timestamp(s: str) -> date: ...
+def typecast_date(s: Optional[str]) -> Optional[date]: ...
+def typecast_time(s: Optional[str]) -> Optional[time]: ...
+def typecast_timestamp(s: Optional[str]) -> Optional[date]: ...
 def rev_typecast_decimal(d: Decimal) -> str: ...
 def split_identifier(identifier: str) -> Tuple[str, str]: ...
 def truncate_name(
     identifier: str, length: Optional[int] = ..., hash_len: int = ...
 ) -> str: ...
 def format_number(
-    value: Optional[Decimal], max_digits: Optional[int], decimal_places: Optional[int]
+    value: Optional[Decimal],
+    max_digits: Optional[int],
+    decimal_places: Optional[int],
 ) -> Optional[str]: ...
 def strip_quotes(table_name: str) -> str: ...