Merge pull request #2129 from isosphere/develop

[postgresql] quote schema and table name
saulpw · Nov 20, 2023 · 81af233 · 81af233
2 parents 0e1afd8 + 216bb5a
commit 81af233
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/visidata/loaders/postgres.py b/visidata/loaders/postgres.py
@@ -123,9 +123,9 @@ class PgTable(Sheet):
     @asyncthread
     def reload(self):
         if self.options.postgres_schema:
-            source = f"{self.options.postgres_schema}.{self.source}"
+            source = f'"{self.options.postgres_schema}"."{self.source}"'
         else:
-            source = self.source
+            source = f'"{self.source}"'
         with self.sql.cur(f"SELECT * FROM {source}") as cur:
             self.rows = []
             r = cur.fetchone()