Fix completion crash when tables are created during refresh

shgol · shgol · commit d1fdb4cdb9e5 · 2026-05-03T16:02:47.000-04:00
diff --git a/AUTHORS b/AUTHORS
@@ -147,6 +147,7 @@ Contributors:
     * Devadathan M B (devadathanmb)
     * Charalampos Stratakis
     * Laszlo Bimba (bimlas)
+    * Shayan Golshani (shgol)
 
 Creator:
 --------
diff --git a/changelog.rst b/changelog.rst
@@ -13,6 +13,7 @@ Bug fixes:
 ----------
 * Add `VERSION` to built-in function completion so `SELECT VERSION();` is suggested.
 * Hide timezone notice at startup when local and server timezones are the same.
+* Fix completion crash when tables are created during refresh.
 
 4.4.0 (2025-12-24)
 ==================
diff --git a/pgcli/pgcompleter.py b/pgcli/pgcompleter.py
@@ -229,6 +229,8 @@ def extend_columns(self, column_data, kind):
                 has_default=has_default,
                 default=default,
             )
+            metadata.setdefault(schema, {})
+            metadata[schema].setdefault(relname, OrderedDict())
             metadata[schema][relname][colname] = column
             self.all_completions.add(colname)
 

Original file line number	Diff line number	Diff line change
`@@ -229,6 +229,8 @@ def extend_columns(self, column_data, kind):`
`229`	`229`	`has_default=has_default,`
`230`	`230`	`default=default,`
`231`	`231`	`)`
	`232`	`+ metadata.setdefault(schema, {})`
	`233`	`+ metadata[schema].setdefault(relname, OrderedDict())`
`232`	`234`	`metadata[schema][relname][colname] = column`
`233`	`235`	`self.all_completions.add(colname)`
`234`	`236`