Skip to content

Commit 6b11a4c

Browse files
committed
fix: handle completion refresh without db
1 parent 112fd66 commit 6b11a4c

2 files changed

Lines changed: 10 additions & 3 deletions

File tree

CHANGELOG.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,10 @@
44

55
- Make LLM support optional and installable via `litecli[ai]`.
66

7+
### Bug Fixes
8+
9+
- Avoid completion refresh crashes when no database is connected.
10+
711
## 1.18.0
812

913
### Internal

litecli/sqlexecute.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,8 @@ def get_result(self, cursor: Any) -> tuple[str | None, list | None, list | None,
179179

180180
def tables(self) -> Generator[tuple[str], None, None]:
181181
"""Yields table names"""
182-
assert self.conn is not None
182+
if not self.conn:
183+
return
183184
with closing(self.conn.cursor()) as cur:
184185
_logger.debug("Tables Query. sql: %r", self.tables_query)
185186
cur.execute(self.tables_query)
@@ -188,7 +189,8 @@ def tables(self) -> Generator[tuple[str], None, None]:
188189

189190
def table_columns(self) -> Generator[tuple[str, str], None, None]:
190191
"""Yields column names"""
191-
assert self.conn is not None
192+
if not self.conn:
193+
return
192194
with closing(self.conn.cursor()) as cur:
193195
_logger.debug("Columns Query. sql: %r", self.table_columns_query)
194196
cur.execute(self.table_columns_query)
@@ -206,7 +208,8 @@ def databases(self) -> Generator[str, None, None]:
206208

207209
def functions(self) -> Iterable[tuple]:
208210
"""Yields tuples of (schema_name, function_name)"""
209-
assert self.conn is not None
211+
if not self.conn:
212+
return
210213
with closing(self.conn.cursor()) as cur:
211214
_logger.debug("Functions Query. sql: %r", self.functions_query)
212215
cur.execute(self.functions_query % self.dbname)

0 commit comments

Comments
 (0)