def __init__(self, parent, mode, expand_on_click=True): super(CoverImageWidget, self).__init__(parent) uic.loadUi(ComicTaggerSettings.getUIFile('coverimagewidget.ui'), self) reduceWidgetFontSize(self.label) self.mode = mode self.comicVine = ComicVineTalker() self.page_loader = None self.showControls = True self.btnLeft.setIcon(QIcon(ComicTaggerSettings.getGraphic('left.png'))) self.btnRight.setIcon( QIcon(ComicTaggerSettings.getGraphic('right.png'))) self.btnLeft.clicked.connect(self.decrementImage) self.btnRight.clicked.connect(self.incrementImage) self.resetWidget() if expand_on_click: clickable(self.lblImage).connect(self.showPopup) else: self.lblImage.setToolTip("") self.updateContent()
def __init__(self, parent, metadata): super(PageBrowserWindow, self).__init__(parent) uic.loadUi(ComicTaggerSettings.getUIFile('pagebrowser.ui' ), self) self.pageWidget = CoverImageWidget( self.pageContainer, CoverImageWidget.ArchiveMode ) gridlayout = QtGui.QGridLayout( self.pageContainer ) gridlayout.addWidget( self.pageWidget ) gridlayout.setContentsMargins(0,0,0,0) self.pageWidget.showControls = False self.setWindowFlags(self.windowFlags() | QtCore.Qt.WindowSystemMenuHint | QtCore.Qt.WindowMaximizeButtonHint) self.comic_archive = None self.page_count = 0 self.current_page_num = 0 self.metadata = metadata self.buttonBox.button(QtGui.QDialogButtonBox.Close).setDefault(True) if platform.system() == "Darwin": self.btnPrev.setText("<<") self.btnNext.setText(">>") else: self.btnPrev.setIcon(QtGui.QIcon( ComicTaggerSettings.getGraphic('left.png' ))) self.btnNext.setIcon(QtGui.QIcon( ComicTaggerSettings.getGraphic('right.png'))) self.btnNext.clicked.connect( self.nextPage ) self.btnPrev.clicked.connect( self.prevPage ) self.show() self.btnNext.setEnabled( False ) self.btnPrev.setEnabled( False )
def __init__(self, parent, image_pixmap): super(ImagePopup, self).__init__(parent) uic.loadUi(ComicTaggerSettings.getUIFile('imagepopup.ui' ), self) QtGui.QApplication.setOverrideCursor(QtGui.QCursor(QtCore.Qt.WaitCursor)) #self.setWindowModality(QtCore.Qt.WindowModal) self.setWindowFlags(QtCore.Qt.Popup) self.setWindowState(QtCore.Qt.WindowFullScreen) self.imagePixmap = image_pixmap screen_size = QtGui.QDesktopWidget().screenGeometry() self.resize(screen_size.width(), screen_size.height()) self.move( 0, 0) # This is a total hack. Uses a snapshot of the desktop, and overlays a # translucent screen over it. Probably can do it better by setting opacity of a # widget self.desktopBg = QtGui.QPixmap.grabWindow(QtGui.QApplication.desktop ().winId(), 0,0, screen_size.width(), screen_size.height()) bg = QtGui.QPixmap(ComicTaggerSettings.getGraphic('popup_bg.png')) self.clientBgPixmap = bg.scaled(screen_size.width(), screen_size.height()) self.setMask(self.clientBgPixmap.mask()) self.applyImagePixmap() self.showFullScreen() self.raise_( ) QtGui.QApplication.restoreOverrideCursor()
def __init__( self, path, settings ): self.path = path self.ci_xml_filename = 'ComicInfo.xml' self.comet_default_filename = 'CoMet.xml' self.resetCache() self.settings = settings if self.zipTest(): self.archive_type = self.ArchiveType.Zip self.archiver = ZipArchiver( self.path ) elif self.rarTest(): self.archive_type = self.ArchiveType.Rar self.archiver = RarArchiver( self.path, settings ) elif os.path.isdir( self.path ): self.archive_type = self.ArchiveType.Folder self.archiver = FolderArchiver( self.path ) else: self.archive_type = self.ArchiveType.Unknown self.archiver = UnknownArchiver( self.path ) if ComicArchive.logo_data is None: fname = ComicTaggerSettings.getGraphic('nocover.png') with open(fname, 'rb') as fd: ComicArchive.logo_data = fd.read()
def __init__(self, path, rar_exe_path=None): self.path = path self.rar_exe_path = rar_exe_path self.ci_xml_filename = 'ComicInfo.xml' self.comet_default_filename = 'CoMet.xml' self.resetCache() if self.rarTest(): self.archive_type = self.ArchiveType.Rar self.archiver = RarArchiver(self.path, rar_exe_path=self.rar_exe_path) elif self.zipTest(): self.archive_type = self.ArchiveType.Zip self.archiver = ZipArchiver(self.path) elif os.path.isdir(self.path): self.archive_type = self.ArchiveType.Folder self.archiver = FolderArchiver(self.path) else: self.archive_type = self.ArchiveType.Unknown self.archiver = UnknownArchiver(self.path) if ComicArchive.logo_data is None: fname = ComicTaggerSettings.getGraphic('nocover.png') with open(fname, 'rb') as fd: ComicArchive.logo_data = fd.read()
def display_match_set_for_choice(label, match_set, opts, settings): print(u"{0} -- {1}:".format(match_set.filename, label)) # sort match list by year match_set.matches.sort(key=lambda k: k['year']) for (counter, m) in enumerate(match_set.matches): counter += 1 print(u" {0}. {1} #{2} [{3}] ({4}/{5}) - {6}".format( counter, m['series'], m['issue_number'], m['publisher'], m['month'], m['year'], m['issue_title'])) if opts.interactive: while True: i = raw_input("Choose a match #, or 's' to skip: ") if (i.isdigit() and int(i) in range( 1, len(match_set.matches) + 1)) or i == 's': break if i != 's': i = int(i) - 1 # save the data! # we know at this point, that the file is all good to go ca = ComicArchive(match_set.filename, settings.rar_exe_path, ComicTaggerSettings.getGraphic('nocover.png')) md = create_local_metadata(opts, ca, ca.hasMetadata(opts.data_style)) cv_md = actual_issue_data_fetch(match_set.matches[int(i)], settings, opts) md.overlay(cv_md) actual_metadata_save(ca, opts, md)
def __init__(self, parent, image_pixmap): super(ImagePopup, self).__init__(parent) uic.loadUi(ComicTaggerSettings.getUIFile('imagepopup.ui'), self) QtGui.QApplication.setOverrideCursor( QtGui.QCursor(QtCore.Qt.WaitCursor)) # self.setWindowModality(QtCore.Qt.WindowModal) self.setWindowFlags(QtCore.Qt.Popup) self.setWindowState(QtCore.Qt.WindowFullScreen) self.imagePixmap = image_pixmap screen_size = QtGui.QDesktopWidget().screenGeometry() self.resize(screen_size.width(), screen_size.height()) self.move(0, 0) # This is a total hack. Uses a snapshot of the desktop, and overlays a # translucent screen over it. Probably can do it better by setting opacity of a # widget self.desktopBg = QtGui.QPixmap.grabWindow( QtGui.QApplication.desktop().winId(), 0, 0, screen_size.width(), screen_size.height()) bg = QtGui.QPixmap(ComicTaggerSettings.getGraphic('popup_bg.png')) self.clientBgPixmap = bg.scaled(screen_size.width(), screen_size.height()) self.setMask(self.clientBgPixmap.mask()) self.applyImagePixmap() self.showFullScreen() self.raise_() QtGui.QApplication.restoreOverrideCursor()
def addPathItem(self, path): path = unicode(path) path = os.path.abspath(path) # print "processing", path if self.isListDupe(path): return self.getCurrentListRow(path) ca = ComicArchive( path, self.settings.rar_exe_path, ComicTaggerSettings.getGraphic('nocover.png')) if ca.seemsToBeAComicArchive(): row = self.twList.rowCount() self.twList.insertRow(row) fi = FileInfo(ca) filename_item = QTableWidgetItem() folder_item = QTableWidgetItem() cix_item = FileTableWidgetItem() cbi_item = FileTableWidgetItem() readonly_item = FileTableWidgetItem() type_item = QTableWidgetItem() filename_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) filename_item.setData(Qt.UserRole, fi) self.twList.setItem( row, FileSelectionList.fileColNum, filename_item) folder_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) self.twList.setItem( row, FileSelectionList.folderColNum, folder_item) type_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) self.twList.setItem(row, FileSelectionList.typeColNum, type_item) cix_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) cix_item.setTextAlignment(Qt.AlignHCenter) self.twList.setItem(row, FileSelectionList.CRFlagColNum, cix_item) cbi_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) cbi_item.setTextAlignment(Qt.AlignHCenter) self.twList.setItem(row, FileSelectionList.CBLFlagColNum, cbi_item) readonly_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) readonly_item.setTextAlignment(Qt.AlignHCenter) self.twList.setItem( row, FileSelectionList.readonlyColNum, readonly_item) self.updateRow(row) return row
def addPathItem(self, path): path = unicode(path) path = os.path.abspath(path) # print "processing", path if self.isListDupe(path): return self.getCurrentListRow(path) ca = ComicArchive(path, self.settings.rar_exe_path, ComicTaggerSettings.getGraphic('nocover.png')) if ca.seemsToBeAComicArchive(): row = self.twList.rowCount() self.twList.insertRow(row) fi = FileInfo(ca) filename_item = QTableWidgetItem() folder_item = QTableWidgetItem() cix_item = FileTableWidgetItem() cbi_item = FileTableWidgetItem() readonly_item = FileTableWidgetItem() type_item = QTableWidgetItem() filename_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) filename_item.setData(Qt.UserRole, fi) self.twList.setItem(row, FileSelectionList.fileColNum, filename_item) folder_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) self.twList.setItem(row, FileSelectionList.folderColNum, folder_item) type_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) self.twList.setItem(row, FileSelectionList.typeColNum, type_item) cix_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) cix_item.setTextAlignment(Qt.AlignHCenter) self.twList.setItem(row, FileSelectionList.CRFlagColNum, cix_item) cbi_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) cbi_item.setTextAlignment(Qt.AlignHCenter) self.twList.setItem(row, FileSelectionList.CBLFlagColNum, cbi_item) readonly_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) readonly_item.setTextAlignment(Qt.AlignHCenter) self.twList.setItem(row, FileSelectionList.readonlyColNum, readonly_item) self.updateRow(row) return row
def ctmain(): utils.fix_output_encoding() settings = ComicTaggerSettings() opts = Options() opts.parseCmdLineArgs() # manage the CV API key if opts.cv_api_key: if opts.cv_api_key != settings.cv_api_key: settings.cv_api_key = opts.cv_api_key settings.save() if opts.only_set_key: print("Key set") return ComicVineTalker.api_key = settings.cv_api_key signal.signal(signal.SIGINT, signal.SIG_DFL) if not qt_available and not opts.no_gui: opts.no_gui = True print >> sys.stderr, "PyQt4 is not available. ComicTagger is limited to command-line mode." if opts.no_gui: cli.cli_mode(opts, settings) else: app = QtGui.QApplication(sys.argv) if platform.system() != "Linux": img = QtGui.QPixmap(ComicTaggerSettings.getGraphic('tags.png')) splash = QtGui.QSplashScreen(img) splash.show() splash.raise_() app.processEvents() try: tagger_window = TaggerWindow(opts.file_list, settings, opts=opts) tagger_window.show() if platform.system() != "Linux": splash.finish(tagger_window) sys.exit(app.exec_()) except Exception as e: QtGui.QMessageBox.critical( QtGui.QMainWindow(), "Error", "Unhandled exception in app:\n" + traceback.format_exc())
def ctmain(): utils.fix_output_encoding() settings = ComicTaggerSettings() opts = Options() opts.parseCmdLineArgs() # manage the CV API key if opts.cv_api_key: if opts.cv_api_key != settings.cv_api_key: settings.cv_api_key = opts.cv_api_key settings.save() if opts.only_set_key: print "Key set" return ComicVineTalker.api_key = settings.cv_api_key signal.signal(signal.SIGINT, signal.SIG_DFL) if not qt_available and not opts.no_gui: opts.no_gui = True print "PyQt4 is not available. ComicTagger is limited to command-line mode." if opts.no_gui: cli.cli_mode(opts, settings) else: app = QtGui.QApplication(sys.argv) if platform.system() != "Linux": img = QtGui.QPixmap(ComicTaggerSettings.getGraphic('tags.png')) splash = QtGui.QSplashScreen(img) splash.show() splash.raise_() app.processEvents() try: tagger_window = TaggerWindow(opts.file_list, settings, opts=opts) tagger_window.show() if platform.system() != "Linux": splash.finish(tagger_window) sys.exit(app.exec_()) except Exception, e: QtGui.QMessageBox.critical( QtGui.QMainWindow(), "Error", "Unhandled exception in app:\n" + traceback.format_exc())
def getQImageFromData(image_data): img = QtGui.QImage() success = img.loadFromData(image_data) if not success: try: if pil_available: # Qt doesn't understand the format, but maybe PIL does # so try to convert the image data to uncompressed tiff format im = Image.open(StringIO.StringIO(image_data)) output = StringIO.StringIO() im.save(output, format="TIFF") img.loadFromData(output.getvalue()) success = True except Exception as e: pass # if still nothing, go with default image if not success: img.load(ComicTaggerSettings.getGraphic('nocover.png')) return img
def getQImageFromData(image_data): img = QtGui.QImage() success = img.loadFromData( image_data ) if not success: try: if pil_available: # Qt doesn't understand the format, but maybe PIL does # so try to convert the image data to uncompressed tiff format im = Image.open(StringIO.StringIO(image_data)) output = StringIO.StringIO() im.save(output, format="TIFF") img.loadFromData( output.getvalue() ) success = True except Exception as e: pass # if still nothing, go with default image if not success: img.load(ComicTaggerSettings.getGraphic('nocover.png')) return img
def display_match_set_for_choice(label, match_set, opts, settings): print(u"{0} -- {1}:".format(match_set.filename, label)) # sort match list by year match_set.matches.sort(key=lambda k: k['year']) for (counter, m) in enumerate(match_set.matches): counter += 1 print( u" {0}. {1} #{2} [{3}] ({4}/{5}) - {6}".format( counter, m['series'], m['issue_number'], m['publisher'], m['month'], m['year'], m['issue_title'])) if opts.interactive: while True: i = raw_input("Choose a match #, or 's' to skip: ") if (i.isdigit() and int(i) in range( 1, len(match_set.matches) + 1)) or i == 's': break if i != 's': i = int(i) - 1 # save the data! # we know at this point, that the file is all good to go ca = ComicArchive( match_set.filename, settings.rar_exe_path, ComicTaggerSettings.getGraphic('nocover.png')) md = create_local_metadata( opts, ca, ca.hasMetadata(opts.data_style)) cv_md = actual_issue_data_fetch( match_set.matches[int(i)], settings, opts) md.overlay(cv_md) actual_metadata_save(ca, opts, md)
def process_file_cli(filename, opts, settings, match_results): batch_mode = len(opts.file_list) > 1 ca = ComicArchive( filename, settings.rar_exe_path, ComicTaggerSettings.getGraphic('nocover.png')) if not os.path.lexists(filename): print >> sys.stderr, "Cannot find " + filename return if not ca.seemsToBeAComicArchive(): print >> sys.stderr, "Sorry, but " + \ filename + " is not a comic archive!" return # if not ca.isWritableForStyle(opts.data_style) and (opts.delete_tags or # opts.save_tags or opts.rename_file): if not ca.isWritable() and ( opts.delete_tags or opts.copy_tags or opts.save_tags or opts.rename_file): print >> sys.stderr, "This archive is not writable for that tag type" return has = [False, False, False] if ca.hasCIX(): has[MetaDataStyle.CIX] = True if ca.hasCBI(): has[MetaDataStyle.CBI] = True if ca.hasCoMet(): has[MetaDataStyle.COMET] = True if opts.print_tags: if opts.data_style is None: page_count = ca.getNumberOfPages() brief = "" if batch_mode: brief = u"{0}: ".format(filename) if ca.isZip(): brief += "ZIP archive " elif ca.isRar(): brief += "RAR archive " elif ca.isFolder(): brief += "Folder archive " brief += "({0: >3} pages)".format(page_count) brief += " tags:[ " if not (has[MetaDataStyle.CBI] or has[ MetaDataStyle.CIX] or has[MetaDataStyle.COMET]): brief += "none " else: if has[MetaDataStyle.CBI]: brief += "CBL " if has[MetaDataStyle.CIX]: brief += "CR " if has[MetaDataStyle.COMET]: brief += "CoMet " brief += "]" print brief if opts.terse: return print if opts.data_style is None or opts.data_style == MetaDataStyle.CIX: if has[MetaDataStyle.CIX]: print("--------- ComicRack tags ---------") if opts.raw: print( u"{0}".format( unicode( ca.readRawCIX(), errors='ignore'))) else: print(u"{0}".format(ca.readCIX())) if opts.data_style is None or opts.data_style == MetaDataStyle.CBI: if has[MetaDataStyle.CBI]: print("------- ComicBookLover tags -------") if opts.raw: pprint(json.loads(ca.readRawCBI())) else: print(u"{0}".format(ca.readCBI())) if opts.data_style is None or opts.data_style == MetaDataStyle.COMET: if has[MetaDataStyle.COMET]: print("----------- CoMet tags -----------") if opts.raw: print(u"{0}".format(ca.readRawCoMet())) else: print(u"{0}".format(ca.readCoMet())) elif opts.delete_tags: style_name = MetaDataStyle.name[opts.data_style] if has[opts.data_style]: if not opts.dryrun: if not ca.removeMetadata(opts.data_style): print(u"{0}: Tag removal seemed to fail!".format(filename)) else: print( u"{0}: Removed {1} tags.".format(filename, style_name)) else: print( u"{0}: dry-run. {1} tags not removed".format(filename, style_name)) else: print(u"{0}: This archive doesn't have {1} tags to remove.".format( filename, style_name)) elif opts.copy_tags: dst_style_name = MetaDataStyle.name[opts.data_style] if opts.no_overwrite and has[opts.data_style]: print(u"{0}: Already has {1} tags. Not overwriting.".format( filename, dst_style_name)) return if opts.copy_source == opts.data_style: print( u"{0}: Destination and source are same: {1}. Nothing to do.".format( filename, dst_style_name)) return src_style_name = MetaDataStyle.name[opts.copy_source] if has[opts.copy_source]: if not opts.dryrun: md = ca.readMetadata(opts.copy_source) if settings.apply_cbl_transform_on_bulk_operation and opts.data_style == MetaDataStyle.CBI: md = CBLTransformer(md, settings).apply() if not ca.writeMetadata(md, opts.data_style): print(u"{0}: Tag copy seemed to fail!".format(filename)) else: print(u"{0}: Copied {1} tags to {2} .".format( filename, src_style_name, dst_style_name)) else: print( u"{0}: dry-run. {1} tags not copied".format(filename, src_style_name)) else: print(u"{0}: This archive doesn't have {1} tags to copy.".format( filename, src_style_name)) elif opts.save_tags: if opts.no_overwrite and has[opts.data_style]: print(u"{0}: Already has {1} tags. Not overwriting.".format( filename, MetaDataStyle.name[opts.data_style])) return if batch_mode: print(u"Processing {0}...".format(filename)) md = create_local_metadata(opts, ca, has[opts.data_style]) if md.issue is None or md.issue == "": if opts.assume_issue_is_one_if_not_set: md.issue = "1" # now, search online if opts.search_online: if opts.issue_id is not None: # we were given the actual ID to search with try: comicVine = ComicVineTalker() comicVine.wait_for_rate_limit = opts.wait_and_retry_on_rate_limit cv_md = comicVine.fetchIssueDataByIssueID( opts.issue_id, settings) except ComicVineTalkerException: print >> sys.stderr, "Network error while getting issue details. Save aborted" match_results.fetchDataFailures.append(filename) return if cv_md is None: print >> sys.stderr, "No match for ID {0} was found.".format( opts.issue_id) match_results.noMatches.append(filename) return if settings.apply_cbl_transform_on_cv_import: cv_md = CBLTransformer(cv_md, settings).apply() else: ii = IssueIdentifier(ca, settings) if md is None or md.isEmpty: print >> sys.stderr, "No metadata given to search online with!" match_results.noMatches.append(filename) return def myoutput(text): if opts.verbose: IssueIdentifier.defaultWriteOutput(text) # use our overlayed MD struct to search ii.setAdditionalMetadata(md) ii.onlyUseAdditionalMetaData = True ii.waitAndRetryOnRateLimit = opts.wait_and_retry_on_rate_limit ii.setOutputFunction(myoutput) ii.cover_page_index = md.getCoverPageIndexList()[0] matches = ii.search() result = ii.search_result found_match = False choices = False low_confidence = False if result == ii.ResultNoMatches: pass elif result == ii.ResultFoundMatchButBadCoverScore: low_confidence = True found_match = True elif result == ii.ResultFoundMatchButNotFirstPage: found_match = True elif result == ii.ResultMultipleMatchesWithBadImageScores: low_confidence = True choices = True elif result == ii.ResultOneGoodMatch: found_match = True elif result == ii.ResultMultipleGoodMatches: choices = True if choices: if low_confidence: print >> sys.stderr, "Online search: Multiple low confidence matches. Save aborted" match_results.lowConfidenceMatches.append( MultipleMatch(filename, matches)) return else: print >> sys.stderr, "Online search: Multiple good matches. Save aborted" match_results.multipleMatches.append( MultipleMatch(filename, matches)) return if low_confidence and opts.abortOnLowConfidence: print >> sys.stderr, "Online search: Low confidence match. Save aborted" match_results.lowConfidenceMatches.append( MultipleMatch(filename, matches)) return if not found_match: print >> sys.stderr, "Online search: No match found. Save aborted" match_results.noMatches.append(filename) return # we got here, so we have a single match # now get the particular issue data cv_md = actual_issue_data_fetch(matches[0], settings, opts) if cv_md is None: match_results.fetchDataFailures.append(filename) return md.overlay(cv_md) # ok, done building our metadata. time to save if not actual_metadata_save(ca, opts, md): match_results.writeFailures.append(filename) else: match_results.goodMatches.append(filename) elif opts.rename_file: msg_hdr = "" if batch_mode: msg_hdr = u"{0}: ".format(filename) if opts.data_style is not None: use_tags = has[opts.data_style] else: use_tags = False md = create_local_metadata(opts, ca, use_tags) if md.series is None: print >> sys.stderr, msg_hdr + "Can't rename without series name" return new_ext = None # default if settings.rename_extension_based_on_archive: if ca.isZip(): new_ext = ".cbz" elif ca.isRar(): new_ext = ".cbr" renamer = FileRenamer(md) renamer.setTemplate(settings.rename_template) renamer.setIssueZeroPadding(settings.rename_issue_number_padding) renamer.setSmartCleanup(settings.rename_use_smart_string_cleanup) new_name = renamer.determineName(filename, ext=new_ext) if new_name == os.path.basename(filename): print >> sys.stderr, msg_hdr + "Filename is already good!" return folder = os.path.dirname(os.path.abspath(filename)) new_abs_path = utils.unique_file(os.path.join(folder, new_name)) suffix = "" if not opts.dryrun: # rename the file os.rename(filename, new_abs_path) else: suffix = " (dry-run, no change)" print( u"renamed '{0}' -> '{1}' {2}".format(os.path.basename(filename), new_name, suffix)) elif opts.export_to_zip: msg_hdr = "" if batch_mode: msg_hdr = u"{0}: ".format(filename) if not ca.isRar(): print >> sys.stderr, msg_hdr + "Archive is not a RAR." return rar_file = os.path.abspath(os.path.abspath(filename)) new_file = os.path.splitext(rar_file)[0] + ".cbz" if opts.abort_export_on_conflict and os.path.lexists(new_file): print msg_hdr + "{0} already exists in the that folder.".format(os.path.split(new_file)[1]) return new_file = utils.unique_file(os.path.join(new_file)) delete_success = False export_success = False if not opts.dryrun: if ca.exportAsZip(new_file): export_success = True if opts.delete_rar_after_export: try: os.unlink(rar_file) except: print >> sys.stderr, msg_hdr + \ "Error deleting original RAR after export" delete_success = False else: delete_success = True else: # last export failed, so remove the zip, if it exists if os.path.lexists(new_file): os.remove(new_file) else: msg = msg_hdr + \ u"Dry-run: Would try to create {0}".format( os.path.split(new_file)[1]) if opts.delete_rar_after_export: msg += u" and delete orginal." print(msg) return msg = msg_hdr if export_success: msg += u"Archive exported successfully to: {0}".format( os.path.split(new_file)[1]) if opts.delete_rar_after_export and delete_success: msg += u" (Original deleted) " else: msg += u"Archive failed to export!" print(msg)
def process_file_cli(filename, opts, settings, match_results): batch_mode = len(opts.file_list) > 1 ca = ComicArchive(filename, settings.rar_exe_path, ComicTaggerSettings.getGraphic('nocover.png')) if not os.path.lexists(filename): print >> sys.stderr, "Cannot find " + filename return if not ca.seemsToBeAComicArchive(): print >> sys.stderr, "Sorry, but " + \ filename + " is not a comic archive!" return # if not ca.isWritableForStyle(opts.data_style) and (opts.delete_tags or # opts.save_tags or opts.rename_file): if not ca.isWritable() and (opts.delete_tags or opts.copy_tags or opts.save_tags or opts.rename_file): print >> sys.stderr, "This archive is not writable for that tag type" return has = [False, False, False] if ca.hasCIX(): has[MetaDataStyle.CIX] = True if ca.hasCBI(): has[MetaDataStyle.CBI] = True if ca.hasCoMet(): has[MetaDataStyle.COMET] = True if opts.print_tags: if opts.data_style is None: page_count = ca.getNumberOfPages() brief = "" if batch_mode: brief = u"{0}: ".format(filename) if ca.isZip(): brief += "ZIP archive " elif ca.isRar(): brief += "RAR archive " elif ca.isFolder(): brief += "Folder archive " brief += "({0: >3} pages)".format(page_count) brief += " tags:[ " if not (has[MetaDataStyle.CBI] or has[MetaDataStyle.CIX] or has[MetaDataStyle.COMET]): brief += "none " else: if has[MetaDataStyle.CBI]: brief += "CBL " if has[MetaDataStyle.CIX]: brief += "CR " if has[MetaDataStyle.COMET]: brief += "CoMet " brief += "]" print brief if opts.terse: return print if opts.data_style is None or opts.data_style == MetaDataStyle.CIX: if has[MetaDataStyle.CIX]: print("--------- ComicRack tags ---------") if opts.raw: print(u"{0}".format( unicode(ca.readRawCIX(), errors='ignore'))) else: print(u"{0}".format(ca.readCIX())) if opts.data_style is None or opts.data_style == MetaDataStyle.CBI: if has[MetaDataStyle.CBI]: print("------- ComicBookLover tags -------") if opts.raw: pprint(json.loads(ca.readRawCBI())) else: print(u"{0}".format(ca.readCBI())) if opts.data_style is None or opts.data_style == MetaDataStyle.COMET: if has[MetaDataStyle.COMET]: print("----------- CoMet tags -----------") if opts.raw: print(u"{0}".format(ca.readRawCoMet())) else: print(u"{0}".format(ca.readCoMet())) elif opts.delete_tags: style_name = MetaDataStyle.name[opts.data_style] if has[opts.data_style]: if not opts.dryrun: if not ca.removeMetadata(opts.data_style): print(u"{0}: Tag removal seemed to fail!".format(filename)) else: print(u"{0}: Removed {1} tags.".format( filename, style_name)) else: print(u"{0}: dry-run. {1} tags not removed".format( filename, style_name)) else: print(u"{0}: This archive doesn't have {1} tags to remove.".format( filename, style_name)) elif opts.copy_tags: dst_style_name = MetaDataStyle.name[opts.data_style] if opts.no_overwrite and has[opts.data_style]: print(u"{0}: Already has {1} tags. Not overwriting.".format( filename, dst_style_name)) return if opts.copy_source == opts.data_style: print(u"{0}: Destination and source are same: {1}. Nothing to do.". format(filename, dst_style_name)) return src_style_name = MetaDataStyle.name[opts.copy_source] if has[opts.copy_source]: if not opts.dryrun: md = ca.readMetadata(opts.copy_source) if settings.apply_cbl_transform_on_bulk_operation and opts.data_style == MetaDataStyle.CBI: md = CBLTransformer(md, settings).apply() if not ca.writeMetadata(md, opts.data_style): print(u"{0}: Tag copy seemed to fail!".format(filename)) else: print(u"{0}: Copied {1} tags to {2} .".format( filename, src_style_name, dst_style_name)) else: print(u"{0}: dry-run. {1} tags not copied".format( filename, src_style_name)) else: print(u"{0}: This archive doesn't have {1} tags to copy.".format( filename, src_style_name)) elif opts.save_tags: if opts.no_overwrite and has[opts.data_style]: print(u"{0}: Already has {1} tags. Not overwriting.".format( filename, MetaDataStyle.name[opts.data_style])) return if batch_mode: print(u"Processing {0}...".format(filename)) md = create_local_metadata(opts, ca, has[opts.data_style]) if md.issue is None or md.issue == "": if opts.assume_issue_is_one_if_not_set: md.issue = "1" # now, search online if opts.search_online: if opts.issue_id is not None: # we were given the actual ID to search with try: comicVine = ComicVineTalker() comicVine.wait_for_rate_limit = opts.wait_and_retry_on_rate_limit cv_md = comicVine.fetchIssueDataByIssueID( opts.issue_id, settings) except ComicVineTalkerException: print >> sys.stderr, "Network error while getting issue details. Save aborted" match_results.fetchDataFailures.append(filename) return if cv_md is None: print >> sys.stderr, "No match for ID {0} was found.".format( opts.issue_id) match_results.noMatches.append(filename) return if settings.apply_cbl_transform_on_cv_import: cv_md = CBLTransformer(cv_md, settings).apply() else: ii = IssueIdentifier(ca, settings) if md is None or md.isEmpty: print >> sys.stderr, "No metadata given to search online with!" match_results.noMatches.append(filename) return def myoutput(text): if opts.verbose: IssueIdentifier.defaultWriteOutput(text) # use our overlayed MD struct to search ii.setAdditionalMetadata(md) ii.onlyUseAdditionalMetaData = True ii.waitAndRetryOnRateLimit = opts.wait_and_retry_on_rate_limit ii.setOutputFunction(myoutput) ii.cover_page_index = md.getCoverPageIndexList()[0] matches = ii.search() result = ii.search_result found_match = False choices = False low_confidence = False if result == ii.ResultNoMatches: pass elif result == ii.ResultFoundMatchButBadCoverScore: low_confidence = True found_match = True elif result == ii.ResultFoundMatchButNotFirstPage: found_match = True elif result == ii.ResultMultipleMatchesWithBadImageScores: low_confidence = True choices = True elif result == ii.ResultOneGoodMatch: found_match = True elif result == ii.ResultMultipleGoodMatches: choices = True if choices: if low_confidence: print >> sys.stderr, "Online search: Multiple low confidence matches. Save aborted" match_results.lowConfidenceMatches.append( MultipleMatch(filename, matches)) return else: print >> sys.stderr, "Online search: Multiple good matches. Save aborted" match_results.multipleMatches.append( MultipleMatch(filename, matches)) return if low_confidence and opts.abortOnLowConfidence: print >> sys.stderr, "Online search: Low confidence match. Save aborted" match_results.lowConfidenceMatches.append( MultipleMatch(filename, matches)) return if not found_match: print >> sys.stderr, "Online search: No match found. Save aborted" match_results.noMatches.append(filename) return # we got here, so we have a single match # now get the particular issue data cv_md = actual_issue_data_fetch(matches[0], settings, opts) if cv_md is None: match_results.fetchDataFailures.append(filename) return md.overlay(cv_md) # ok, done building our metadata. time to save if not actual_metadata_save(ca, opts, md): match_results.writeFailures.append(filename) else: match_results.goodMatches.append(filename) elif opts.rename_file: msg_hdr = "" if batch_mode: msg_hdr = u"{0}: ".format(filename) if opts.data_style is not None: use_tags = has[opts.data_style] else: use_tags = False md = create_local_metadata(opts, ca, use_tags) if md.series is None: print >> sys.stderr, msg_hdr + "Can't rename without series name" return new_ext = None # default if settings.rename_extension_based_on_archive: if ca.isZip(): new_ext = ".cbz" elif ca.isRar(): new_ext = ".cbr" renamer = FileRenamer(md) renamer.setTemplate(settings.rename_template) renamer.setIssueZeroPadding(settings.rename_issue_number_padding) renamer.setSmartCleanup(settings.rename_use_smart_string_cleanup) new_name = renamer.determineName(filename, ext=new_ext) if new_name == os.path.basename(filename): print >> sys.stderr, msg_hdr + "Filename is already good!" return folder = os.path.dirname(os.path.abspath(filename)) new_abs_path = utils.unique_file(os.path.join(folder, new_name)) suffix = "" if not opts.dryrun: # rename the file os.rename(filename, new_abs_path) else: suffix = " (dry-run, no change)" print(u"renamed '{0}' -> '{1}' {2}".format(os.path.basename(filename), new_name, suffix)) elif opts.export_to_zip: msg_hdr = "" if batch_mode: msg_hdr = u"{0}: ".format(filename) if not ca.isRar(): print >> sys.stderr, msg_hdr + "Archive is not a RAR." return rar_file = os.path.abspath(os.path.abspath(filename)) new_file = os.path.splitext(rar_file)[0] + ".cbz" if opts.abort_export_on_conflict and os.path.lexists(new_file): print msg_hdr + "{0} already exists in the that folder.".format( os.path.split(new_file)[1]) return new_file = utils.unique_file(os.path.join(new_file)) delete_success = False export_success = False if not opts.dryrun: if ca.exportAsZip(new_file): export_success = True if opts.delete_rar_after_export: try: os.unlink(rar_file) except: print >> sys.stderr, msg_hdr + \ "Error deleting original RAR after export" delete_success = False else: delete_success = True else: # last export failed, so remove the zip, if it exists if os.path.lexists(new_file): os.remove(new_file) else: msg = msg_hdr + \ u"Dry-run: Would try to create {0}".format( os.path.split(new_file)[1]) if opts.delete_rar_after_export: msg += u" and delete orginal." print(msg) return msg = msg_hdr if export_success: msg += u"Archive exported successfully to: {0}".format( os.path.split(new_file)[1]) if opts.delete_rar_after_export and delete_success: msg += u" (Original deleted) " else: msg += u"Archive failed to export!" print(msg)
def loadDefault( self ): self.current_pixmap = QPixmap(ComicTaggerSettings.getGraphic('nocover.png')) #print "loadDefault called" self.setDisplayPixmap( 0, 0)
def loadDefault(self): self.current_pixmap = QPixmap( ComicTaggerSettings.getGraphic('nocover.png')) #print("loadDefault called") self.setDisplayPixmap(0, 0)