Add sqlean stubs

Author: kracekumar

litecli/main.py

@@ -13,7 +13,7 @@
 from io import open
 
 try:
-    from sqlean import OperationalError, sqlite_version  # type: ignore[import-untyped]
+    from sqlean import OperationalError, sqlite_version
 except ImportError:
     from sqlite3 import OperationalError, sqlite_version
 from time import time

litecli/sqlexecute.py

@@ -5,8 +5,8 @@
 from typing import Any, Generator, Iterable
 
 try:
-    import sqlean as sqlite3  # type: ignore[import-untyped]
-    from sqlean import OperationalError  # type: ignore[import-untyped]
+    import sqlean as sqlite3
+    from sqlean import OperationalError
 
     sqlite3.extensions.enable_all()
 except ImportError:
@@ -87,8 +87,7 @@ def connect(self, database: str | None = None) -> None:
             if not os.path.exists(db_dir_name):
                 raise Exception("Path does not exist: {}".format(db_dir_name))
 
-        # sqlean exposes the connect method during run-time
-        conn = sqlite3.connect(database=db_name, isolation_level=None, uri=uri)  # type: ignore[attr-defined]
+        conn = sqlite3.connect(database=db_name, isolation_level=None, uri=uri)
         conn.text_factory = lambda x: x.decode("utf-8", "backslashreplace")
         if self.conn:
             self.conn.close()

pyproject.toml

@@ -46,6 +46,7 @@ dev = [
     "pdbpp>=0.10.3",
     "llm>=0.19.0",
     "ty>=0.0.4",
+    "sqlean-stubs>=0.0.3",
 ]
 
 [tool.setuptools.packages.find]

tests/test_sqlexecute.py

@@ -7,7 +7,7 @@
 from .utils import assert_result_equal, dbtest, is_expanded_output, run, set_expanded_output
 
 try:
-    from sqlean import OperationalError, ProgrammingError  # type: ignore[import-untyped]
+    from sqlean import OperationalError, ProgrammingError
 except ImportError:
     from sqlite3 import OperationalError, ProgrammingError