@@ -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