def data(self, index, role=QtCore.Qt.DisplayRole):
if role not in (QtCore.Qt.DisplayRole, QtCore.Qt.ToolTipRole, QtCore.Qt.ForegroundRole):
return None
column = index.column()
info = self.raw.getInfo()
if role == QtCore.Qt.ForegroundRole:
if self.raw.getStatus() == tv_types.EpisodeRenameItem.MISSING_NEW:
return QtGui.QBrush(QtCore.Qt.red)
elif self.raw.getStatus() == tv_types.EpisodeRenameItem.MISSING_OLD:
return QtGui.QBrush(QtCore.Qt.gray)
if column == Columns.COL_OLD_NAME:
if role == QtCore.Qt.ToolTipRole:
return self.raw.filename
else:
return file_helper.FileHelper.basename(self.raw.filename)
elif column == Columns.COL_NEW_NUM:
if info.ep_num == tv_types.UNRESOLVED_KEY:
return None
else:
return info.ep_num
elif column == Columns.COL_NEW_NAME:
return info.ep_name
elif column == Columns.COL_STATUS:
return self.raw.getStatus()
elif column == Columns.COL_FILE_SIZE:
if self.raw.canEditInfo():
return utils.bytesToString(file_helper.FileHelper.getFileSize(self.raw.filename))
return None
def data(self, index, role):
if not index.isValid():
return None
if role not in (QtCore.Qt.ForegroundRole,
QtCore.Qt.DisplayRole,
QtCore.Qt.ToolTipRole,
QtCore.Qt.CheckStateRole,
RAW_DATA_ROLE) or \
(role == QtCore.Qt.CheckStateRole and index.column() != Columns.COL_CHECK):
return None
item = self._movies[index.row()]
movie = item.movie
info = movie.getInfo()
if role == RAW_DATA_ROLE:
return copy.copy(movie)
col = index.column()
if role == QtCore.Qt.ForegroundRole and not self._isItemValid(item):
return QtGui.QBrush(QtCore.Qt.red)
elif col == Columns.COL_CHECK:
if role == QtCore.Qt.CheckStateRole:
if self._performMoveOnItem(item):
return QtCore.Qt.Checked
else:
return QtCore.Qt.Unchecked
elif col == Columns.COL_OLD_NAME:
if role == QtCore.Qt.DisplayRole:
return file_helper.FileHelper.basename(movie.filename)
else:
return movie.filename
elif col == Columns.COL_NEW_NAME:
return info.title
elif col == Columns.COL_DISC:
return info.part
elif col == Columns.COL_STATUS:
return item.cached_status_text
elif col == Columns.COL_YEAR:
return info.year
elif col == Columns.COL_GENRE:
return info.getGenre()
elif col == Columns.COL_FILE_SIZE:
if movie.canEditInfo():
return utils.bytesToString(file_helper.FileHelper.getFileSize(movie.filename))
elif col == Columns.COL_SERIES:
return info.series
def set_config(self, data):
data = data or config.InputConfig()
self.folderEdit.setText(data.folder or os.path.abspath(os.path.curdir))
self.isRecursiveCheckBox.setChecked(data.recursive)
if data.allExtensions:
self.anyExtRadioButton.setChecked(True)
else:
self.restrictedExtRadioButton.setChecked(True)
if data.allFileSizes:
self.anySizeRadioButton.setChecked(True)
else:
self.restrictedSizeRadioButton.setChecked(True)
self.fileExtensionEdit.setText(data.extensions)
fileSize, fileDenom = utils.bytesToString(data.minFileSizeBytes).split()
self.sizeSpinBox.setValue(int(float(fileSize)))
self.sizeComboBox.setCurrentIndex(self.sizeComboBox.findText(fileDenom))
self._holder.set_config(data.sources or [])
self.onSourcesWidgetFinished()
def setConfig(self, data):
data = data or config.InputConfig()
self.folder_edit.setText(data.folder or os.path.abspath(os.path.curdir))
self.is_recursive_check_box.setChecked(data.recursive)
if data.all_extensions:
self.any_extension_radio_button.setChecked(True)
else:
self.restricted_extension_radio_button.setChecked(True)
if data.all_file_sizes:
self.any_size_radio_button.setChecked(True)
else:
self.restricted_size_radio_button.setChecked(True)
self.file_extension_edit.setText(data.extensions)
file_size, file_denom = utils.bytesToString(data.min_file_size_bytes).split()
self.filesize_spin_box.setValue(int(float(file_size)))
self.filesize_combo_box.setCurrentIndex(self.filesize_combo_box.findText(file_denom))
self._holder.setConfig(data.sources or [])
self.onSourcesWidgetFinished()
