def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
# Config
defaults = {
'show_advanced_settings' : 'false',
'data_source' : '0'
}
self.config = ConfigParser.RawConfigParser(defaults)
self.configfile = os.path.expanduser('~/.config/episoderenamer.conf')
configfiles = self.config.read(['episoderenamer.conf', self.configfile])
if len(configfiles) > 0:
self.configfile = configfiles[0]
if self.config.getboolean('DEFAULT', 'show_advanced_settings'):
self.ui.modeAction.setChecked(True)
self.ui.sourceComboBox.setCurrentIndex(self.config.getint('DEFAULT', 'data_source'))
if self.config.has_option('DEFAULT', 'file_mask'):
if self.config.get('DEFAULT', 'file_mask') == self.ui.maskComboBox.itemText(0):
self.config.remove_option('DEFAULT', 'file_mask')
else:
self.ui.maskComboBox.addItem(self.config.get('DEFAULT', 'file_mask'))
self.ui.maskComboBox.setCurrentIndex(1)
# Icon
fallbackIcon = QtGui.QIcon()
if os.path.exists("episoderenamer_gui.svg"):
fallbackIcon = QtGui.QIcon("episoderenamer_gui.svg")
self.setWindowIcon(QtGui.QIcon().fromTheme("episoderenamer_gui", fallbackIcon))
# Model
self.model = QtGui.QStandardItemModel()
header = QtCore.QStringList("Episode")
header.append("Renamed Episode")
self.model.setHorizontalHeaderLabels(header)
self.ui.file_list.setModel(self.model)
# Connecting signals/actions with slots
QtCore.QObject.connect(self.ui.addFilesAction, QtCore.SIGNAL("triggered()"), self.file_dialog)
QtCore.QObject.connect(self.ui.removeFilesAction, QtCore.SIGNAL("triggered()"), self.remove_files)
QtCore.QObject.connect(self.ui.clearFilesAction, QtCore.SIGNAL("triggered()"), self.clear_files)
QtCore.QObject.connect(self.ui.previewAction, QtCore.SIGNAL("triggered()"), self.get_new_filenames)
QtCore.QObject.connect(self.ui.renameFilesAction, QtCore.SIGNAL("triggered()"), self.rename_files)
QtCore.QObject.connect(self.ui.quitAction, QtCore.SIGNAL("triggered()"), self.close)
QtCore.QObject.connect(self.ui.modeAction, QtCore.SIGNAL("changed()"), self.switch_mode)
QtCore.QObject.connect(self.ui.maskComboBox, QtCore.SIGNAL("editTextChanged(QString)"), self.mask_changed)
QtCore.QObject.connect(self.ui.aboutAction, QtCore.SIGNAL("triggered()"), self.showAbout)
# Shortcuts
self.ui.removeFilesAction.setShortcuts(QtGui.QKeySequence.Delete)
self.ui.quitAction.setShortcuts(QtGui.QKeySequence.Quit)
# Icons
genericIcon = QtGui.QIcon.fromTheme("document-open")
self.ui.addFilesAction.setIcon(QtGui.QIcon.fromTheme("list-add", genericIcon)) #folder-new
self.ui.quitAction.setIcon(QtGui.QIcon.fromTheme("window-close", genericIcon))
self.ui.removeFilesAction.setIcon(QtGui.QIcon.fromTheme("edit-delete", genericIcon))
self.ui.clearFilesAction.setIcon(QtGui.QIcon.fromTheme("edit-clear", genericIcon))
# Context Menu
self.ui.file_list.addAction(self.ui.addFilesAction)
self.ui.file_list.addAction(self.ui.removeFilesAction)
self.ui.file_list.addAction(self.ui.clearFilesAction)
# start in simple mode
self.switch_mode()
# Connecting drag'n'drop events
self.ui.file_list.__class__.dragEnterEvent = self.dragEnterEvent
self.ui.file_list.__class__.dragMoveEvent = self.dragMoveEvent
self.ui.file_list.__class__.dropEvent = self.dropEvent
class EpisodeRenamerGUI(QtGui.QMainWindow):
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
# Config
defaults = {
'show_advanced_settings' : 'false',
'data_source' : '0'
}
self.config = ConfigParser.RawConfigParser(defaults)
self.configfile = os.path.expanduser('~/.config/episoderenamer.conf')
configfiles = self.config.read(['episoderenamer.conf', self.configfile])
if len(configfiles) > 0:
self.configfile = configfiles[0]
if self.config.getboolean('DEFAULT', 'show_advanced_settings'):
self.ui.modeAction.setChecked(True)
self.ui.sourceComboBox.setCurrentIndex(self.config.getint('DEFAULT', 'data_source'))
if self.config.has_option('DEFAULT', 'file_mask'):
if self.config.get('DEFAULT', 'file_mask') == self.ui.maskComboBox.itemText(0):
self.config.remove_option('DEFAULT', 'file_mask')
else:
self.ui.maskComboBox.addItem(self.config.get('DEFAULT', 'file_mask'))
self.ui.maskComboBox.setCurrentIndex(1)
# Icon
fallbackIcon = QtGui.QIcon()
if os.path.exists("episoderenamer_gui.svg"):
fallbackIcon = QtGui.QIcon("episoderenamer_gui.svg")
self.setWindowIcon(QtGui.QIcon().fromTheme("episoderenamer_gui", fallbackIcon))
# Model
self.model = QtGui.QStandardItemModel()
header = QtCore.QStringList("Episode")
header.append("Renamed Episode")
self.model.setHorizontalHeaderLabels(header)
self.ui.file_list.setModel(self.model)
# Connecting signals/actions with slots
QtCore.QObject.connect(self.ui.addFilesAction, QtCore.SIGNAL("triggered()"), self.file_dialog)
QtCore.QObject.connect(self.ui.removeFilesAction, QtCore.SIGNAL("triggered()"), self.remove_files)
QtCore.QObject.connect(self.ui.clearFilesAction, QtCore.SIGNAL("triggered()"), self.clear_files)
QtCore.QObject.connect(self.ui.previewAction, QtCore.SIGNAL("triggered()"), self.get_new_filenames)
QtCore.QObject.connect(self.ui.renameFilesAction, QtCore.SIGNAL("triggered()"), self.rename_files)
QtCore.QObject.connect(self.ui.quitAction, QtCore.SIGNAL("triggered()"), self.close)
QtCore.QObject.connect(self.ui.modeAction, QtCore.SIGNAL("changed()"), self.switch_mode)
QtCore.QObject.connect(self.ui.maskComboBox, QtCore.SIGNAL("editTextChanged(QString)"), self.mask_changed)
QtCore.QObject.connect(self.ui.aboutAction, QtCore.SIGNAL("triggered()"), self.showAbout)
# Shortcuts
self.ui.removeFilesAction.setShortcuts(QtGui.QKeySequence.Delete)
self.ui.quitAction.setShortcuts(QtGui.QKeySequence.Quit)
# Icons
genericIcon = QtGui.QIcon.fromTheme("document-open")
self.ui.addFilesAction.setIcon(QtGui.QIcon.fromTheme("list-add", genericIcon)) #folder-new
self.ui.quitAction.setIcon(QtGui.QIcon.fromTheme("window-close", genericIcon))
self.ui.removeFilesAction.setIcon(QtGui.QIcon.fromTheme("edit-delete", genericIcon))
self.ui.clearFilesAction.setIcon(QtGui.QIcon.fromTheme("edit-clear", genericIcon))
# Context Menu
self.ui.file_list.addAction(self.ui.addFilesAction)
self.ui.file_list.addAction(self.ui.removeFilesAction)
self.ui.file_list.addAction(self.ui.clearFilesAction)
# start in simple mode
self.switch_mode()
# Connecting drag'n'drop events
self.ui.file_list.__class__.dragEnterEvent = self.dragEnterEvent
self.ui.file_list.__class__.dragMoveEvent = self.dragMoveEvent
self.ui.file_list.__class__.dropEvent = self.dropEvent
def switch_mode(self):
if self.ui.modeAction.isChecked():
self.ui.sourceLabel.show()
self.ui.sourceComboBox.show()
# unsupported for now
# self.ui.atomicParsleyLabel.show()
# self.ui.atomicParsleyCheckBox.show()
self.ui.maskLabel.show()
self.ui.maskComboBox.show()
else:
self.ui.sourceLabel.hide()
self.ui.sourceComboBox.hide()
# self.ui.atomicParsleyLabel.hide()
# self.ui.atomicParsleyCheckBox.hide()
self.ui.maskLabel.hide()
self.ui.maskComboBox.hide()
self.ui.atomicParsleyLabel.hide()
self.ui.atomicParsleyCheckBox.hide()
def file_dialog(self):
fd = QtGui.QFileDialog(self)
fd.setFileMode(QtGui.QFileDialog.ExistingFiles)
last_directory = ''
if self.config.has_option('DEFAULT', 'last_directory'):
last_directory = unicode(self.config.get('DEFAULT', 'last_directory'), "utf-8")
filenames = fd.getOpenFileNames(self, "Episode Renamer - Add Files", last_directory)
# Save dir
if len(filenames) > 0:
self.config.set('DEFAULT', 'last_directory', os.path.dirname(unicode(filenames[0].toUtf8(), "utf-8")).encode('UTF-8'))
for filename in filenames:
self.add_file(filename)
def add_file(self, filename):
episode_item = QtGui.QStandardItem(filename)
episode_item.setEditable(False)
renamed_item = QtGui.QStandardItem()
renamed_item.setEditable(False)
self.model.appendRow([episode_item,renamed_item])
self.ui.file_list.resizeColumnToContents(0)
def remove_files(self):
selected_items = self.ui.file_list.selectionModel().selectedIndexes()
while len(selected_items):
self.model.removeRows(selected_items[0].row(), 1)
selected_items = self.ui.file_list.selectionModel().selectedIndexes()
def clear_files(self):
self.model.removeRows(0, self.model.rowCount())
def mask_changed(self):
self.config.set('DEFAULT', 'file_mask', str(self.ui.maskComboBox.currentText()))
def dragEnterEvent(self, event):
if event.mimeData().hasUrls():
event.acceptProposedAction()
def dragMoveEvent(self, event):
if event.mimeData().hasUrls():
event.acceptProposedAction()
def dropEvent(self, event):
if event.mimeData().hasUrls():
event.acceptProposedAction()
for url in event.mimeData().urls():
if url.scheme() == "file":
self.add_file(url.path())
def showMsgBox(self, title, text, icon, info_text=''):
msgBox = QtGui.QMessageBox(self)
msgBox.setWindowTitle(title)
msgBox.setIcon(icon)
msgBox.setText(text)
msgBox.exec_()
def showAbout(self):
dialog = QtGui.QDialog(self)
dialog.ui = Ui_Dialog()
dialog.ui.setupUi(dialog)
dialog.setAttribute(QtCore.Qt.WA_DeleteOnClose)
dialog.ui.logo_label.setPixmap(QtGui.QPixmap(self.windowIcon().pixmap(100,100)))
dialog.exec_()
def get_new_filenames(self):
if self.model.rowCount() == 0:
self.showMsgBox(
"Episode Renamer - No Files",
"Please add some files first, so new names can be found for them!",
QtGui.QMessageBox.Information
)
return
show_name = unicode(self.ui.showLineEdit.text().toUtf8(), "utf-8")
import optparse
options = optparse.Values
options.google = True
options.preview = True
options.use_atomic_parsley = False
if self.ui.sourceComboBox.currentText() == 'IMDb API':
parser = episoderenamer.parse_imdbapi
elif self.ui.sourceComboBox.currentText() == 'epguides.com':
parser = episoderenamer.parse_epguides
else:
parser = episoderenamer.parse_imdb
try:
if show_name == '':
self.showMsgBox(
"Episode Renamer - No Show Name",
"Please enter the name of the show the episodes belong to!",
QtGui.QMessageBox.Information
)
return
# Get Show Name
show = parser(show_name, options)
except BaseException, e:
self.showMsgBox(
"Episode Renamer - No Show Name",
"The show '%s' could not be found with the current data source engine. Please try a different one." % show_name,
QtGui.QMessageBox.Warning
)
return
# Get List of Files
files = {}
for row in range(0,self.model.rowCount()):
full_filename = unicode(self.model.item(row).text().toUtf8(), "utf-8")
files[os.path.basename(full_filename)] = full_filename
# Get Mapping from Old Files to New Files
new_filenames = episoderenamer.rename_files(show, files.keys(), str(self.ui.maskComboBox.currentText()), True, False)
# Update Model with New Files
for filename in files:
if filename in new_filenames:
full_filename = files[filename]
row = self.model.findItems(full_filename)[0].row()
self.model.item(row, 1).setText(new_filenames[filename])
self.ui.file_list.resizeColumnToContents(1)
self.ui.file_list.horizontalScrollBar().setSliderPosition(self.ui.file_list.horizontalScrollBar().maximum())