From 1a4768a539bfdf65b2f5a6d68a133730683f9261 Mon Sep 17 00:00:00 2001
From: Charles Haley <>
Date: Sun, 22 May 2011 18:32:03 +0100
Subject: [PATCH] Make coloring work across change_libraries

---
 src/calibre/gui2/init.py           | 1 -
 src/calibre/gui2/library/models.py | 2 ++
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/calibre/gui2/init.py b/src/calibre/gui2/init.py
index 079e1814c3..a75ff01b21 100644
--- a/src/calibre/gui2/init.py
+++ b/src/calibre/gui2/init.py
@@ -65,7 +65,6 @@ class LibraryViewMixin(object): # {{{
 
         self.build_context_menus()
         self.library_view.model().set_highlight_only(config['highlight_search_matches'])
-        self.library_view.model().set_color_templates()
 
     def build_context_menus(self):
         lm = QMenu(self)
diff --git a/src/calibre/gui2/library/models.py b/src/calibre/gui2/library/models.py
index 83bf5868ba..a3e7438908 100644
--- a/src/calibre/gui2/library/models.py
+++ b/src/calibre/gui2/library/models.py
@@ -157,6 +157,7 @@ class BooksModel(QAbstractTableModel): # {{{
         self.database_changed.emit(db)
         self.stop_metadata_backup()
         self.start_metadata_backup()
+        self.set_color_templates()
 
     def start_metadata_backup(self):
         self.metadata_backup = MetadataBackup(self.db)
@@ -535,6 +536,7 @@ class BooksModel(QAbstractTableModel): # {{{
         return img
 
     def set_color_templates(self):
+        print 'here'
         self.column_color_map = {}
         for i in range(1,self.db.column_color_count+1):
             name = self.db.prefs.get('column_color_name_'+str(i))