Merge pull request #101 from Naruto0/master

sqlite3 stub for 2.7  and 3

stdlib/2.7/sqlite3/dbapi2.pyi

@@ -1,15 +1,17 @@
-# Stubs for sqlite3.dbapi2 (Python 3.5)
-#
-# NOTE: This dynamically typed stub was automatically generated by stubgen.
+# Filip Hron <filip.hron@gmail.com>
+# based heavily on Andrey Vlasovskikh's python-skeletons https://github.com/JetBrains/python-skeletons/blob/master/sqlite3.py
 
-from typing import Any
+from typing import Any, Union, List
+from numbers import Integral
+from datetime import time, datetime
+from collections import Iterable
 
-paramstyle = ... # type: Any
-threadsafety = ... # type: Any
-apilevel = ... # type: Any
-Date = ... # type: Any
-Time = ... # type: Any
-Timestamp = ... # type: Any
+paramstyle = ... # type: str
+threadsafety = ... # type: int
+apilevel = ... # type: str
+Date = ... # type: datetime
+Time = ... # type: time
+Timestamp = ... # type: datetime
 
 def DateFromTicks(ticks): ...
 def TimeFromTicks(ticks): ...
@@ -61,18 +63,26 @@ converters = ... # type: Any
 sqlite_version = ... # type: str
 version = ... # type: str
 
+# TODO: adapt needs to get probed
 def adapt(obj, protocol, alternate): ...
-def complete_statement(sql): ...
-def connect(*args, **kwargs): ...
-def enable_callback_tracebacks(flag): ...
-def enable_shared_cache(do_enable): ...
-def register_adapter(type, callable): ...
-def register_converter(typename, callable): ...
+def complete_statement(sql: str) -> bool: ...
+def connect(database: Union[bytes, unicode],
+            timeout: float = ... ,
+            detect_types: int = ...,
+            isolation_level: Union[str, None] = ...,
+            check_same_thread: bool = ...,
+            factory: Union[Connection, None] = ...,
+            cached_statements: int = ...) -> Connection: ...
+def enable_callback_tracebacks(flag: bool) -> None: ...
+def enable_shared_cache(do_enable: int) -> None: ...
+def register_adapter(type: type, callable: Any) -> None: ...
+# TODO: sqlite3.register_converter.__doc__ specifies callable as unknown
+def register_converter(typename: str, callable: bytes) -> None: ...
 
 class Cache:
-    def __init__(self, *args, **kwargs): ...
-    def display(self, *args, **kwargs): ...
-    def get(self, *args, **kwargs): ...
+    def __init__(self, *args, **kwargs) -> None: ...
+    def display(self, *args, **kwargs) -> None: ...
+    def get(self, *args, **kwargs) -> None: ...
 
 class Connection:
     DataError = ... # type: Any
@@ -91,20 +101,25 @@ class Connection:
     text_factory = ... # type: Any
     total_changes = ... # type: Any
     def __init__(self, *args, **kwargs): ...
-    def close(self, *args, **kwargs): ...
-    def commit(self, *args, **kwargs): ...
-    def create_aggregate(self, *args, **kwargs): ...
-    def create_collation(self, *args, **kwargs): ...
-    def create_function(self, *args, **kwargs): ...
-    def cursor(self, *args, **kwargs): ...
-    def execute(self, *args, **kwargs): ...
-    def executemany(self, *args, **kwargs): ...
-    def executescript(self, *args, **kwargs): ...
-    def interrupt(self, *args, **kwargs): ...
-    def iterdump(self, *args, **kwargs): ...
+    def close(self) -> None: ...
+    def commit(self) -> None: ...
+    def create_aggregate(self, name: str, num_params: int, aggregate_class: type) -> None: ...
+    def create_collation(self, name: str, callable: Any) -> None: ...
+    def create_function(self, name: str, num_params: int, func: Any) -> None: ...
+    def cursor(self, cursorClass= Union[type, None]) -> Cursor: ...
+    def execute(self, sql: str, parameters: Iterable) -> Cursor: ...
+    # TODO: please check in executemany() if seq_of_parameters type is possible like this
+    def executemany(self, sql: str, seq_of_parameters: Iterable[Iterable]) -> Cursor: ...
+    def executescript(self, sql_script: Union[bytes, unicode]) -> Cursor: ...
+    def interrupt(self, *args, **kwargs) -> None: ...
+    def iterdump(self, *args, **kwargs) -> None: ...
     def rollback(self, *args, **kwargs): ...
-    def set_authorizer(self, *args, **kwargs): ...
-    def set_progress_handler(self, *args, **kwargs): ...
+    # TODO: set_authorizer(authorzer_callback)
+    # see https://docs.python.org/2/library/sqlite3.html#sqlite3.Connection.set_authorizer
+    # returns [SQLITE_OK, SQLITE_DENY, SQLITE_IGNORE] so perhaps int
+    def set_authorizer(self, *args, **kwargs) -> None: ...
+    # set_progress_handler(handler, n) -> see https://docs.python.org/2/library/sqlite3.html#sqlite3.Connection.set_progress_handler
+    def set_progress_handler(self, *args, **kwargs) -> None: ...
     def set_trace_callback(self, *args, **kwargs): ...
     def __call__(self, *args, **kwargs): ...
     def __enter__(self, *args, **kwargs): ...
@@ -117,14 +132,17 @@ class Cursor:
     lastrowid = ... # type: Any
     row_factory = ... # type: Any
     rowcount = ... # type: Any
+    # TODO: Cursor class accepts exactly 1 argument
+    # required type is sqlite3.Connection (which is imported as _Connection)
+    # however, the name of the __init__ variable is unknown
     def __init__(self, *args, **kwargs): ...
     def close(self, *args, **kwargs): ...
-    def execute(self, *args, **kwargs): ...
-    def executemany(self, *args, **kwargs): ...
-    def executescript(self, *args, **kwargs): ...
-    def fetchall(self, *args, **kwargs): ...
-    def fetchmany(self, *args, **kwargs): ...
-    def fetchone(self, *args, **kwargs): ...
+    def execute(self, sql: str, parameters: Iterable) -> Cursor: ...
+    def executemany(self, sql: str, seq_of_parameters: Iterable[Iterable]): ...
+    def executescript(self, sql_script: Union[bytes, unicode]) -> Cursor: ...
+    def fetchall(self) -> List[tuple]: ...
+    def fetchmany(self, size: Integral = ...) -> List[tuple]: ...
+    def fetchone(self) -> Union[tuple, None]: ...
     def setinputsizes(self, *args, **kwargs): ...
     def setoutputsize(self, *args, **kwargs): ...
     def __iter__(self): ...