inkstitch/inkstitch.py

# 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 cProfile
import pstats
import logging
import os
import sys
import traceback
from argparse import ArgumentParser
from io import StringIO

if getattr(sys, 'frozen', None) is None:
    # When running in development mode, we want to use the inkex installed by
    # pip install, not the one bundled with Inkscape which is not new enough.
    if sys.platform == "darwin":
        extensions_path = "/Applications/Inkscape.app/Contents/Resources/share/inkscape/extensions"
    else:
        extensions_path = "/usr/share/inkscape/extensions"

    sys.path.remove(extensions_path)
    sys.path.append(extensions_path)

from inkex import errormsg
from lxml.etree import XMLSyntaxError

import lib.debug as debug
from lib import extensions
from lib.i18n import _
from lib.utils import restore_stderr, save_stderr, version

# ignore warnings in releases
if getattr(sys, 'frozen', None):
    import warnings
    warnings.filterwarnings('ignore')

logger = logging.getLogger('shapely.geos')
logger.setLevel(logging.DEBUG)
shapely_errors = StringIO()
ch = logging.StreamHandler(shapely_errors)
ch.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(name)s - %(levelname)s - %(message)s')
ch.setFormatter(formatter)
logger.addHandler(ch)

parser = ArgumentParser()
parser.add_argument("--extension")
my_args, remaining_args = parser.parse_known_args()

if os.path.exists(os.path.join(os.path.dirname(os.path.realpath(__file__)), "DEBUG")):
    debug.enable()

profiler = None
if os.path.exists(os.path.join(os.path.dirname(os.path.realpath(__file__)), "PROFILE")):
    profiler = cProfile.Profile()
    profiler.enable()

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()

if (hasattr(sys, 'gettrace') and sys.gettrace()) or profiler is not None:
    extension.run(args=remaining_args)
    if profiler:
        path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "profile_stats")
        profiler.disable()
        profiler.dump_stats(path + ".prof")

        with open(path, 'w') as stats_file:
            stats = pstats.Stats(profiler, stream=stats_file)
            stats.sort_stats(pstats.SortKey.CUMULATIVE)
            stats.print_stats()

        print(f"profiling stats written to {path} and {path}.prof", file=sys.stderr)
else:
    save_stderr()
    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 Exception:
        exception = traceback.format_exc()
    finally:
        restore_stderr()

        if shapely_errors.tell():
            errormsg(shapely_errors.getvalue())

    if exception:
        errormsg(_("Ink/Stitch experienced an unexpected error. This means it is a bug in Ink/Stitch.") + "\n")
        errormsg(_("If you'd like to help please\n"
                   "- copy the entire error message below\n"
                   "- save your SVG file and\n"
                   "- create a new issue at https://github.com/inkstitch/inkstitch/issues") + "\n")
        errormsg(_("Include the error description and also (if possible) "
                   "the svg file.") + "\n")
        errormsg(version.get_inkstitch_version() + "\n")
        errormsg(exception)
        sys.exit(1)
    else:
        sys.exit(0)