mirror of https://github.com/evilhero/mylar
246 lines
8.3 KiB
Python
246 lines
8.3 KiB
Python
"""A PyQT4 dialog to select from automated issue matches"""
|
|
|
|
# Copyright 2012-2014 Anthony Beville
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import os
|
|
#import sys
|
|
|
|
from PyQt4 import QtCore, QtGui, uic
|
|
#from PyQt4.QtCore import QUrl, pyqtSignal, QByteArray
|
|
|
|
from settings import ComicTaggerSettings
|
|
from comicarchive import MetaDataStyle
|
|
from coverimagewidget import CoverImageWidget
|
|
from comictaggerlib.ui.qtutils import reduceWidgetFontSize
|
|
#from imagefetcher import ImageFetcher
|
|
#from comicvinetalker import ComicVineTalker
|
|
#import utils
|
|
|
|
|
|
class AutoTagMatchWindow(QtGui.QDialog):
|
|
|
|
volume_id = 0
|
|
|
|
def __init__(self, parent, match_set_list, style, fetch_func):
|
|
super(AutoTagMatchWindow, self).__init__(parent)
|
|
|
|
uic.loadUi(
|
|
ComicTaggerSettings.getUIFile('matchselectionwindow.ui'), self)
|
|
|
|
self.altCoverWidget = CoverImageWidget(
|
|
self.altCoverContainer, CoverImageWidget.AltCoverMode)
|
|
gridlayout = QtGui.QGridLayout(self.altCoverContainer)
|
|
gridlayout.addWidget(self.altCoverWidget)
|
|
gridlayout.setContentsMargins(0, 0, 0, 0)
|
|
|
|
self.archiveCoverWidget = CoverImageWidget(
|
|
self.archiveCoverContainer, CoverImageWidget.ArchiveMode)
|
|
gridlayout = QtGui.QGridLayout(self.archiveCoverContainer)
|
|
gridlayout.addWidget(self.archiveCoverWidget)
|
|
gridlayout.setContentsMargins(0, 0, 0, 0)
|
|
|
|
reduceWidgetFontSize(self.twList)
|
|
reduceWidgetFontSize(self.teDescription, 1)
|
|
|
|
self.setWindowFlags(self.windowFlags() |
|
|
QtCore.Qt.WindowSystemMenuHint |
|
|
QtCore.Qt.WindowMaximizeButtonHint)
|
|
|
|
self.skipButton = QtGui.QPushButton(self.tr("Skip to Next"))
|
|
self.buttonBox.addButton(
|
|
self.skipButton, QtGui.QDialogButtonBox.ActionRole)
|
|
self.buttonBox.button(QtGui.QDialogButtonBox.Ok).setText(
|
|
"Accept and Write Tags")
|
|
|
|
self.match_set_list = match_set_list
|
|
self.style = style
|
|
self.fetch_func = fetch_func
|
|
|
|
self.current_match_set_idx = 0
|
|
|
|
self.twList.currentItemChanged.connect(self.currentItemChanged)
|
|
self.twList.cellDoubleClicked.connect(self.cellDoubleClicked)
|
|
self.skipButton.clicked.connect(self.skipToNext)
|
|
|
|
self.updateData()
|
|
|
|
def updateData(self):
|
|
|
|
self.current_match_set = self.match_set_list[
|
|
self.current_match_set_idx]
|
|
|
|
if self.current_match_set_idx + 1 == len(self.match_set_list):
|
|
self.buttonBox.button(
|
|
QtGui.QDialogButtonBox.Cancel).setDisabled(True)
|
|
# self.buttonBox.button(QtGui.QDialogButtonBox.Ok).setText("Accept")
|
|
self.skipButton.setText(self.tr("Skip"))
|
|
|
|
self.setCoverImage()
|
|
self.populateTable()
|
|
self.twList.resizeColumnsToContents()
|
|
self.twList.selectRow(0)
|
|
|
|
path = self.current_match_set.ca.path
|
|
self.setWindowTitle(
|
|
u"Select correct match or skip ({0} of {1}): {2}".format(
|
|
self.current_match_set_idx + 1,
|
|
len(self.match_set_list),
|
|
os.path.split(path)[1])
|
|
)
|
|
|
|
def populateTable(self):
|
|
|
|
while self.twList.rowCount() > 0:
|
|
self.twList.removeRow(0)
|
|
|
|
self.twList.setSortingEnabled(False)
|
|
|
|
row = 0
|
|
for match in self.current_match_set.matches:
|
|
self.twList.insertRow(row)
|
|
|
|
item_text = match['series']
|
|
item = QtGui.QTableWidgetItem(item_text)
|
|
item.setData(QtCore.Qt.ToolTipRole, item_text)
|
|
item.setData(QtCore.Qt.UserRole, (match,))
|
|
item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
|
|
self.twList.setItem(row, 0, item)
|
|
|
|
if match['publisher'] is not None:
|
|
item_text = u"{0}".format(match['publisher'])
|
|
else:
|
|
item_text = u"Unknown"
|
|
item = QtGui.QTableWidgetItem(item_text)
|
|
item.setData(QtCore.Qt.ToolTipRole, item_text)
|
|
item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
|
|
self.twList.setItem(row, 1, item)
|
|
|
|
month_str = u""
|
|
year_str = u"????"
|
|
if match['month'] is not None:
|
|
month_str = u"-{0:02d}".format(int(match['month']))
|
|
if match['year'] is not None:
|
|
year_str = u"{0}".format(match['year'])
|
|
|
|
item_text = year_str + month_str
|
|
item = QtGui.QTableWidgetItem(item_text)
|
|
item.setData(QtCore.Qt.ToolTipRole, item_text)
|
|
item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
|
|
self.twList.setItem(row, 2, item)
|
|
|
|
item_text = match['issue_title']
|
|
if item_text is None:
|
|
item_text = ""
|
|
item = QtGui.QTableWidgetItem(item_text)
|
|
item.setData(QtCore.Qt.ToolTipRole, item_text)
|
|
item.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
|
|
self.twList.setItem(row, 3, item)
|
|
|
|
row += 1
|
|
|
|
self.twList.resizeColumnsToContents()
|
|
self.twList.setSortingEnabled(True)
|
|
self.twList.sortItems(2, QtCore.Qt.AscendingOrder)
|
|
self.twList.selectRow(0)
|
|
self.twList.resizeColumnsToContents()
|
|
self.twList.horizontalHeader().setStretchLastSection(True)
|
|
|
|
def cellDoubleClicked(self, r, c):
|
|
self.accept()
|
|
|
|
def currentItemChanged(self, curr, prev):
|
|
|
|
if curr is None:
|
|
return
|
|
if prev is not None and prev.row() == curr.row():
|
|
return
|
|
|
|
self.altCoverWidget.setIssueID(self.currentMatch()['issue_id'])
|
|
if self.currentMatch()['description'] is None:
|
|
self.teDescription.setText("")
|
|
else:
|
|
self.teDescription.setText(self.currentMatch()['description'])
|
|
|
|
def setCoverImage(self):
|
|
ca = self.current_match_set.ca
|
|
self.archiveCoverWidget.setArchive(ca)
|
|
|
|
def currentMatch(self):
|
|
row = self.twList.currentRow()
|
|
match = self.twList.item(row, 0).data(
|
|
QtCore.Qt.UserRole).toPyObject()[0]
|
|
return match
|
|
|
|
def accept(self):
|
|
|
|
self.saveMatch()
|
|
self.current_match_set_idx += 1
|
|
|
|
if self.current_match_set_idx == len(self.match_set_list):
|
|
# no more items
|
|
QtGui.QDialog.accept(self)
|
|
else:
|
|
self.updateData()
|
|
|
|
def skipToNext(self):
|
|
self.current_match_set_idx += 1
|
|
|
|
if self.current_match_set_idx == len(self.match_set_list):
|
|
# no more items
|
|
QtGui.QDialog.reject(self)
|
|
else:
|
|
self.updateData()
|
|
|
|
def reject(self):
|
|
reply = QtGui.QMessageBox.question(
|
|
self,
|
|
self.tr("Cancel Matching"),
|
|
self.tr("Are you sure you wish to cancel the matching process?"),
|
|
QtGui.QMessageBox.Yes,
|
|
QtGui.QMessageBox.No)
|
|
|
|
if reply == QtGui.QMessageBox.No:
|
|
return
|
|
|
|
QtGui.QDialog.reject(self)
|
|
|
|
def saveMatch(self):
|
|
|
|
match = self.currentMatch()
|
|
ca = self.current_match_set.ca
|
|
|
|
md = ca.readMetadata(self.style)
|
|
if md.isEmpty:
|
|
md = ca.metadataFromFilename()
|
|
|
|
# now get the particular issue data
|
|
cv_md = self.fetch_func(match)
|
|
if cv_md is None:
|
|
QtGui.QMessageBox.critical(self, self.tr("Network Issue"), self.tr(
|
|
"Could not connect to Comic Vine to get issue details!"))
|
|
return
|
|
|
|
QtGui.QApplication.setOverrideCursor(
|
|
QtGui.QCursor(QtCore.Qt.WaitCursor))
|
|
md.overlay(cv_md)
|
|
success = ca.writeMetadata(md, self.style)
|
|
ca.loadCache([MetaDataStyle.CBI, MetaDataStyle.CIX])
|
|
|
|
QtGui.QApplication.restoreOverrideCursor()
|
|
|
|
if not success:
|
|
QtGui.QMessageBox.warning(self, self.tr("Write Error"), self.tr(
|
|
"Saving the tags to the archive seemed to fail!"))
|