class SetDialog(QDialog):
def __init__(self, parent=None):
QDialog.__init__(self,parent)
winsettings('setdialog', self)
vbox = QVBoxLayout()
self._previndex = 0
self.setscombo = QComboBox()
setlabel = QLabel('&Sets')
setlabel.setBuddy(self.setscombo)
vbox.addWidget(setlabel)
comboadd = QToolButton()
comboadd.setIcon(QIcon(':/filenew.png'))
comboadd.setToolTip('Add set')
self.connect(comboadd, SIGNAL('clicked()'), self.addSet)
hbox = QHBoxLayout()
hbox.addWidget(self.setscombo)
hbox.addWidget(comboadd)
vbox.addLayout(hbox)
conditions = QLabel('&Conditions')
vbox.addWidget(conditions)
self.listbox = ListBox()
conditions.setBuddy(self.listbox)
listbuttons = ListButtons()
listhbox = QHBoxLayout()
listhbox.addWidget(self.listbox)
listhbox.addLayout(listbuttons)
vbox.addLayout(listhbox)
label = QLabel('Retrieve values via: ')
self.maintag = QComboBox()
self.maintag.addItems(['artist', 'title', 'genre', 'album', 'year'])
maintaghbox = QHBoxLayout()
maintaghbox.addWidget(label)
maintaghbox.addWidget(self.maintag)
maintaghbox.addStretch()
vbox.addLayout(maintaghbox)
dispformat = QLabel('Display Format')
vbox.addWidget(dispformat)
self.texts = [QLineEdit(), QLineEdit()]
t = ['Original', 'Duplicates']
for i, text in enumerate(self.texts):
label = QLabel(t[i])
label.setBuddy(text)
dispbox = QHBoxLayout()
dispbox.addWidget(label)
dispbox.addWidget(text)
vbox.addLayout(dispbox)
okcancel = OKCancel()
vbox.addLayout(okcancel)
self.connect(okcancel, SIGNAL('ok'), self.okClicked)
self.connect(okcancel, SIGNAL('cancel'), self.cancelClicked)
self.setLayout(vbox)
self.fill(loadsets())
listbuttons.connectToWidget(self)
def addSet(self):
def gettext():
(text, ok) = QInputDialog.getText (self, 'puddletag', 'Enter a name'
'for the set', QLineEdit.Normal)
if ok:
if self.setscombo.findText(text) > -1:
QMessageBox.information (self, 'puddletag', 'The name entered already exists.')
return gettext()
return text
text = gettext()
if text:
self.setscombo.addItem(text)
self._sets.append([unicode(text), ['', ''], []])
self.setscombo.setCurrentIndex(self.setscombo.count() - 1)
def fill(self, sets):
if not sets:
return
self._sets = sets
self.setscombo.clear()
self.setscombo.addItems([z[0] for z in sets])
self.currentSet = sets[0]
self.setscombo.setCurrentIndex(0)
self.connect(self.setscombo, SIGNAL('currentIndexChanged (int)'),
self.changeSet)
def _setCurrentSet(self, s):
[text.setText(disp) for text, disp in zip(self.texts, s[1])]
self.listbox.clear()
[self.listbox.addItem(alg.pprint()) for alg in s[2]]
index = self.maintag.findText(s[3])
if index > -1:
self.maintag.setCurrentIndex(index)
else:
self.maintag.addItem(s[3])
self.maintag.setCurrentIndex(self.maintag.count() - 1)
def _getCurrentSet(self):
return self._sets[self.setscombo.currentIndex()]
currentSet = property(_getCurrentSet, _setCurrentSet)
def changeSet(self, index):
i = self._previndex
prevset = {'setname': self._sets[i][0],
'disp': [unicode(text.text()) for text in self.texts],
'algs': self._sets[i][2],
'maintag': unicode(self.maintag.currentText())}
self._sets[i][1] = prevset['disp']
self._sets[i][2] = prevset['algs']
self._sets[i][3] = prevset['maintag']
saveset(**prevset)
self.currentSet = self._sets[index]
self._previndex = index
def add(self):
win = AlgWin(self)
win.setModal(True)
self.connect(win, SIGNAL('okClicked'), self.addBuddy)
win.show()
def addBuddy(self, alg):
self.listbox.addItem(alg.pprint())
self.currentSet[2].append(alg)
def edit(self):
win = AlgWin(self, self.currentSet[2][self.listbox.currentRow()])
win.setModal(True)
self.connect(win, SIGNAL('okClicked'), self.editBuddy)
win.show()
def editBuddy(self, alg):
self.listbox.item(self.listbox.currentRow()).setText(alg.pprint())
self._sets[self._previndex][2][self.listbox.currentRow()] = alg
def moveUp(self):
self.listbox.moveUp(self.currentSet[2])
def moveDown(self):
self.listbox.moveDown(self.currentSet[2])
def remove(self):
del(self.currentSet[2][self.listbox.currentRow()])
self.listbox.takeItem(self.listbox.currentRow())
def okClicked(self):
i = self.setscombo.currentIndex()
prevset = {'setname': self._sets[i][0],
'disp': [unicode(text.text()) for text in self.texts],
'algs': self._sets[i][2],
'maintag': unicode(self.maintag.currentText())}
saveset(**prevset)
self.close()
self.emit(SIGNAL('setAvailable'), *self.currentSet)
def cancelClicked(self):
self.close()
class MTProfileEdit(QDialog):
def __init__(self, tag_sources, profile=None, parent=None):
super(MTProfileEdit, self).__init__(parent)
self.setWindowTitle(translate('Profile Editor', 'Edit Masstagging Profile'))
winsettings('masstag_profile', self)
self._configs = []
self.tag_sources = tag_sources
self._tsps = []
self._name = QLineEdit(translate('Profile Editor',
'Masstagging Profile'))
self._desc = QLineEdit()
self.listbox = ListBox()
self.okcancel = OKCancel()
self.okcancel.ok.setDefault(True)
self.buttonlist = ListButtons()
self.pattern = QLineEdit('%artist% - %album%/%track% - %title%')
self.pattern.setToolTip(translate('Profile Editor',
"<p>If no tag information is found in a file, "
"the tags retrieved using this pattern will be used instead.</p>"))
self.albumBound = QSpinBox()
self.albumBound.setRange(0,100)
self.albumBound.setValue(70)
self.albumBound.setToolTip(translate('Profile Editor',
"<p>The artist and album fields will be used in "
"determining whether an album matches the retrieved one. "
"Each field will be compared using a fuzzy matching algorithm. "
"If the resulting average match percentage is greater "
"or equal than what you specify here it'll be "
"considered to match.</p>"))
self.matchFields = QLineEdit('artist, title')
self.matchFields.setToolTip(translate('Profile Editor',
'<p>The fields listed here will be used in '
'determining whether a file matches a retrieved track. '
'Each field will be compared using a fuzzy matching '
'algorithm. If the resulting average match '
'percentage is greater than the "Minimum Percentage" '
'it\'ll be considered to match.</p>'))
self.trackBound = QSpinBox()
self.trackBound.setRange(0,100)
self.trackBound.setValue(80)
self.jfdi = QCheckBox(translate('Profile Editor',
'Brute force unmatched files.'))
self.jfdi.setToolTip(translate('Profile Editor',
"<p>Check to enable brute forcing matches. "
" If a proper match isn't found for a file, "
'the files will get sorted by filename, '
'the retrieved tag sources by filename and '
'corresponding (unmatched) tracks will matched.</p>'))
self.existing = QCheckBox(translate('Profile Editor',
'Update empty fields only.'))
self.grid = QGridLayout()
self.setLayout(self.grid)
self.grid.addLayout(create_buddy(
translate('Profile Editor', '&Name:'), self._name), 0, 0, 1, 2)
self.grid.addLayout(create_buddy(
translate('Profile Editor', '&Description'), self._desc), 1, 0, 1, 2)
self.grid.addWidget(self.listbox, 2, 0)
self.grid.setRowStretch(2, 1)
self.grid.addLayout(self.buttonlist, 2, 1)
self.grid.addLayout(create_buddy(translate('Profile Editor',
'Pattern to match filenames against.'),
self.pattern, QVBoxLayout()), 3, 0, 1, 2)
self.grid.addLayout(create_buddy(translate('Profile Editor',
'Minimum percentage required for album matches.'),
self.albumBound), 4, 0, 1, 2)
self.grid.addLayout(create_buddy(translate('Profile Editor',
'Match tracks using fields: '),
self.matchFields, QVBoxLayout()), 5, 0, 1, 2)
self.grid.addLayout(create_buddy(translate('Profile Editor',
'Minimum percentage required for track match.'),
self.trackBound), 6, 0, 1, 2)
self.grid.addWidget(self.jfdi, 7, 0, 1, 2)
self.grid.addWidget(self.existing, 8, 0, 1, 2)
self.grid.addLayout(self.okcancel, 9, 0, 1, 2)
connect = lambda control, signal, slot: self.connect(
control, SIGNAL(signal), slot)
connect(self.okcancel, "ok" , self.okClicked)
connect(self.okcancel, "cancel",self.close)
connect(self.buttonlist, "add", self.addClicked)
connect(self.buttonlist, "edit", self.editClicked)
connect(self.buttonlist, "moveup", self.moveUp)
connect(self.buttonlist, "movedown", self.moveDown)
connect(self.buttonlist, "remove", self.remove)
connect(self.buttonlist, "duplicate", self.dupClicked)
connect(self.listbox, "itemDoubleClicked (QListWidgetItem *)",
self.editClicked)
connect(self.listbox, "currentRowChanged(int)", self.enableListButtons)
if profile is not None:
self.setProfile(profile)
self.enableListButtons(self.listbox.currentRow())
def addClicked(self):
win = TSProfileEdit(self.tag_sources, None, self)
win.setModal(True)
self.connect(win, SIGNAL('profileChanged'), self.addTSProfile)
win.show()
def addTSProfile(self, profile):
row = self.listbox.count()
self.listbox.addItem(profile.tag_source.name)
self._tsps.append(profile)
def dupClicked(self):
row = self.listbox.currentRow()
if row == -1:
return
win = TSProfileEdit(self.tag_sources, self._tsps[row], self)
win.setModal(True)
self.connect(win, SIGNAL('profileChanged'), self.addTSProfile)
win.show()
def editClicked(self, item=None):
if item:
row = self.listbox.row(item)
else:
row = self.listbox.currentRow()
if row == -1:
return
win = TSProfileEdit(self.tag_sources, self._tsps[row], self)
win.setModal(True)
self.connect(win, SIGNAL('profileChanged'),
partial(self.replaceTSProfile, row))
win.show()
def replaceTSProfile(self, row, profile):
self._tsps[row] = profile
self.listbox.item(row).setText(profile.tag_source.name)
def enableListButtons(self, val):
if val == -1:
[button.setEnabled(False) for button in self.buttonlist.widgets[1:]]
else:
[button.setEnabled(True) for button in self.buttonlist.widgets[1:]]
def moveDown(self):
self.listbox.moveDown(self._tsps)
def moveUp(self):
self.listbox.moveUp(self._tsps)
def okClicked(self):
fields = [z.strip() for z in
unicode(self.matchFields.text()).split(u',')]
mtp = MassTagProfile(unicode(self._name.text()),
unicode(self._desc.text()), fields, None,
unicode(self.pattern.text()), self._tsps,
self.albumBound.value() / 100.0,
self.trackBound.value() / 100.0, self.jfdi.isChecked(),
self.existing.isChecked(), u'')
self.emit(SIGNAL('profileChanged'), mtp)
self.close()
def remove(self):
row = self.listbox.currentRow()
if row == -1:
return
del(self._tsps[row])
self.listbox.takeItem(row)
def setProfile(self, profile):
self._tsps = [tsp for tsp in profile.profiles if tsp.tag_source]
[self.listbox.addItem(tsp.tag_source.name) for tsp in self._tsps]
self.albumBound.setValue(profile.album_bound * 100)
self.pattern.setText(profile.file_pattern)
self.matchFields.setText(u', '.join(profile.fields))
self.trackBound.setValue(profile.track_bound * 100)
self.jfdi.setChecked(profile.jfdi)
self._name.setText(profile.name)
self._desc.setText(profile.desc)
self.existing.setChecked(profile.leave_existing)
class MassTagEdit(QDialog):
def __init__(self, tag_sources, profiles=None, parent = None):
super(MassTagEdit, self).__init__(parent)
self.setWindowTitle(translate('Profile Config',
'Configure Mass Tagging Profiles'))
winsettings('masstag_edit', self)
self.listbox = ListBox()
self.tag_sources = tag_sources
okcancel = OKCancel()
okcancel.ok.setDefault(True)
self.buttonlist = ListButtons()
connect = lambda control, signal, slot: self.connect(
control, SIGNAL(signal), slot)
connect(okcancel, "ok" , self.okClicked)
connect(okcancel, "cancel",self.close)
connect(self.buttonlist, "add", self.addClicked)
connect(self.buttonlist, "edit", self.editClicked)
connect(self.buttonlist, "moveup", self.moveUp)
connect(self.buttonlist, "movedown", self.moveDown)
connect(self.buttonlist, "remove", self.remove)
connect(self.buttonlist, "duplicate", self.dupClicked)
connect(self.listbox, "itemDoubleClicked (QListWidgetItem *)",
self.editClicked)
connect(self.listbox, "currentRowChanged(int)", self.enableListButtons)
self.enableListButtons(self.listbox.currentRow())
layout = QVBoxLayout()
self.setLayout(layout)
layout.addWidget(QLabel(translate('Profile Config',
'Masstagging Profiles')))
list_layout = QHBoxLayout()
list_layout.addWidget(self.listbox, 1)
list_layout.addLayout(self.buttonlist)
layout.addLayout(list_layout)
layout.addLayout(okcancel)
if profiles is not None:
self.setProfiles(profiles)
def addClicked(self):
win = MTProfileEdit(self.tag_sources, parent=self)
win.setModal(True)
self.connect(win, SIGNAL('profileChanged'), self.addProfile)
win.show()
def addProfile(self, profile):
row = self.listbox.count()
self.listbox.addItem(profile.name)
self._profiles.append(profile)
def dupClicked(self):
row = self.listbox.currentRow()
if row == -1:
return
win = MTProfileEdit(self.tag_sources, deepcopy(self._profiles[row]), self)
win.setModal(True)
self.connect(win, SIGNAL('profileChanged'), self.addProfile)
win.show()
def editClicked(self, item=None):
if item:
row = self.listbox.row(item)
else:
row = self.listbox.currentRow()
if row == -1:
return
win = MTProfileEdit(self.tag_sources, self._profiles[row], self)
win.setModal(True)
self.connect(win, SIGNAL('profileChanged'),
partial(self.replaceProfile, row))
win.show()
def enableListButtons(self, val):
if val == -1:
[button.setEnabled(False) for button in self.buttonlist.widgets[1:]]
else:
[button.setEnabled(True) for button in self.buttonlist.widgets[1:]]
def loadProfiles(self, dirpath=PROFILEDIR):
profiles = load_all_mtps(dirpath, self.tag_sources)
self.setProfiles(filter(None, profiles))
def moveDown(self):
self.listbox.moveDown(self._profiles)
def moveUp(self):
self.listbox.moveUp(self._profiles)
def okClicked(self):
self.emit(SIGNAL('profilesChanged'), self._profiles)
self.saveProfiles(PROFILEDIR, self._profiles)
self.close()
def remove(self):
row = self.listbox.currentRow()
if row == -1:
return
del(self._profiles[row])
self.listbox.takeItem(row)
def replaceProfile(self, row, profile):
self._profiles[row] = profile
self.listbox.item(row).setText(profile.name)
def saveProfiles(self, dirpath, profiles):
filenames = {}
order = []
for profile in profiles:
filename = profile.name + u'.mtp'
i = 0
while filename in filenames:
filename = u'%s_%d%s' % (profile.name, i, u'.mtp')
i += 1
filenames[filename] = profile
order.append(profile.name)
files = glob.glob(os.path.join(dirpath, '*.mtp'))
for f in files:
if f not in filenames:
try:
os.remove(f)
except EnvironmentError:
pass
for filename, profile in filenames.items():
save_mtp(profile, os.path.join(dirpath, filename))
f = open(os.path.join(dirpath, 'order'), 'w')
f.write(u'\n'.join(order))
f.close()
def setProfiles(self, profiles):
self._profiles = profiles
for profile in self._profiles:
self.listbox.addItem(profile.name)
class SetDialog(QDialog):
def __init__(self, parent=None):
QDialog.__init__(self,parent)
winsettings('setdialog', self)
vbox = QVBoxLayout()
self._previndex = 0
self.setscombo = QComboBox()
setlabel = QLabel('&Sets')
setlabel.setBuddy(self.setscombo)
vbox.addWidget(setlabel)
comboadd = QToolButton()
comboadd.setIcon(QIcon(':/filenew.png'))
comboadd.setToolTip('Add set')
self.connect(comboadd, SIGNAL('clicked()'), self.addSet)
hbox = QHBoxLayout()
hbox.addWidget(self.setscombo)
hbox.addWidget(comboadd)
vbox.addLayout(hbox)
conditions = QLabel('&Conditions')
vbox.addWidget(conditions)
self.listbox = ListBox()
conditions.setBuddy(self.listbox)
listbuttons = ListButtons()
listhbox = QHBoxLayout()
listhbox.addWidget(self.listbox)
listhbox.addLayout(listbuttons)
vbox.addLayout(listhbox)
label = QLabel('Retrieve values via: ')
self.maintag = QComboBox()
self.maintag.addItems(['artist', 'title', 'genre', 'album', 'year'])
maintaghbox = QHBoxLayout()
maintaghbox.addWidget(label)
maintaghbox.addWidget(self.maintag)
maintaghbox.addStretch()
vbox.addLayout(maintaghbox)
dispformat = QLabel('Display Format')
vbox.addWidget(dispformat)
self.texts = [QLineEdit(), QLineEdit()]
t = ['Original', 'Duplicates']
for i, text in enumerate(self.texts):
label = QLabel(t[i])
label.setBuddy(text)
dispbox = QHBoxLayout()
dispbox.addWidget(label)
dispbox.addWidget(text)
vbox.addLayout(dispbox)
okcancel = OKCancel()
vbox.addLayout(okcancel)
self.connect(okcancel, SIGNAL('ok'), self.okClicked)
self.connect(okcancel, SIGNAL('cancel'), self.cancelClicked)
self.setLayout(vbox)
self.fill(loadsets())
listbuttons.connectToWidget(self)
def addSet(self):
def gettext():
(text, ok) = QInputDialog.getText (self, 'puddletag', 'Enter a name'
'for the set', QLineEdit.Normal)
if ok:
if self.setscombo.findText(text) > -1:
QMessageBox.information (self, 'puddletag', 'The name entered already exists.')
return gettext()
return text
text = gettext()
if text:
self.setscombo.addItem(text)
self._sets.append([unicode(text), ['', ''], []])
self.setscombo.setCurrentIndex(self.setscombo.count() - 1)
def fill(self, sets):
if not sets:
return
self._sets = sets
self.setscombo.clear()
self.setscombo.addItems([z[0] for z in sets])
self.currentSet = sets[0]
self.setscombo.setCurrentIndex(0)
self.connect(self.setscombo, SIGNAL('currentIndexChanged (int)'),
self.changeSet)
def _setCurrentSet(self, s):
[text.setText(disp) for text, disp in zip(self.texts, s[1])]
self.listbox.clear()
[self.listbox.addItem(alg.pprint()) for alg in s[2]]
index = self.maintag.findText(s[3])
if index > -1:
self.maintag.setCurrentIndex(index)
else:
self.maintag.addItem(s[3])
self.maintag.setCurrentIndex(self.maintag.count() - 1)
def _getCurrentSet(self):
return self._sets[self.setscombo.currentIndex()]
currentSet = property(_getCurrentSet, _setCurrentSet)
def changeSet(self, index):
i = self._previndex
prevset = {'setname': self._sets[i][0],
'disp': [unicode(text.text()) for text in self.texts],
'algs': self._sets[i][2],
'maintag': unicode(self.maintag.currentText())}
self._sets[i][1] = prevset['disp']
self._sets[i][2] = prevset['algs']
self._sets[i][3] = prevset['maintag']
saveset(**prevset)
self.currentSet = self._sets[index]
self._previndex = index
def add(self):
win = AlgWin(self)
win.setModal(True)
self.connect(win, SIGNAL('okClicked'), self.addBuddy)
win.show()
def addBuddy(self, alg):
self.listbox.addItem(alg.pprint())
self.currentSet[2].append(alg)
def edit(self):
win = AlgWin(self, self.currentSet[2][self.listbox.currentRow()])
win.setModal(True)
self.connect(win, SIGNAL('okClicked'), self.editBuddy)
win.show()
def editBuddy(self, alg):
self.listbox.item(self.listbox.currentRow()).setText(alg.pprint())
self._sets[self._previndex][2][self.listbox.currentRow()] = alg
def moveUp(self):
self.listbox.moveUp(self.currentSet[2])
def moveDown(self):
self.listbox.moveDown(self.currentSet[2])
def remove(self):
del(self.currentSet[2][self.listbox.currentRow()])
self.listbox.takeItem(self.listbox.currentRow())
def okClicked(self):
i = self.setscombo.currentIndex()
prevset = {'setname': self._sets[i][0],
'disp': [unicode(text.text()) for text in self.texts],
'algs': self._sets[i][2],
'maintag': unicode(self.maintag.currentText())}
saveset(**prevset)
self.close()
self.emit(SIGNAL('setAvailable'), *self.currentSet)
def cancelClicked(self):
self.close()
