fix "local variable 'txt' referenced before assignment"

2014-01-13 19:07:34 +09:00 · 2014-01-13 19:07:34 +09:00 · c5df294a28
commit c5df294a28
parent cc98ef3763
1 changed files with 4 additions and 0 deletions
--- a/aqt/browser.py
+++ b/aqt/browser.py
@ -96,10 +96,14 @@ class DataModel(QAbstractTableModel):
            return
        elif role == Qt.DisplayRole and section < len(self.activeCols):
            type = self.columnType(section)
+            txt = None
            for stype, name in self.browser.columns:
                if type == stype:
                    txt = name
                    break
+            # handle case where extension has set an invalid column type
+            if not txt:
+                txt = self.browser.columns[0][1]
            return txt
        else:
            return