From e55598eb261166fc649f2f5fe19059246f472b1c Mon Sep 17 00:00:00 2001
From: Damien Elmes <git@ichi2.net>
Date: Mon, 21 May 2012 12:28:46 +0900
Subject: [PATCH] mod deck when column added

---
 aqt/browser.py | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

diff --git a/aqt/browser.py b/aqt/browser.py
index 255d7c4a5..3d1c689b8 100644
--- a/aqt/browser.py
+++ b/aqt/browser.py
@@ -83,7 +83,7 @@ class DataModel(QAbstractTableModel):
     def headerData(self, section, orientation, role):
         if orientation == Qt.Vertical:
             return
-        elif role == Qt.DisplayRole:
+        elif role == Qt.DisplayRole and section < len(self.activeCols):
             type = self.columnType(section)
             for stype, name in self.browser.columns:
                 if type == stype:
@@ -186,13 +186,7 @@ class DataModel(QAbstractTableModel):
     ######################################################################
 
     def columnType(self, column):
-        try:
-            type = self.activeCols[column]
-        except:
-            # debugging
-            print column, self.activeCols
-            return "noteFld"
-        return type
+        return self.activeCols[column]
 
     def columnData(self, index):
         row = index.row()
@@ -398,6 +392,7 @@ class Browser(QMainWindow):
         saveState(self, "editor")
         saveHeader(self.form.tableView.horizontalHeader(), "editor")
         self.col.conf['activeCols'] = self.model.activeCols
+        self.col.setMod()
         self.hide()
         aqt.dialogs.close("Browser")
         self.teardownHooks()