Fix completion crash when tables are created during refresh

shgol · shgol · commit 1da54360a4dc · 2026-06-26T00:49:46.000-04:00
diff --git a/AUTHORS b/AUTHORS
@@ -148,6 +148,7 @@ Contributors:
     * Charalampos Stratakis
     * Laszlo Bimba (bimlas)
     * Anjanna
+    * Shayan Golshani (shgol)
 
 Creator:
 --------
diff --git a/changelog.rst b/changelog.rst
@@ -20,6 +20,7 @@ Bug fixes:
 * Fix trailing SQL comments preventing query submission and execution.
     * ``SELECT 1; -- note`` now submits correctly in multiline mode
     * ``rstrip(";")`` in ``pgexecute.py`` now handles comments after the semicolon
+* 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`