From 3855183213dff718e82c6feb8468745c7184de9e Mon Sep 17 00:00:00 2001
From: Damien Elmes <git@ichi2.net>
Date: Thu, 14 Apr 2011 03:52:32 +0900
Subject: [PATCH] find & replace

---
 aqt/browser.py          | 55 ++++++++++++------------
 aqt/editor.py           |  4 +-
 designer/browser.ui     |  3 ++
 designer/findreplace.ui | 94 +++++++++++++++++++++++------------------
 4 files changed, 84 insertions(+), 72 deletions(-)

diff --git a/aqt/browser.py b/aqt/browser.py
index 6a13ff32b..c39cdf2e8 100644
--- a/aqt/browser.py
+++ b/aqt/browser.py
@@ -116,6 +116,7 @@ class DeckModel(QAbstractTableModel):
 
     def beginReset(self):
         self.browser.editor.saveNow()
+        self.browser.editor.setFact(None, hide=False)
         self.browser.mw.progress.start()
         self.saveSelection()
         self.beginResetModel()
@@ -740,6 +741,7 @@ where id in %s""" % ids2str(sf))
             ChangeModel(self, fids)
 
     def cram(self):
+        return showInfo("not yet implemented")
         self.close()
         self.mw.onCram(self.selectedCards())
 
@@ -822,6 +824,7 @@ where id in %s""" % ids2str(sf))
     ######################################################################
 
     def reschedule(self):
+        return showInfo("not yet implemented")
         n = _("Reschedule")
         d = QDialog(self)
         frm = aqt.forms.reschedule.Ui_Dialog()
