335 lines
12 KiB
Python
335 lines
12 KiB
Python
|
# find.py - Find and Replace
|
||
|
|
import win32con, win32api
|
||
|
|
import win32ui
|
||
|
|
from pywin.mfc import dialog
|
||
|
|
import afxres
|
||
|
|
from pywin.framework import scriptutils
|
||
|
|
|
||
|
|
FOUND_NOTHING=0
|
||
|
|
FOUND_NORMAL=1
|
||
|
|
FOUND_LOOPED_BACK=2
|
||
|
|
FOUND_NEXT_FILE=3
|
||
|
|
|
||
|
|
class SearchParams:
|
||
|
|
def __init__(self, other=None):
|
||
|
|
if other is None:
|
||
|
|
self.__dict__['findText'] = ""
|
||
|
|
self.__dict__['replaceText'] = ""
|
||
|
|
self.__dict__['matchCase'] = 0
|
||
|
|
self.__dict__['matchWords'] = 0
|
||
|
|
self.__dict__['acrossFiles'] = 0
|
||
|
|
self.__dict__['remember'] = 1
|
||
|
|
self.__dict__['sel'] = (-1,-1)
|
||
|
|
self.__dict__['keepDialogOpen']=0
|
||
|
|
else:
|
||
|
|
self.__dict__.update(other.__dict__)
|
||
|
|
# Helper so we cant misspell attributes :-)
|
||
|
|
def __setattr__(self, attr, val):
|
||
|
|
if not hasattr(self, attr):
|
||
|
|
raise AttributeError(attr)
|
||
|
|
self.__dict__[attr]=val
|
||
|
|
|
||
|
|
curDialog = None
|
||
|
|
lastSearch = defaultSearch = SearchParams()
|
||
|
|
searchHistory = []
|
||
|
|
|
||
|
|
def ShowFindDialog():
|
||
|
|
_ShowDialog(FindDialog)
|
||
|
|
|
||
|
|
def ShowReplaceDialog():
|
||
|
|
_ShowDialog(ReplaceDialog)
|
||
|
|
|
||
|
|
def _ShowDialog(dlgClass):
|
||
|
|
global curDialog
|
||
|
|
if curDialog is not None:
|
||
|
|
if curDialog.__class__ != dlgClass:
|
||
|
|
curDialog.DestroyWindow()
|
||
|
|
curDialog = None
|
||
|
|
else:
|
||
|
|
curDialog.SetFocus()
|
||
|
|
if curDialog is None:
|
||
|
|
curDialog = dlgClass()
|
||
|
|
curDialog.CreateWindow()
|
||
|
|
|
||
|
|
def FindNext():
|
||
|
|
params = SearchParams(lastSearch)
|
||
|
|
params.sel = (-1,-1)
|
||
|
|
if not params.findText:
|
||
|
|
ShowFindDialog()
|
||
|
|
else:
|
||
|
|
return _FindIt(None, params)
|
||
|
|
|
||
|
|
def _GetControl(control=None):
|
||
|
|
if control is None:
|
||
|
|
control = scriptutils.GetActiveEditControl()
|
||
|
|
return control
|
||
|
|
|
||
|
|
def _FindIt(control, searchParams):
|
||
|
|
global lastSearch, defaultSearch
|
||
|
|
control = _GetControl(control)
|
||
|
|
if control is None: return FOUND_NOTHING
|
||
|
|
|
||
|
|
# Move to the next char, so we find the next one.
|
||
|
|
flags = 0
|
||
|
|
if searchParams.matchWords: flags = flags | win32con.FR_WHOLEWORD
|
||
|
|
if searchParams.matchCase: flags = flags | win32con.FR_MATCHCASE
|
||
|
|
if searchParams.sel == (-1,-1):
|
||
|
|
sel = control.GetSel()
|
||
|
|
# If the position is the same as we found last time,
|
||
|
|
# then we assume it is a "FindNext"
|
||
|
|
if sel==lastSearch.sel:
|
||
|
|
sel = sel[0]+1, sel[0]+1
|
||
|
|
else:
|
||
|
|
sel = searchParams.sel
|
||
|
|
|
||
|
|
if sel[0]==sel[1]: sel=sel[0], control.GetTextLength()
|
||
|
|
|
||
|
|
rc = FOUND_NOTHING
|
||
|
|
# (Old edit control will fail here!)
|
||
|
|
posFind, foundSel = control.FindText(flags, sel, searchParams.findText)
|
||
|
|
lastSearch = SearchParams(searchParams)
|
||
|
|
if posFind >= 0:
|
||
|
|
rc = FOUND_NORMAL
|
||
|
|
lineno = control.LineFromChar(posFind)
|
||
|
|
control.SCIEnsureVisible(lineno)
|
||
|
|
control.SetSel(foundSel)
|
||
|
|
control.SetFocus()
|
||
|
|
win32ui.SetStatusText(win32ui.LoadString(afxres.AFX_IDS_IDLEMESSAGE))
|
||
|
|
if rc == FOUND_NOTHING and lastSearch.acrossFiles:
|
||
|
|
# Loop around all documents. First find this document.
|
||
|
|
try:
|
||
|
|
try:
|
||
|
|
doc = control.GetDocument()
|
||
|
|
except AttributeError:
|
||
|
|
try:
|
||
|
|
doc = control.GetParent().GetDocument()
|
||
|
|
except AttributeError:
|
||
|
|
print("Cant find a document for the control!")
|
||
|
|
doc = None
|
||
|
|
if doc is not None:
|
||
|
|
template = doc.GetDocTemplate()
|
||
|
|
alldocs = template.GetDocumentList()
|
||
|
|
mypos = lookpos = alldocs.index(doc)
|
||
|
|
while 1:
|
||
|
|
lookpos = (lookpos+1) % len(alldocs)
|
||
|
|
if lookpos == mypos:
|
||
|
|
break
|
||
|
|
view = alldocs[lookpos].GetFirstView()
|
||
|
|
posFind, foundSel = view.FindText(flags, (0, view.GetTextLength()), searchParams.findText)
|
||
|
|
if posFind >= 0:
|
||
|
|
nChars = foundSel[1]-foundSel[0]
|
||
|
|
lineNo = view.LineFromChar(posFind) # zero based.
|
||
|
|
lineStart = view.LineIndex(lineNo)
|
||
|
|
colNo = posFind - lineStart # zero based.
|
||
|
|
scriptutils.JumpToDocument(alldocs[lookpos].GetPathName(), lineNo+1, colNo+1, nChars)
|
||
|
|
rc = FOUND_NEXT_FILE
|
||
|
|
break
|
||
|
|
except win32ui.error:
|
||
|
|
pass
|
||
|
|
if rc == FOUND_NOTHING:
|
||
|
|
# Loop around this control - attempt to find from the start of the control.
|
||
|
|
posFind, foundSel = control.FindText(flags, (0, sel[0]-1), searchParams.findText)
|
||
|
|
if posFind >= 0:
|
||
|
|
control.SCIEnsureVisible(control.LineFromChar(foundSel[0]))
|
||
|
|
control.SetSel(foundSel)
|
||
|
|
control.SetFocus()
|
||
|
|
win32ui.SetStatusText("Not found! Searching from the top of the file.")
|
||
|
|
rc = FOUND_LOOPED_BACK
|
||
|
|
else:
|
||
|
|
lastSearch.sel=-1,-1
|
||
|
|
win32ui.SetStatusText("Can not find '%s'" % searchParams.findText )
|
||
|
|
|
||
|
|
if rc != FOUND_NOTHING:
|
||
|
|
lastSearch.sel = foundSel
|
||
|
|
|
||
|
|
if lastSearch.remember:
|
||
|
|
defaultSearch = lastSearch
|
||
|
|
|
||
|
|
# track search history
|
||
|
|
try:
|
||
|
|
ix = searchHistory.index(searchParams.findText)
|
||
|
|
except ValueError:
|
||
|
|
if len(searchHistory) > 50:
|
||
|
|
searchHistory[50:] = []
|
||
|
|
else:
|
||
|
|
del searchHistory[ix]
|
||
|
|
searchHistory.insert(0, searchParams.findText)
|
||
|
|
|
||
|
|
return rc
|
||
|
|
|
||
|
|
def _ReplaceIt(control):
|
||
|
|
control = _GetControl(control)
|
||
|
|
statusText = "Can not find '%s'." % lastSearch.findText
|
||
|
|
rc = FOUND_NOTHING
|
||
|
|
if control is not None and lastSearch.sel != (-1,-1):
|
||
|
|
control.ReplaceSel(lastSearch.replaceText)
|
||
|
|
rc = FindNext()
|
||
|
|
if rc !=FOUND_NOTHING:
|
||
|
|
statusText = win32ui.LoadString(afxres.AFX_IDS_IDLEMESSAGE)
|
||
|
|
win32ui.SetStatusText(statusText)
|
||
|
|
return rc
|
||
|
|
|
||
|
|
class FindReplaceDialog(dialog.Dialog):
|
||
|
|
def __init__(self):
|
||
|
|
dialog.Dialog.__init__(self,self._GetDialogTemplate())
|
||
|
|
self.HookCommand(self.OnFindNext, 109)
|
||
|
|
|
||
|
|
def OnInitDialog(self):
|
||
|
|
self.editFindText = self.GetDlgItem(102)
|
||
|
|
self.butMatchWords = self.GetDlgItem(105)
|
||
|
|
self.butMatchCase = self.GetDlgItem(107)
|
||
|
|
self.butKeepDialogOpen = self.GetDlgItem(115)
|
||
|
|
self.butAcrossFiles = self.GetDlgItem(116)
|
||
|
|
self.butRemember = self.GetDlgItem(117)
|
||
|
|
|
||
|
|
self.editFindText.SetWindowText(defaultSearch.findText)
|
||
|
|
control = _GetControl()
|
||
|
|
# _GetControl only gets normal MDI windows; if the interactive
|
||
|
|
# window is docked and no document open, we get None.
|
||
|
|
if control:
|
||
|
|
# If we have a selection, default to that.
|
||
|
|
sel = control.GetSelText()
|
||
|
|
if (len(sel) != 0):
|
||
|
|
self.editFindText.SetWindowText(sel)
|
||
|
|
if (defaultSearch.remember):
|
||
|
|
defaultSearch.findText = sel
|
||
|
|
for hist in searchHistory:
|
||
|
|
self.editFindText.AddString(hist)
|
||
|
|
|
||
|
|
if hasattr(self.editFindText, 'SetEditSel'):
|
||
|
|
self.editFindText.SetEditSel(0, -2)
|
||
|
|
else:
|
||
|
|
self.editFindText.SetSel(0, -2)
|
||
|
|
self.editFindText.SetFocus()
|
||
|
|
self.butMatchWords.SetCheck(defaultSearch.matchWords)
|
||
|
|
self.butMatchCase.SetCheck(defaultSearch.matchCase)
|
||
|
|
self.butKeepDialogOpen.SetCheck(defaultSearch.keepDialogOpen)
|
||
|
|
self.butAcrossFiles.SetCheck(defaultSearch.acrossFiles)
|
||
|
|
self.butRemember.SetCheck(defaultSearch.remember)
|
||
|
|
return dialog.Dialog.OnInitDialog(self)
|
||
|
|
|
||
|
|
def OnDestroy(self, msg):
|
||
|
|
global curDialog
|
||
|
|
curDialog = None
|
||
|
|
return dialog.Dialog.OnDestroy(self, msg)
|
||
|
|
|
||
|
|
def DoFindNext(self):
|
||
|
|
params = SearchParams()
|
||
|
|
params.findText = self.editFindText.GetWindowText()
|
||
|
|
params.matchCase = self.butMatchCase.GetCheck()
|
||
|
|
params.matchWords = self.butMatchWords.GetCheck()
|
||
|
|
params.acrossFiles = self.butAcrossFiles.GetCheck()
|
||
|
|
params.remember = self.butRemember.GetCheck()
|
||
|
|
return _FindIt(None, params)
|
||
|
|
|
||
|
|
def OnFindNext(self, id, code):
|
||
|
|
if not self.editFindText.GetWindowText():
|
||
|
|
win32api.MessageBeep()
|
||
|
|
return
|
||
|
|
if self.DoFindNext() != FOUND_NOTHING:
|
||
|
|
if not self.butKeepDialogOpen.GetCheck():
|
||
|
|
self.DestroyWindow()
|
||
|
|
|
||
|
|
class FindDialog(FindReplaceDialog):
|
||
|
|
def _GetDialogTemplate(self):
|
||
|
|
style = win32con.DS_MODALFRAME | win32con.WS_POPUP | win32con.WS_VISIBLE | win32con.WS_CAPTION | win32con.WS_SYSMENU | win32con.DS_SETFONT
|
||
|
|
visible = win32con.WS_CHILD | win32con.WS_VISIBLE
|
||
|
|
dt = [
|
||
|
|
["Find", (0, 2, 240, 75), style, None, (8, "MS Sans Serif")],
|
||
|
|
["Static", "Fi&nd What:", 101, (5, 8, 40, 10), visible],
|
||
|
|
["ComboBox", "", 102, (50, 7, 120, 120), visible | win32con.WS_BORDER | win32con.WS_TABSTOP |
|
||
|
|
win32con.WS_VSCROLL |win32con.CBS_DROPDOWN |win32con.CBS_AUTOHSCROLL],
|
||
|
|
["Button", "Match &whole word only", 105, (5, 23, 100, 10), visible | win32con.BS_AUTOCHECKBOX | win32con.WS_TABSTOP],
|
||
|
|
["Button", "Match &case", 107, (5, 33, 100, 10), visible | win32con.BS_AUTOCHECKBOX | win32con.WS_TABSTOP],
|
||
|
|
["Button", "Keep &dialog open", 115, (5, 43, 100, 10), visible | win32con.BS_AUTOCHECKBOX | win32con.WS_TABSTOP],
|
||
|
|
["Button", "Across &open files", 116, (5, 52, 100, 10), visible | win32con.BS_AUTOCHECKBOX | win32con.WS_TABSTOP],
|
||
|
|
["Button", "&Remember as default search", 117, (5, 61, 150, 10), visible | win32con.BS_AUTOCHECKBOX | win32con.WS_TABSTOP],
|
||
|
|
["Button", "&Find Next", 109, (185, 5, 50, 14), visible | win32con.BS_DEFPUSHBUTTON | win32con.WS_TABSTOP],
|
||
|
|
["Button", "Cancel", win32con.IDCANCEL, (185, 23, 50, 14), visible | win32con.WS_TABSTOP],
|
||
|
|
]
|
||
|
|
return dt
|
||
|
|
|
||
|
|
class ReplaceDialog(FindReplaceDialog):
|
||
|
|
def _GetDialogTemplate(self):
|
||
|
|
style = win32con.DS_MODALFRAME | win32con.WS_POPUP | win32con.WS_VISIBLE | win32con.WS_CAPTION | win32con.WS_SYSMENU | win32con.DS_SETFONT
|
||
|
|
visible = win32con.WS_CHILD | win32con.WS_VISIBLE
|
||
|
|
dt = [
|
||
|
|
["Replace", (0, 2, 240, 95), style, 0, (8, "MS Sans Serif")],
|
||
|
|
["Static", "Fi&nd What:", 101, (5, 8, 40, 10), visible],
|
||
|
|
["ComboBox", "", 102, (60, 7, 110, 120), visible | win32con.WS_BORDER | win32con.WS_TABSTOP |
|
||
|
|
win32con.WS_VSCROLL |win32con.CBS_DROPDOWN |win32con.CBS_AUTOHSCROLL],
|
||
|
|
["Static", "Re&place with:", 103, (5, 25, 50, 10), visible],
|
||
|
|
["ComboBox", "", 104, (60, 24, 110, 120), visible | win32con.WS_BORDER | win32con.WS_TABSTOP |
|
||
|
|
win32con.WS_VSCROLL |win32con.CBS_DROPDOWN |win32con.CBS_AUTOHSCROLL],
|
||
|
|
["Button", "Match &whole word only", 105, (5, 42, 100, 10), visible | win32con.BS_AUTOCHECKBOX | win32con.WS_TABSTOP],
|
||
|
|
["Button", "Match &case", 107, (5, 52, 100, 10), visible | win32con.BS_AUTOCHECKBOX | win32con.WS_TABSTOP],
|
||
|
|
["Button", "Keep &dialog open", 115, (5, 62, 100, 10), visible | win32con.BS_AUTOCHECKBOX | win32con.WS_TABSTOP],
|
||
|
|
["Button", "Across &open files", 116, (5, 72, 100, 10), visible | win32con.BS_AUTOCHECKBOX | win32con.WS_TABSTOP],
|
||
|
|
["Button", "&Remember as default search", 117, (5, 81, 150, 10), visible | win32con.BS_AUTOCHECKBOX | win32con.WS_TABSTOP],
|
||
|
|
["Button", "&Find Next", 109, (185, 5, 50, 14), visible | win32con.BS_DEFPUSHBUTTON | win32con.WS_TABSTOP],
|
||
|
|
["Button", "&Replace", 110, (185, 23, 50, 14), visible | win32con.WS_TABSTOP],
|
||
|
|
["Button", "Replace &All", 111, (185, 41, 50, 14), visible | win32con.WS_TABSTOP],
|
||
|
|
["Button", "Cancel", win32con.IDCANCEL, (185, 59, 50, 14), visible | win32con.WS_TABSTOP],
|
||
|
|
|
||
|
|
|
||
|
|
]
|
||
|
|
return dt
|
||
|
|
|
||
|
|
def OnInitDialog(self):
|
||
|
|
rc = FindReplaceDialog.OnInitDialog(self)
|
||
|
|
self.HookCommand(self.OnReplace, 110)
|
||
|
|
self.HookCommand(self.OnReplaceAll, 111)
|
||
|
|
self.HookMessage(self.OnActivate, win32con.WM_ACTIVATE)
|
||
|
|
self.editReplaceText = self.GetDlgItem(104)
|
||
|
|
self.editReplaceText.SetWindowText(lastSearch.replaceText)
|
||
|
|
if hasattr(self.editReplaceText, 'SetEditSel'):
|
||
|
|
self.editReplaceText.SetEditSel(0, -2)
|
||
|
|
else:
|
||
|
|
self.editReplaceText.SetSel(0, -2)
|
||
|
|
self.butReplace = self.GetDlgItem(110)
|
||
|
|
self.butReplaceAll = self.GetDlgItem(111)
|
||
|
|
self.CheckButtonStates()
|
||
|
|
return rc
|
||
|
|
|
||
|
|
def CheckButtonStates(self):
|
||
|
|
# We can do a "Replace" or "Replace All" if the current selection
|
||
|
|
# is the same as the search text.
|
||
|
|
ft = self.editFindText.GetWindowText()
|
||
|
|
control = _GetControl()
|
||
|
|
# bCanReplace = len(ft)>0 and control.GetSelText() == ft
|
||
|
|
bCanReplace = control is not None and lastSearch.sel == control.GetSel()
|
||
|
|
self.butReplace.EnableWindow(bCanReplace)
|
||
|
|
# self.butReplaceAll.EnableWindow(bCanReplace)
|
||
|
|
|
||
|
|
def OnActivate(self, msg):
|
||
|
|
wparam = msg[2]
|
||
|
|
fActive = win32api.LOWORD(wparam)
|
||
|
|
if fActive != win32con.WA_INACTIVE:
|
||
|
|
self.CheckButtonStates()
|
||
|
|
|
||
|
|
def OnFindNext(self, id, code):
|
||
|
|
self.DoFindNext()
|
||
|
|
self.CheckButtonStates()
|
||
|
|
|
||
|
|
def OnReplace(self, id, code):
|
||
|
|
lastSearch.replaceText = self.editReplaceText.GetWindowText()
|
||
|
|
_ReplaceIt(None)
|
||
|
|
|
||
|
|
def OnReplaceAll(self, id, code):
|
||
|
|
control = _GetControl(None)
|
||
|
|
if control is not None:
|
||
|
|
control.SetSel(0)
|
||
|
|
num = 0
|
||
|
|
if self.DoFindNext() == FOUND_NORMAL:
|
||
|
|
num = 1
|
||
|
|
lastSearch.replaceText = self.editReplaceText.GetWindowText()
|
||
|
|
while _ReplaceIt(control) == FOUND_NORMAL:
|
||
|
|
num = num + 1
|
||
|
|
|
||
|
|
win32ui.SetStatusText("Replaced %d occurrences" % num)
|
||
|
|
if num > 0 and not self.butKeepDialogOpen.GetCheck():
|
||
|
|
self.DestroyWindow()
|
||
|
|
|
||
|
|
if __name__=='__main__':
|
||
|
|
ShowFindDialog()
|