def renameAll(self):
mod = self.model
for row in range(mod.rowCount()):
newname = unicode(mod.data(mod.index(row, 1)).toString())
if newname:
oldname = unicode(mod.data(mod.index(row, 0)).toString())
done = rename_file(self.episodeInfos[row]['dir'],
oldname, newname)
if done:
self.ui.infoLabel.setText(i18n("Renamed {0} to {1}") \
% (oldname, newname))
else:
self.ui.infoLabel.setText(i18n("Couldn't renamed %s.") \
% oldname)
def newname(self, info):
if info['show'] == '' or info['season'] == 0 or \
info['episode'] == 0:
return ""
episodeName = self.api_parser.getEpisodeName(info)
if episodeName == '':
return ""
name = self.ui.nameMask.text()
name = name.replace(i18n("%season"), unicode(info['season']))
name = name.replace(i18n("%Season"), unicode(info['season']).zfill(2))
name = name.replace(i18n("%episode"), unicode(info['episode']).zfill(2))
name = name.replace(i18n("%show"), info['show'])
name = name.replace(i18n("%name"), episodeName)
return name + "." + info['extension']
def episodeSelected(self):
indexes = self.ui.episodeList.selectionModel().selectedIndexes()
if len(indexes) == 2: #one row contains two columns
row = indexes[0].row()
self.selectedRows = [row]
self.ui.showProperties.show()
self.ui.showName.setDisabled(False)
self.ui.seasonNo.setDisabled(False)
self.ui.episodeNo.setDisabled(False)
self.ui.showName.setPlaceholderText("")
self.ui.showName.setText(self.episodeInfos[row]['show'])
self.ui.seasonNo.setText(unicode(self.episodeInfos[row]['season']))
self.ui.episodeNo.setText(unicode(self.episodeInfos[row]['episode']))
elif len(indexes) > 2:
for index in indexes:
self.selectedRows.append(index.row())
self.selectRows = set(self.selectedRows)
firstRow = indexes[0].row()
self.ui.showProperties.show()
self.ui.showName.setPlaceholderText(i18n("Multiple Edit is on"))
self.ui.showName.setText("")
self.ui.showName.setDisabled(False)
self.ui.seasonNo.setDisabled(True)
self.ui.episodeNo.setDisabled(True)
else:
self.ui.showProperties.hide()
self.ui.showName.setText("")
self.ui.seasonNo.setText("")
self.ui.episodeNo.setText("")
self.ui.showName.setPlaceholderText("")
def get_video_file_filter():
"""returns video file filter for QFileDialog"""
filter_ = i18n("Video Files")+' ('
for ext in consts.VIDEO_EXTS:
filter_ += ' *.'+ext
filter_+= ")"
return filter_
def __init__(self):
QMainWindow.__init__(self)
self.ui = Ui_main()
myWidget = QWidget()
self.ui.setupUi(myWidget)
self.setCentralWidget(myWidget)
self.setWindowTitle("ENBea")
self._setupEpisodeList()
self.parser = EpisodeParser()
self.api_parser = IMDbApiParser()
self.ui.showProperties.hide()
self.ui.progressBar.hide()
self.ui.infoLabel.setText(i18n("Add files"))
self.ui.nameMask.setText(i18n("%season%episode - %name"))
self.selectedRows = []
self.connect_signals()
def showFound(self):
self.ui.progressBar.hide()
self.ui.infoLabel.setText(i18n("Download Successful!"))
self.updateNewNames()
def showNotFound(self):
self.ui.progressBar.hide()
self.ui.infoLabel.setText(i18n("Show list couldn't be parsed"))
def openFolderDialog(self):
dirname = QFileDialog.getExistingDirectory(self, i18n("Select folder"),
'~')
for video in get_videos(unicode(dirname)):
self.addFile(video)
def openFileDialog(self):
fullnames = QFileDialog.getOpenFileNames(self, i18n("Open file"),
'~', filter=get_video_file_filter()
+ ";;"+i18n("All Files")+" (*)")
for fullname in fullnames:
self.addFile(fullname)
def __init__(self, table_data, parent=None, *args):
"""init"""
QAbstractTableModel.__init__(self, parent, *args)
self.table_data = table_data
self.header_data = [i18n('Original Name'), i18n('New Name')]
def startDownloadProgress(self, total, show):
self.ui.infoLabel.setText(i18n("Downloading %s episode list") % show)
self.ui.progressBar.setMinimum(0)
self.ui.progressBar.setMaximum(total)
self.ui.progressBar.show()
def addedToQueue(self, show):
self.ui.infoLabel.setText(i18n("%s list added to download queue.") \
% show)
