fix(sqlite): list temp tables when no database is specified

ibis-project · Sep 7, 2024 · 415e298 · 415e298
1 parent 7bfc9f7
commit 415e298
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/ibis/backends/sqlite/__init__.py b/ibis/backends/sqlite/__init__.py
@@ -167,22 +167,24 @@ def list_tables(
             Database to list tables from. Default behavior is to show tables in
             the current database.
         """
-        if database is None:
+        if show_temp := (database is None):
             database = "main"
 
+        schemas = [sge.convert(database)]
+        if show_temp:
+            schemas.append(sge.convert("temp"))
+
         sql = (
             sg.select("name")
             .from_(F.pragma_table_list())
             .where(
-                C.schema.eq(sge.convert(database)),
+                C.schema.isin(*schemas),
                 C.type.isin(sge.convert("table"), sge.convert("view")),
-                ~(
-                    C.name.isin(
-                        sge.convert("sqlite_schema"),
-                        sge.convert("sqlite_master"),
-                        sge.convert("sqlite_temp_schema"),
-                        sge.convert("sqlite_temp_master"),
-                    )
+                ~C.name.isin(
+                    sge.convert("sqlite_schema"),
+                    sge.convert("sqlite_master"),
+                    sge.convert("sqlite_temp_schema"),
+                    sge.convert("sqlite_temp_master"),
                 ),
             )
             .sql(self.name)