@@ -911,52 +914,46 @@ where id in %s""" % ids2str(sf))
         sf = self.selectedFacts()
         if not sf:
             return
-        mods = self.deck.db.column0("""
-select distinct modelId from facts
-where id in %s""" % ids2str(sf))
-        if not len(mods) == 1:
-            ui.utils.showInfo(
-                _("Can only operate on one model at a time."),
-                parent=self)
-            return
+        import anki.find
+        fields = sorted(anki.find.fieldNames(self.deck, downcase=False))
         d = QDialog(self)
         frm = aqt.forms.findreplace.Ui_Dialog()
         frm.setupUi(d)
-        fields = sorted(self.card.fact.model.fieldModels, key=attrgetter("name"))
-        frm.field.addItems(QStringList(
-            [_("All Fields")] + [f.name for f in fields]))
+        frm.field.addItems(QStringList([_("All Fields")] + fields))
         self.connect(frm.buttonBox, SIGNAL("helpRequested()"),
                      self.onFindReplaceHelp)
         if not d.exec_():
             return
-        n = _("Find and Replace")
-        self.deck.startProgress(2)
-        self.deck.updateProgress(_("Replacing..."))
-        self.deck.setUndoStart(n)
-        self.deck.updateProgress()
-        changed = None
+        if frm.field.currentIndex() == 0:
+            field = None
+        else:
+            field = fields[frm.field.currentIndex()-1]
+        self.mw.checkpoint(_("Find and Replace"))
+        self.mw.progress.start()
+        self.model.beginReset()
         try:
-            if frm.field.currentIndex() == 0:
-                field = None
-            else:
-                field = fields[frm.field.currentIndex()-1].id
             changed = self.deck.findReplace(sf,
                                             unicode(frm.find.text()),
                                             unicode(frm.replace.text()),
                                             frm.re.isChecked(),
-                                            field)
+                                            field,
+                                            frm.ignoreCase.isChecked())
         except sre_constants.error:
             ui.utils.showInfo(_("Invalid regular expression."),
                               parent=self)
-        self.deck.setUndoEnd(n)
-        self.deck.finishProgress()
-        self.mw.reset()
-        self.onSearch()
-        if changed is not None:
-            ui.utils.showInfo(ngettext("%(a)d of %(b)d fact updated", "%(a)d of %(b)d facts updated", len(sf)) % {
+            return
+        else:
+            self.onSearch()
+            self.resetDeck()
+        finally:
+            self.model.endReset()
+            self.mw.progress.finish()
+        showInfo(ngettext(
+            "%(a)d of %(b)d fact updated",
+            "%(a)d of %(b)d facts updated", len(sf)) % {
                 'a': changed,
                 'b': len(sf),
-                }, parent=self)
+            })
 
     def onFindReplaceHelp(self):
         aqt.openHelp("Browser#FindReplace")
diff --git a/aqt/editor.py b/aqt/editor.py
index 459232326..edf8a4180 100644
--- a/aqt/editor.py
+++ b/aqt/editor.py
@@ -350,7 +350,7 @@ class Editor(object):
         if self.fact:
             self.loadFact()
 
-    def setFact(self, fact):
+    def setFact(self, fact, hide=True):
         "Make FACT the current fact."
         self.fact = fact
         # change timer
@@ -359,7 +359,7 @@ class Editor(object):
                              loadCB=self._loadFinished)
             self.updateTagsAndGroup()
             self.updateKeyboard()
-        else:
+        elif hide:
             self.widget.hide()
 
     def loadFact(self, field=0):
diff --git a/designer/browser.ui b/designer/browser.ui
index 5213ed4dc..962cc4ba7 100644
--- a/designer/browser.ui
+++ b/designer/browser.ui
@@ -457,6 +457,9 @@
    <property name="text">
     <string>Find and Re&amp;place...</string>
    </property>
+   <property name="shortcut">
+    <string>Ctrl+Alt+F</string>
+   </property>
   </action>
   <action name="actionCram">
    <property name="icon">
diff --git a/designer/findreplace.ui b/designer/findreplace.ui
index d27b9d7cf..4b94cf51f 100644
--- a/designer/findreplace.ui
+++ b/designer/findreplace.ui
@@ -1,65 +1,76 @@
-<ui version="4.0" >
+<?xml version="1.0" encoding="UTF-8"?>
+<ui version="4.0">
  <class>Dialog</class>
- <widget class="QDialog" name="Dialog" >
-  <property name="geometry" >
+ <widget class="QDialog" name="Dialog">
+  <property name="geometry">
    <rect>
     <x>0</x>
     <y>0</y>
-    <width>322</width>
-    <height>159</height>
+    <width>367</width>
+    <height>209</height>
    </rect>
   </property>
-  <property name="windowTitle" >
+  <property name="windowTitle">
    <string>Find and Replace</string>
   </property>
-  <layout class="QVBoxLayout" name="verticalLayout" >
+  <layout class="QVBoxLayout" name="verticalLayout">
    <item>
-    <layout class="QGridLayout" name="gridLayout" >
-     <item row="0" column="0" >
-      <widget class="QLabel" name="label" >
-       <property name="text" >
-        <string>&lt;b>Find&lt;/b>:</string>
+    <layout class="QGridLayout" name="gridLayout">
+     <item row="0" column="0">
+      <widget class="QLabel" name="label">
+       <property name="text">
+        <string>&lt;b&gt;Find&lt;/b&gt;:</string>
        </property>
       </widget>
      </item>
-     <item row="0" column="1" >
-      <widget class="QLineEdit" name="find" />
+     <item row="0" column="1">
+      <widget class="QLineEdit" name="find"/>
      </item>
-     <item row="2" column="0" >
-      <widget class="QLabel" name="label_2" >
-       <property name="text" >
-        <string>&lt;b>Replace With&lt;/b>:</string>
+     <item row="1" column="0">
+      <widget class="QLabel" name="label_2">
+       <property name="text">
+        <string>&lt;b&gt;Replace With&lt;/b&gt;:</string>
        </property>
       </widget>
      </item>
-     <item row="2" column="1" >
-      <widget class="QLineEdit" name="replace" />
+     <item row="1" column="1">
+      <widget class="QLineEdit" name="replace"/>
      </item>
-     <item row="3" column="1" >
-      <widget class="QCheckBox" name="re" >
-       <property name="text" >
+     <item row="2" column="0">
+      <widget class="QLabel" name="label_3">
+       <property name="text">
+        <string>&lt;b&gt;In&lt;/b&gt;:</string>
+       </property>
+      </widget>
+     </item>
+     <item row="2" column="1">
+      <widget class="QComboBox" name="field"/>
+     </item>
+     <item row="4" column="1">
+      <widget class="QCheckBox" name="re">
+       <property name="text">
         <string>Treat input as regular expression</string>
        </property>
       </widget>
      </item>
-     <item row="1" column="0" >
-      <widget class="QLabel" name="label_3" >
-       <property name="text" >
-        <string>&lt;b>In&lt;/b>:</string>
+     <item row="3" column="1">
+      <widget class="QCheckBox" name="ignoreCase">
+       <property name="text">
+        <string>Ignore case</string>
+       </property>
+       <property name="checked">
+        <bool>true</bool>
        </property>
       </widget>
      </item>
-     <item row="1" column="1" >
-      <widget class="QComboBox" name="field" />
-     </item>
     </layout>
    </item>
    <item>
-    <spacer name="verticalSpacer" >
-     <property name="orientation" >
+    <spacer name="verticalSpacer">
+     <property name="orientation">
       <enum>Qt::Vertical</enum>
      </property>
-     <property name="sizeHint" stdset="0" >
+     <property name="sizeHint" stdset="0">
       <size>
        <width>20</width>
        <height>40</height>
@@ -68,11 +79,11 @@
     </spacer>
    </item>
    <item>
-    <widget class="QDialogButtonBox" name="buttonBox" >
-     <property name="orientation" >
+    <widget class="QDialogButtonBox" name="buttonBox">
+     <property name="orientation">
       <enum>Qt::Horizontal</enum>
      </property>
-     <property name="standardButtons" >
+     <property name="standardButtons">
       <set>QDialogButtonBox::Cancel|QDialogButtonBox::Help|QDialogButtonBox::Ok</set>
      </property>
     </widget>
@@ -81,8 +92,9 @@
  </widget>
  <tabstops>
   <tabstop>find</tabstop>
-  <tabstop>field</tabstop>
   <tabstop>replace</tabstop>
+  <tabstop>field</tabstop>
+  <tabstop>ignoreCase</tabstop>
   <tabstop>re</tabstop>
   <tabstop>buttonBox</tabstop>
  </tabstops>
@@ -94,11 +106,11 @@
    <receiver>Dialog</receiver>
    <slot>accept()</slot>
    <hints>
-    <hint type="sourcelabel" >
+    <hint type="sourcelabel">
      <x>256</x>
      <y>154</y>
     </hint>
-    <hint type="destinationlabel" >
+    <hint type="destinationlabel">
      <x>157</x>
      <y>274</y>
     </hint>
@@ -110,11 +122,11 @@
    <receiver>Dialog</receiver>
    <slot>reject()</slot>
    <hints>
-    <hint type="sourcelabel" >
+    <hint type="sourcelabel">
      <x>290</x>
      <y>154</y>
     </hint>
-    <hint type="destinationlabel" >
+    <hint type="destinationlabel">
      <x>286</x>
      <y>274</y>
     </hint>