kopia lustrzana https://github.com/inkstitch/inkstitch
180 wiersze
7.7 KiB
Python
180 wiersze
7.7 KiB
Python
# Authors: see git history
|
|
#
|
|
# Copyright (c) 2010 Authors
|
|
# Licensed under the GNU GPL version 3.0 or later. See the file LICENSE for details.
|
|
|
|
import os
|
|
import sys
|
|
from pathlib import Path # to work with paths as objects
|
|
from argparse import ArgumentParser # to parse arguments and remove --extension
|
|
|
|
if sys.version_info >= (3, 11):
|
|
import tomllib # built-in in Python 3.11+
|
|
else:
|
|
import tomli as tomllib
|
|
|
|
import logging
|
|
|
|
import lib.debug.utils as debug_utils
|
|
import lib.debug.logging as debug_logging
|
|
from lib.debug.utils import safe_get # mimic get method of dict with default value
|
|
|
|
# --------------------------------------------------------------------------------------------
|
|
|
|
SCRIPTDIR = Path(__file__).parent.absolute()
|
|
|
|
logger = logging.getLogger("inkstitch") # create module logger with name 'inkstitch'
|
|
|
|
# TODO --- temporary --- catch old DEBUG.ini file and inform user to reformat it to DEBUG.toml
|
|
old_debug_ini = SCRIPTDIR / "DEBUG.ini"
|
|
if old_debug_ini.exists():
|
|
print("ERROR: old DEBUG.ini exists, please reformat it to DEBUG.toml and remove DEBUG.ini file", file=sys.stderr)
|
|
exit(1)
|
|
# --- end of temporary ---
|
|
|
|
debug_toml = SCRIPTDIR / "DEBUG.toml"
|
|
if debug_toml.exists():
|
|
with debug_toml.open("rb") as f:
|
|
ini = tomllib.load(f) # read DEBUG.toml file if exists, otherwise use default values in ini object
|
|
else:
|
|
ini = {}
|
|
# --------------------------------------------------------------------------------------------
|
|
|
|
running_as_frozen = getattr(sys, 'frozen', None) is not None # check if running from pyinstaller bundle
|
|
|
|
if not running_as_frozen: # override running_as_frozen from DEBUG.toml - for testing
|
|
if safe_get(ini, "DEBUG", "force_frozen", default=False):
|
|
running_as_frozen = True
|
|
|
|
if len(sys.argv) < 2:
|
|
# no arguments - prevent accidentally running this script
|
|
msg = "No arguments given, exiting!" # without gettext localization see _()
|
|
if running_as_frozen: # we show dialog only when running from pyinstaller bundle - using wx
|
|
try:
|
|
import wx
|
|
app = wx.App()
|
|
dlg = wx.MessageDialog(None, msg, "Inkstitch", wx.OK | wx.ICON_ERROR)
|
|
dlg.ShowModal()
|
|
dlg.Destroy()
|
|
except ImportError:
|
|
print(msg, file=sys.stderr)
|
|
else:
|
|
print(msg, file=sys.stderr)
|
|
exit(1)
|
|
|
|
# activate logging - must be done before any logging is done
|
|
debug_logging.activate_logging(running_as_frozen, ini, SCRIPTDIR)
|
|
# --------------------------------------------------------------------------------------------
|
|
|
|
# check if running from inkscape, given by environment variable
|
|
if os.environ.get('INKSTITCH_OFFLINE_SCRIPT', '').lower() in ['true', '1', 'yes', 'y']:
|
|
running_from_inkscape = False
|
|
else:
|
|
running_from_inkscape = True
|
|
|
|
# initialize debug and profiler type
|
|
debug_active = bool((gettrace := getattr(sys, 'gettrace')) and gettrace()) # check if debugger is active on startup
|
|
debug_type = 'none'
|
|
profiler_type = 'none'
|
|
|
|
if not running_as_frozen: # debugging/profiling only in development mode
|
|
# specify debugger type
|
|
# but if script was already started from debugger then don't read debug type from ini file or cmd line
|
|
if not debug_active:
|
|
debug_type = debug_utils.resolve_debug_type(ini) # read debug type from ini file or cmd line
|
|
|
|
profiler_type = debug_utils.resolve_profiler_type(ini) # read profile type from ini file or cmd line
|
|
|
|
if running_from_inkscape:
|
|
# process creation of the Bash script - should be done before sys.path is modified, see below in prefere_pip_inkex
|
|
if safe_get(ini, "DEBUG", "create_bash_script", default=False): # create script only if enabled in DEBUG.toml
|
|
debug_utils.write_offline_debug_script(SCRIPTDIR, ini)
|
|
|
|
# disable debugger when running from inkscape
|
|
disable_from_inkscape = safe_get(ini, "DEBUG", "disable_from_inkscape", default=False)
|
|
if disable_from_inkscape:
|
|
debug_type = 'none' # do not start debugger when running from inkscape
|
|
|
|
# prefer pip installed inkex over inkscape bundled inkex, pip version is bundled with Inkstitch
|
|
# - must be be done before importing inkex
|
|
prefere_pip_inkex = safe_get(ini, "LIBRARY", "prefer_pip_inkex", default=True)
|
|
if prefere_pip_inkex and 'PYTHONPATH' in os.environ:
|
|
debug_utils.reorder_sys_path()
|
|
|
|
# enabling of debug depends on value of debug_type in DEBUG.toml file
|
|
if debug_type != 'none':
|
|
from lib.debug.debugger import init_debugger
|
|
init_debugger(debug_type, ini)
|
|
# check if debugger is really activated
|
|
debug_active = bool((gettrace := getattr(sys, 'gettrace')) and gettrace())
|
|
|
|
# activate logging for svg
|
|
# we need to import only after possible modification of sys.path, we disable here flake8 E402
|
|
from lib.debug.debug import debug # noqa: E402 # import global variable debug - don't import whole module
|
|
debug.enable() # perhaps it would be better to find a more relevant name; in fact, it's about logging and svg creation.
|
|
|
|
# log startup info
|
|
debug_logging.startup_info(logger, SCRIPTDIR, running_as_frozen, running_from_inkscape, debug_active, debug_type, profiler_type)
|
|
|
|
# --------------------------------------------------------------------------------------------
|
|
|
|
# pop '--extension' from arguments and generate extension class name from extension name
|
|
# example: --extension=params will instantiate Params() class from lib.extensions.
|
|
|
|
# we need to import only after possible modification of sys.path, we disable here flake8 E402
|
|
from lib import extensions # noqa: E402 # import all supported extensions of institch
|
|
|
|
# TODO: if we run this earlier the warnings ignore filter for releases will not work properly
|
|
if running_as_frozen and not debug_logging.frozen_debug_active():
|
|
debug_logging.disable_warnings()
|
|
|
|
parser = ArgumentParser()
|
|
parser.add_argument("--extension")
|
|
my_args, remaining_args = parser.parse_known_args()
|
|
|
|
extension_name = my_args.extension
|
|
|
|
# example: foo_bar_baz -> FooBarBaz
|
|
extension_class_name = extension_name.title().replace("_", "")
|
|
|
|
extension_class = getattr(extensions, extension_class_name)
|
|
extension = extension_class() # create instance of extension class - call __init__ method
|
|
|
|
# extension run(), we differentiate between debug and normal mode
|
|
# - in debug or profile mode we debug or profile extension.run() method
|
|
# - in normal mode we run extension.run() in try/except block to catch all exceptions and hide GTK spam
|
|
if debug_active or profiler_type != "none": # if debug or profile mode
|
|
if profiler_type == 'none': # only debugging
|
|
extension.run(args=remaining_args)
|
|
else: # do profiling
|
|
debug_utils.profile(profiler_type, SCRIPTDIR, ini, extension, remaining_args)
|
|
|
|
else: # if not debug nor profile mode
|
|
from lib.exceptions import InkstitchException, format_uncaught_exception
|
|
from inkex import errormsg # to show error message in inkscape
|
|
from lxml.etree import XMLSyntaxError # to catch XMLSyntaxError from inkex
|
|
from lib.i18n import _ # see gettext translation function _()
|
|
from lib.utils import restore_stderr, save_stderr # to hide GTK spam
|
|
|
|
save_stderr() # hide GTK spam
|
|
exception = None
|
|
try:
|
|
extension.run(args=remaining_args)
|
|
except (SystemExit, KeyboardInterrupt):
|
|
raise
|
|
except XMLSyntaxError:
|
|
msg = _("Ink/Stitch cannot read your SVG file. "
|
|
"This is often the case when you use a file which has been created with Adobe Illustrator.")
|
|
msg += "\n\n"
|
|
msg += _("Try to import the file into Inkscape through 'File > Import...' (Ctrl+I)")
|
|
errormsg(msg)
|
|
except InkstitchException as exc:
|
|
errormsg(str(exc))
|
|
except Exception:
|
|
errormsg(format_uncaught_exception())
|
|
sys.exit(1)
|
|
finally:
|
|
restore_stderr()
|
|
|
|
sys.exit(0)
|