Store search: Ability to control what stores are searched.

src/calibre/gui2/store/search.py

@@ -11,7 +11,8 @@ from threading import Event, Thread
 from Queue import Queue
 
 from PyQt4.Qt import Qt, QAbstractItemModel, QDialog, QTimer, QVariant, \
-    QModelIndex, QPixmap, QSize
+    QModelIndex, QPixmap, QSize, QCheckBox, QVBoxLayout, QHBoxLayout, \
+    QPushButton 
 
 from calibre import browser
 from calibre.customize.ui import store_plugins
@@ -41,12 +42,30 @@ class SearchDialog(QDialog, Ui_Dialog):
         self.model = Matches()
         self.results_view.setModel(self.model)
 
+        stores_group_layout = QVBoxLayout()
+        self.stores_group.setLayout(stores_group_layout)
         for x in store_plugins():
             self.store_plugins[x.name] = x
+            cbox = QCheckBox(x.name)
+            cbox.setChecked(True)
+            stores_group_layout.addWidget(cbox)
+            setattr(self, 'store_check_' + x.name, cbox)
+        
+        store_button_layout = QHBoxLayout()
+        stores_group_layout.addLayout(store_button_layout)
+        self.select_all_stores = QPushButton(_('All'))
+        self.select_invert_stores = QPushButton(_('Invert'))
+        self.select_none_stores = QPushButton(_('None'))
+        store_button_layout.addWidget(self.select_all_stores)
+        store_button_layout.addWidget(self.select_invert_stores)
+        store_button_layout.addWidget(self.select_none_stores)
 
         self.search.clicked.connect(self.do_search)
         self.checker.timeout.connect(self.get_results)
         self.results_view.activated.connect(self.open_store)
+        self.select_all_stores.clicked.connect(self.stores_select_all)
+        self.select_invert_stores.clicked.connect(self.stores_select_invert)
+        self.select_none_stores.clicked.connect(self.stores_select_none)
         
         self.resize_columns()
         
@@ -80,6 +99,7 @@ class SearchDialog(QDialog, Ui_Dialog):
             return
         
         for n in self.store_plugins:
+            if getattr(self, 'store_check_' + n).isChecked():
                 t = SearchThread(query, (n, self.store_plugins[n]), self.results, self.abort, self.TIMEOUT)
                 self.running_threads.append(t)
                 t.start()
@@ -115,6 +135,25 @@ class SearchDialog(QDialog, Ui_Dialog):
         result = self.results_view.model().get_result(index)
         self.store_plugins[result.store].open(self.gui, self, result.detail_item)
 
+    def get_store_checks(self):
+        checks = []
+        for x in self.store_plugins:
+            check = getattr(self, 'store_check_' + x, None)
+            if check:
+                checks.append(check)
+        return checks
+
+    def stores_select_all(self):
+        for check in self.get_store_checks():
+            check.setChecked(True)
+    
+    def stores_select_invert(self):
+        for check in self.get_store_checks():
+            check.setChecked(not check.isChecked())
+    
+    def stores_select_none(self):
+        for check in self.get_store_checks():
+            check.setChecked(False)
 
 class SearchThread(Thread):
     

src/calibre/gui2/store/search.ui

@@ -6,8 +6,8 @@
    <rect>
     <x>0</x>
     <y>0</y>
-    <width>616</width>
-    <height>545</height>
+    <width>937</width>
+    <height>669</height>
    </rect>
   </property>
   <property name="windowTitle">
@@ -16,19 +16,45 @@
   <property name="sizeGripEnabled">
    <bool>true</bool>
   </property>
-  <layout class="QGridLayout" name="gridLayout">
-   <item row="0" column="0">
+  <layout class="QVBoxLayout" name="verticalLayout">
+   <item>
+    <layout class="QHBoxLayout" name="horizontalLayout_2">
+     <item>
+      <widget class="QLabel" name="label">
+       <property name="text">
+        <string>Query:</string>
+       </property>
+      </widget>
+     </item>
+     <item>
       <widget class="QLineEdit" name="search_edit"/>
      </item>
-   <item row="0" column="1">
+     <item>
       <widget class="QPushButton" name="search">
        <property name="text">
         <string>Search</string>
        </property>
       </widget>
      </item>
-   <item row="1" column="0" colspan="2">
+    </layout>
+   </item>
+   <item>
+    <widget class="QSplitter" name="splitter">
+     <property name="orientation">
+      <enum>Qt::Horizontal</enum>
+     </property>
+     <widget class="QGroupBox" name="stores_group">
+      <property name="title">
+       <string>Stores</string>
+      </property>
+     </widget>
      <widget class="QTreeView" name="results_view">
+      <property name="sizePolicy">
+       <sizepolicy hsizetype="Expanding" vsizetype="Expanding">
+        <horstretch>2</horstretch>
+        <verstretch>0</verstretch>
+       </sizepolicy>
+      </property>
       <property name="alternatingRowColors">
        <bool>true</bool>
       </property>
@@ -54,43 +80,45 @@
        <bool>false</bool>
       </property>
      </widget>
+    </widget>
    </item>
-   <item row="2" column="0" colspan="2">
-    <widget class="QDialogButtonBox" name="buttonBox">
-     <property name="standardButtons">
-      <set>QDialogButtonBox::Close</set>
+   <item>
+    <layout class="QHBoxLayout" name="horizontalLayout">
+     <item>
+      <spacer name="horizontalSpacer">
+       <property name="orientation">
+        <enum>Qt::Horizontal</enum>
+       </property>
+       <property name="sizeHint" stdset="0">
+        <size>
+         <width>40</width>
+         <height>20</height>
+        </size>
+       </property>
+      </spacer>
+     </item>
+     <item>
+      <widget class="QPushButton" name="close">
+       <property name="text">
+        <string>Close</string>
        </property>
       </widget>
      </item>
     </layout>
+   </item>
+  </layout>
  </widget>
  <resources/>
  <connections>
   <connection>
-   <sender>buttonBox</sender>
-   <signal>accepted()</signal>
+   <sender>close</sender>
+   <signal>clicked()</signal>
    <receiver>Dialog</receiver>
    <slot>accept()</slot>
    <hints>
     <hint type="sourcelabel">
-     <x>307</x>
-     <y>524</y>
-    </hint>
-    <hint type="destinationlabel">
-     <x>307</x>
-     <y>272</y>
-    </hint>
-   </hints>
-  </connection>
-  <connection>
-   <sender>buttonBox</sender>
-   <signal>rejected()</signal>
-   <receiver>Dialog</receiver>
-   <slot>accept()</slot>
-   <hints>
-    <hint type="sourcelabel">
-     <x>307</x>
-     <y>524</y>
+     <x>526</x>
+     <y>525</y>
     </hint>
     <hint type="destinationlabel">
      <x>307</x>