Merge pull request #243 from dbcli/krace/sqlean-stubs

Add sqlean stubs

Author: kracekumar

CHANGELOG.md

@@ -3,6 +3,7 @@
 ### Internal
 
 - Switch mypy to ty for type checking. [(#242)](https://github.com/dbcli/litecli/pull/242/files)
+- Add sqlean-stubs for type checking. [(#243)(https://github.com/dbcli/litecli/pull/243/files)]
 
 ## 1.17.0 - 2025-09-28
 

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

@@ -34,7 +34,8 @@ litecli = "litecli.main:cli"
 
 [project.optional-dependencies]
 ai = ["llm"]
-sqlean = ["sqlean-py>=3.47.0"]
+sqlean = ["sqlean-py>=3.47.0", 
+    "sqlean-stubs>=0.0.3"]
 
 dev = [
     "behave>=1.2.6",
@@ -45,7 +46,7 @@ dev = [
     "tox>=4.8.0",
     "pdbpp>=0.10.3",
     "llm>=0.19.0",
-    "ty>=0.0.4",
+    "ty>=0.0.4"
 ]
 
 [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
 

tox.ini

@@ -9,7 +9,7 @@ commands = uv pip install -e .[dev]
 
 [testenv:sqlean]
 deps = uv
-commands = uv pip install -e .[dev,sqlean]
+commands = uv pip install -e .[dev,sqlean,sqlean-stubs]
         coverage run -m pytest -v tests
         coverage report -m