218 lines
8.1 KiB
Python
Executable File
218 lines
8.1 KiB
Python
Executable File
"""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.
|
|
from __future__ import annotations
|
|
|
|
import importlib.metadata
|
|
import json
|
|
import logging.handlers
|
|
import os
|
|
import pathlib
|
|
import platform
|
|
import signal
|
|
import sys
|
|
import traceback
|
|
import types
|
|
|
|
from comicapi import utils
|
|
from comictaggerlib import cli
|
|
from comictaggerlib.comicvinetalker import ComicVineTalker
|
|
from comictaggerlib.ctversion import version
|
|
from comictaggerlib.options import parse_cmd_line
|
|
from comictaggerlib.settings import ComicTaggerSettings
|
|
|
|
logger = logging.getLogger("comictagger")
|
|
logging.getLogger("comicapi").setLevel(logging.DEBUG)
|
|
logging.getLogger("comictaggerlib").setLevel(logging.DEBUG)
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
try:
|
|
qt_available = True
|
|
from PyQt5 import QtCore, QtGui, QtWidgets
|
|
|
|
def show_exception_box(log_msg: str) -> None:
|
|
"""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(f"Oops. An unexpected error occurred:\n{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) -> None:
|
|
super().__init__()
|
|
|
|
# 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: type[BaseException], exc_value: BaseException, exc_traceback: types.TracebackType | None
|
|
) -> None:
|
|
"""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)), f"{exc_type.__name__}: {exc_value}"])
|
|
logger.critical("Uncaught exception: %s: %s", 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:
|
|
|
|
def show_exception_box(log_msg: str) -> None:
|
|
pass
|
|
|
|
logger.error(str(e))
|
|
qt_available = False
|
|
|
|
|
|
def rotate(handler: logging.handlers.RotatingFileHandler, filename: pathlib.Path) -> None:
|
|
if filename.is_file() and filename.stat().st_size > 0:
|
|
handler.doRollover()
|
|
|
|
|
|
def update_publishers() -> None:
|
|
json_file = ComicTaggerSettings.get_settings_folder() / "publishers.json"
|
|
if json_file.exists():
|
|
try:
|
|
utils.update_publishers(json.loads(json_file.read_text("utf-8")))
|
|
except Exception as e:
|
|
logger.exception("Failed to load publishers from %s", json_file)
|
|
show_exception_box(str(e))
|
|
|
|
|
|
def ctmain() -> None:
|
|
opts = parse_cmd_line()
|
|
SETTINGS = ComicTaggerSettings(opts.config_path)
|
|
|
|
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",
|
|
)
|
|
# Need to load setting before anything else
|
|
|
|
# manage the CV API key
|
|
# None comparison is used so that the empty string can unset the value
|
|
if opts.cv_api_key is not None or opts.cv_url is not None:
|
|
SETTINGS.cv_api_key = opts.cv_api_key if opts.cv_api_key is not None else SETTINGS.cv_api_key
|
|
SETTINGS.cv_url = opts.cv_url if opts.cv_url is not None else SETTINGS.cv_url
|
|
SETTINGS.save()
|
|
if opts.only_set_cv_key:
|
|
print("Key set") # noqa: T201
|
|
return
|
|
|
|
ComicVineTalker.api_key = SETTINGS.cv_api_key
|
|
ComicVineTalker.api_base_url = SETTINGS.cv_url
|
|
|
|
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(importlib.metadata.distributions(), key=lambda x: x.name):
|
|
logger.debug("%s\t%s", pkg.name, pkg.version)
|
|
|
|
utils.load_publishers()
|
|
update_publishers()
|
|
|
|
if not qt_available and not opts.no_gui:
|
|
opts.no_gui = True
|
|
logger.warn("PyQt5 is not available. ComicTagger is limited to command-line mode.")
|
|
|
|
if opts.no_gui:
|
|
try:
|
|
cli.cli_mode(opts, SETTINGS)
|
|
except Exception:
|
|
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) # type: ignore[attr-defined]
|
|
# force close of console window
|
|
swp_hidewindow = 0x0080
|
|
console_wnd = ctypes.windll.kernel32.GetConsoleWindow() # type: ignore[attr-defined]
|
|
if console_wnd != 0:
|
|
ctypes.windll.user32.SetWindowPos(console_wnd, None, 0, 0, 0, 0, swp_hidewindow) # type: ignore[attr-defined]
|
|
|
|
if platform.system() != "Linux":
|
|
img = QtGui.QPixmap(ComicTaggerSettings.get_graphic("tags.png"))
|
|
|
|
splash = QtWidgets.QSplashScreen(img)
|
|
splash.show()
|
|
splash.raise_()
|
|
QtWidgets.QApplication.processEvents()
|
|
|
|
try:
|
|
tagger_window = TaggerWindow(opts.files, 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()
|
|
)
|