def __init__(self, title, pos, size):
wx.Frame.__init__(self, None, -1, title, pos, size)
self.current_dupe_file = '' # Current shelve db file.
self.current_scan_path = '' # Root directories for currently loaded dupe scan.
self.scan_dataset = {} # Main data set. In format of key:list
self.value_key_map = {} # Reverse lookup of flie to hash
self.dupe_checker = DupeChecker() # Dupe scanner code.
self.clip_count = 0
self.is_dirty = False # If a new scan is done or a file touched, set to true.
### Start of wxPython UI Code:
self.BuildToolbar()
self.SetMenuBar(self.BuildMenubar())
self.CreateStatusBar()
self.SetStatusText('Welcome to Duper')
self.CreateTree()
self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)
class DuperFrame(wx.Frame):
## Constructor
def __init__(self, title, pos, size):
wx.Frame.__init__(self, None, -1, title, pos, size)
self.current_dupe_file = '' # Current shelve db file.
self.current_scan_path = '' # Root directories for currently loaded dupe scan.
self.scan_dataset = {} # Main data set. In format of key:list
self.value_key_map = {} # Reverse lookup of flie to hash
self.dupe_checker = DupeChecker() # Dupe scanner code.
self.clip_count = 0
self.is_dirty = False # If a new scan is done or a file touched, set to true.
### Start of wxPython UI Code:
self.BuildToolbar()
self.SetMenuBar(self.BuildMenubar())
self.CreateStatusBar()
self.SetStatusText('Welcome to Duper')
self.CreateTree()
self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)
## Dupe Tree Code ##
## Create the primary tree control used to display the dupe data.
def CreateTree(self):
sizer = wx.BoxSizer(wx.HORIZONTAL)
self.tree = wx.TreeCtrl(self, id=-1, pos=wx.DefaultPosition,
size=wx.DefaultSize, style=wx.TR_HAS_BUTTONS | wx.TR_HIDE_ROOT,
validator=wx.DefaultValidator, name="treeCtrl")
sizer.Add(self.tree, 1, wx.EXPAND)
self.SetSizer(sizer)
self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnTItemActivate, self.tree)
self.UpdateTree()
# Called after the data set has changed (after a scan or a db load).
def RebuildTree(self):
self.tree.DeleteAllItems()
self.UpdateTree()
# Called any time that the tree needs to paint it's data.
def UpdateTree(self):
root = self.tree.AddRoot("Duplicate Set")
for key in self.scan_dataset:
key_id = self.tree.AppendItem(root, key)
for item in self.scan_dataset[key]:
self.tree.AppendItem(key_id, item)
self.value_key_map[item] = key
## Toolbar Code ##
def BuildToolbar(self):
toolbar = self.CreateToolBar()
toolbar.SetToolBitmapSize(wx.Size(16, 16))
self.AddTool(toolbar, 'icons/database_add.png', 'New Scan',
'Scan for duplicate sound files.', self.OnNewScan)
self.AddTool(toolbar, 'icons/database_refresh.png', 'Re-Scan',
'Rescan the current data set to capture file system changes.', self.OnRescanDataset)
self.AddTool(toolbar, 'icons/database_link.png', 'Validate',
'Revalidate current file paths (verify file existance).', self.OnRevalidateData)
self.AddTool(toolbar, 'icons/database.png', 'Open Dupes File',
'Open a previously saved dupes scan.', self.OnOpenDupes)
self.AddTool(toolbar, 'icons/disk.png', 'Save Dupes File',
'Save the current dupes scan data.', self.OnSaveDupes)
toolbar.AddSeparator()
self.AddTool(toolbar, 'icons/page_delete.png', 'Delete Selected Clip',
'Delete the currently selected clip.', self.OnDoFileRemove)
self.AddTool(toolbar, 'icons/page_link.png', 'Set Single Clip',
'Designate one clip as master. All others will replaced with links.',
self.OnDoFileAliasReplace)
toolbar.Realize()
def AddTool(self, toolbar, icon, help_, long_help, function):
icon_png = wx.Image(icon, wx.BITMAP_TYPE_PNG).ConvertToBitmap()
tool = toolbar.AddSimpleTool(wx.NewId(), icon_png, help_, long_help)
self.Bind(wx.EVT_TOOL, function, tool)
## Menu Bar Code ##
def BuildMenubar(self):
menuFile = wx.Menu()
self.AddMenuItem(menuFile, '&New Scan...', self.OnNewScan)
self.AddMenuItem(menuFile, '&Save Dupe Set...', self.OnSaveDupes)
self.AddMenuItem(menuFile, 'Save Dupe Set As...', self.OnSaveDupesAs)
self.AddMenuItem(menuFile, '&Open Dupe Set...', self.OnOpenDupes)
menuFile.AppendSeparator()
self.AddMenuItem(menuFile, 'Quit', self.OnQuit, wx.ID_EXIT)
menuEdit = wx.Menu()
self.AddMenuItem(menuEdit, "Cut", self.OnCut, wx.ID_CUT)
self.AddMenuItem(menuEdit, "&Copy", self.OnCopy, wx.ID_COPY)
self.AddMenuItem(menuEdit, "Paste", self.OnPaste, wx.ID_PASTE)
menuEdit.AppendSeparator()
self.AddMenuItem(menuEdit, "Delete Selected Clip", self.OnDoFileRemove)
self.AddMenuItem(menuEdit, "Set Single Clip", self.OnDoFileAliasReplace)
menuEdit.AppendSeparator()
self.AddMenuItem(menuEdit, "&Preferences", self.OnPrefs, wx.ID_PREFERENCES)
menuHelp = wx.Menu()
self.AddMenuItem(menuHelp, '&About Duper', self.OnAbout, wx.ID_ABOUT)
self.AddMenuItem(menuHelp, 'Help', self.OnHelp)
menuBar = wx.MenuBar()
menuBar.Append(menuFile, '&File')
menuBar.Append(menuEdit, '&Edit')
menuBar.Append(menuHelp, '&Help')
return menuBar
def AddMenuItem(self, menu, text, function, item_id=None):
if not item_id:
item_id = wx.NewId()
menu.Append(item_id, text)
self.Bind(wx.EVT_MENU, function, id=item_id)
## About Box Uses AboutHTML defined below.
def OnAbout(self, event):
dlg = wx.Dialog(self, -1, 'About Duper', size=(440, 400))
html = AboutHTML(dlg)
html.SetPage(html.GetAboutTxt())
button = wx.Button(dlg, wx.ID_OK, "Ok")
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(html, 1, wx.EXPAND | wx.ALL, 5)
sizer.Add(button, 0, wx.ALIGN_CENTER | wx.ALL, 5)
dlg.SetSizer(sizer)
dlg.Layout()
dlg.ShowModal()
def OnHelp(self, event):
dlg = wx.Dialog(self, -1, 'About Duper', size=(440, 400))
html = wx.html.HtmlWindow(dlg)
html.SetPage(self.GetHelpHTML())
button = wx.Button(dlg, wx.ID_OK, "Ok")
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(html, 1, wx.EXPAND | wx.ALL, 5)
sizer.Add(button, 0, wx.ALIGN_CENTER | wx.ALL, 5)
dlg.SetSizer(sizer)
dlg.Layout()
dlg.ShowModal()
## Event Handlers ##
def OnQuit(self, event):
self.Close()
def OnCloseWindow(self, event):
if self.is_dirty:
dlg = wx.MessageDialog(None,
"The current dupe scan data has not been saved. Do you want to save this data?",
'Unsaved Data', wx.YES_NO | wx.ICON_EXCLAMATION)
retCode = dlg.ShowModal()
if (retCode == wx.ID_YES):
self.DoDataSave(self.current_dupe_file)
self.scan_dataset = None
self.value_key_map = None
self.dupe_checker = None
self.Destroy()
def OnCut(self, event):
pass
def OnCopy(self, event):
pass
def OnPaste(self, event):
pass
def OnPrefs(self, event):
pass
def OnNewScan(self, event):
scan_path = self.DoGetScanPath()
if (scan_path):
self.DoDupeScan(scan_path)
self.current_scan_path = scan_path
def OnRevalidateData(self, event):
invalid_paths = []
for key in self.scan_dataset:
for path in self.scan_dataset[key]:
if not os.path.exists(path):
invalid_paths.append(path)
dlg = None
if not invalid_paths:
dlg = wx.MessageDialog(None, 'The current data files are valid.',
'Validation Complete', wx.OK | wx.ICON_INFORMATION)
else:
dlg = wx.MessageDialog(None, "The following files do not seem to exist any longer."
"Validatioin Complete", wx.YES_NO | wx.ICON_INFORMATION)
retCode = dlg.ShowModal()
if invalid_paths and retCode:
pass
# Display invalid data and ask what to do...
def OnRescanDataset(self, event):
if self.current_scan_path:
self.DoDupeScan(self.current_scan_path)
else:
dlg = wx.MessageDialog(None,
"There is no scan data to re-process.",
'Error', wx.OK | wx.ICON_INFORMATION)
dlg.ShowModal()
def DoGetScanPath(self):
scan_path = ''
dialog = wx.DirDialog(self, "Choose a directory:", defaultPath=os.getcwd(),
style=wx.DD_DEFAULT_STYLE | wx.DD_NEW_DIR_BUTTON)
if dialog.ShowModal() == wx.ID_OK:
scan_path = dialog.GetPath()
dialog.Destroy()
return scan_path
def DoDupeScan(self, scan_path):
self.clip_count = self.dupe_checker.pre_scan(scan_path,
('aiff', 'aif', 'wav'))
if self.clip_count > 0:
self.dupe_checker.do_scan()
display_txt = "Scanning %d audio clips" % (self.clip_count)
prog_dlg = wx.ProgressDialog("Dupe Scan Progress", display_txt, 100,
style=wx.PD_CAN_ABORT | wx.PD_REMAINING_TIME |
wx.PD_ELAPSED_TIME)
current = 0
do_more = True
while current < self.clip_count:
progress = (float(current) / float(self.clip_count)) * 100
wx.Sleep(1)
do_more = prog_dlg.Update(progress, "Scanned %d of %d clips" %
(current, self.clip_count))
if not do_more[0]:
self.dupe_checker.do_shutdown()
break
current = self.dupe_checker.update()
if do_more[0]:
self.scan_dataset = self.dupe_checker.get_data()
self.RebuildTree()
self.is_dirty = True
prog_dlg.Destroy()
def OnDoFileRemove(self, event):
item = self.tree.GetSelection()
item_text = self.tree.GetItemText(item)
# Verify that it's a file and not a hash.
if item_text not in self.scan_dataset:
dlg = wx.MessageDialog(None, 'Do you really want to simply delete the file %s?' % (item_text),
'Delete Confirmation',
wx.YES_NO | wx.ICON_EXCLAMATION)
retCode = dlg.ShowModal()
do_remove = False
if (retCode == wx.ID_YES):
do_remove = True
dlg.Destroy()
if do_remove:
# Delete Item From File System, the tree and the data.
# mark the data as dirty.
dupe_hex = self.value_key_map[item_text]
del self.value_key_map[item_text]
del self.scan_dataset[dupe_hex][item_text]
os.unlink(item_text)
self.tree.Delete(item)
self.is_dirty = True
def OnDoFileAliasReplace(self):
# Get Selected File Path from tree.
# If Path and not Hash.
# Do 'Are You Sure' Dlg.
# Get Path to master file
# Delete Sub File
# Create Alias to master file.
# self.tree.Delete(item)
pass
def OnTItemActivate(self, event):
item = event.GetItem()
item_text = self.tree.GetItemText(item)
hash_key = ''
file_path = ''
if item_text in self.scan_dataset:
if self.tree.IsExpanded(item):
self.tree.Collapse(item)
else:
self.tree.Expand(item)
else:
# Have file path
file_path = item_text
hash_key = self.value_key_map[item_text]
if file_path and hash_key:
file_name = os.path.basename(file_path)
file_data = {"name": file_name}
edit_dlg = DupeEdit(file_data, "Edit %s" % (file_name))
edit_dlg.ShowModal()
edit_dlg.Destroy()
print "Return Data => ", str(file_data)
## TODO: Need to finish edit functionality
def OnSaveDupes(self, event):
if not self.scan_dataset:
dlg = wx.MessageDialog(None,
"There is no dupe scan data to save.",
'Error', wx.OK | wx.ICON_INFORMATION)
dlg.ShowModal()
else:
self.DoDataSave(self.current_dupe_file)
def OnSaveDupesAs(self, event):
if not self.scan_dataset:
dlg = wx.MessageDialog(None,
"There is no dupe scan data to save.",
'Error', wx.OK | wx.ICON_INFORMATION)
dlg.ShowModal()
else:
self.DoDataSave('', True)
def DoOpenFile(self, filename):
if (os.path.exists(os.path.abspath(filename))):
print 'filename => ', filename
d_shelve = DupeShelve()
full_data = d_shelve.ReadDB(filename)
self.scan_dataset = full_data['scan_data']
self.current_scan_path = full_data['scan_path']
self.current_dupe_file = filename
self.RebuildTree()
def OnOpenDupes(self, event):
open_path = self.GetFilePath('Open Dataset', wx.OPEN)
if (open_path):
d_shelve = DupeShelve()
full_data = d_shelve.ReadDB(open_path)
self.scan_dataset = full_data['scan_data']
self.current_scan_path = full_data['scan_path']
self.current_dupe_file = open_path
self.RebuildTree()
def DoDataSave(self, in_path, is_new=False):
if not in_path or is_new:
dlg_txt = 'Save New Dataset'
if is_new:
dlg_txt = 'Save Dataset As...'
in_path = self.GetFilePath(dlg_txt, wx.SAVE)
if in_path:
full_data = {}
full_data['scan_data'] = self.scan_dataset
full_data['scan_path'] = self.current_scan_path
d_shelve = DupeShelve()
d_shelve.WriteDB(in_path, full_data)
self.is_dirty = False
## Helpers and utlitiy funcs ##
def GetFilePath(self, title, dlg_type):
save_path = ''
allowed_types = "Python DB (*.db)|*.db|" \
"All files (*.*)|*.*"
dialog = wx.FileDialog(self, title, os.getcwd(), "", allowed_types, dlg_type)
if dialog.ShowModal() == wx.ID_OK:
save_path = dialog.GetPath()
dialog.Destroy()
return save_path
def GetHelpHTML(self):
return """<html>
