"""A python app to (automatically) tag comic archives""" # 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 logging import logging.handlers import os import pathlib import platform import signal import sys import traceback import pkg_resources from comictaggerlib import cli from comictaggerlib.comicvinetalker import ComicVineTalker from comictaggerlib.ctversion import version from comictaggerlib.options import Options from comictaggerlib.settings import ComicTaggerSettings logger = logging.getLogger("comictagger") logging.getLogger("comicapi").setLevel(logging.DEBUG) logger.setLevel(logging.DEBUG) try: qt_available = True from PyQt5 import QtCore, QtGui, QtWidgets def show_exception_box(log_msg): """Checks if a QApplication instance is available and shows a messagebox with the exception message. If unavailable (non-console application), log an additional notice. """ if QtWidgets.QApplication.instance() is not None: errorbox = QtWidgets.QMessageBox() errorbox.setText("Oops. An unexpected error occured:\n{0}".format(log_msg)) errorbox.exec_() QtWidgets.QApplication.exit(1) else: logger.debug("No QApplication instance available.") class UncaughtHook(QtCore.QObject): _exception_caught = QtCore.pyqtSignal(object) def __init__(self, *args, **kwargs): super(UncaughtHook, self).__init__(*args, **kwargs) # this registers the exception_hook() function as hook with the Python interpreter sys.excepthook = self.exception_hook # connect signal to execute the message box function always on main thread self._exception_caught.connect(show_exception_box) def exception_hook(self, exc_type, exc_value, exc_traceback): """Function handling uncaught exceptions. It is triggered each time an uncaught exception occurs. """ if issubclass(exc_type, KeyboardInterrupt): # ignore keyboard interrupt to support console applications sys.__excepthook__(exc_type, exc_value, exc_traceback) else: exc_info = (exc_type, exc_value, exc_traceback) log_msg = "\n".join( ["".join(traceback.format_tb(exc_traceback)), "{0}: {1}".format(exc_type.__name__, exc_value)] ) logger.critical( "Uncaught exception: %s", "{0}: {1}".format(exc_type.__name__, exc_value), exc_info=exc_info ) # trigger message box show self._exception_caught.emit(log_msg) qt_exception_hook = UncaughtHook() from comictaggerlib.taggerwindow import TaggerWindow except ImportError as e: logging.debug(e) qt_available = False def rotate(handler: logging.handlers.RotatingFileHandler, filename: pathlib.Path): if filename.is_file() and filename.stat().st_size > 0: handler.doRollover() def ctmain(): os.makedirs(ComicTaggerSettings.get_settings_folder() / "logs", exist_ok=True) stream_handler = logging.StreamHandler() stream_handler.setLevel(logging.WARNING) file_handler = logging.handlers.RotatingFileHandler( ComicTaggerSettings.get_settings_folder() / "logs" / "ComicTagger.log", encoding="utf-8", backupCount=10 ) rotate(file_handler, ComicTaggerSettings.get_settings_folder() / "logs" / "ComicTagger.log") logging.basicConfig( handlers=[ stream_handler, file_handler, ], level=logging.WARNING, format="%(asctime)s | %(name)s | %(levelname)s | %(message)s", datefmt="%Y-%m-%dT%H:%M:%S", ) opts = Options() opts.parse_cmd_line_args() # Need to load setting before anything else SETTINGS = ComicTaggerSettings() # 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) logger.info( "ComicTagger Version: %s running on: %s PyInstaller: %s", version, platform.system(), "Yes" if getattr(sys, "frozen", None) else "No", ) logger.debug("Installed Packages") for pkg in sorted(pkg_resources.working_set, key=lambda x: x.project_name): logger.debug("%s\t%s", pkg.project_name, pkg.version) if not qt_available and not opts.no_gui: opts.no_gui = True print("PyQt5 is not available. ComicTagger is limited to command-line mode.") logger.info("PyQt5 is not available. ComicTagger is limited to command-line mode.") if opts.no_gui: try: cli.cli_mode(opts, SETTINGS) except: logger.exception("CLI mode failed") else: os.environ["QtWidgets.QT_AUTO_SCREEN_SCALE_FACTOR"] = "1" args = [] if opts.darkmode: args.extend(["-platform", "windows:darkmode=2"]) args.extend(sys.argv) app = QtWidgets.QApplication(args) if platform.system() == "Darwin": # Set the MacOS dock icon app.setWindowIcon(QtGui.QIcon(ComicTaggerSettings.get_graphic("app.png"))) if platform.system() == "Windows": # For pure python, tell windows that we're not python, # so we can have our own taskbar icon import ctypes myappid = "comictagger" # arbitrary string ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID(myappid) # force close of console window swp_hidewindow = 0x0080 console_wnd = ctypes.windll.kernel32.GetConsoleWindow() if console_wnd != 0: ctypes.windll.user32.SetWindowPos(console_wnd, None, 0, 0, 0, 0, swp_hidewindow) if platform.system() != "Linux": img = QtGui.QPixmap(ComicTaggerSettings.get_graphic("tags.png")) splash = QtWidgets.QSplashScreen(img) splash.show() splash.raise_() app.processEvents() try: tagger_window = TaggerWindow(opts.file_list, SETTINGS, opts=opts) tagger_window.setWindowIcon(QtGui.QIcon(ComicTaggerSettings.get_graphic("app.png"))) tagger_window.show() if platform.system() != "Linux": splash.finish(tagger_window) sys.exit(app.exec()) except Exception: logger.exception("GUI mode failed") QtWidgets.QMessageBox.critical( QtWidgets.QMainWindow(), "Error", "Unhandled exception in app:\n" + traceback.format_exc() )