mercurial/utils/resourceutil.py
author Matt Harbison <matt_harbison@yahoo.com>
Tue, 17 Dec 2019 23:33:35 -0500
changeset 43943 0671f0a19d93
parent 43905 4d59cc8bda65
child 43955 75ec46c63657
permissions -rw-r--r--
patch: fix header.__repr__() to not have `b''` prefixes in file names Spotted by Denis Laxalde. Differential Revision: https://phab.mercurial-scm.org/D7695

# resourceutil.py - utility for looking up resources
#
#  Copyright 2005 K. Thananchayan <thananck@yahoo.com>
#  Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
#  Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.

from __future__ import absolute_import

import imp
import os
import sys

from .. import pycompat


def mainfrozen():
    """return True if we are a frozen executable.

    The code supports py2exe (most common, Windows only) and tools/freeze
    (portable, not much used).
    """
    return (
        pycompat.safehasattr(sys, "frozen")
        or pycompat.safehasattr(sys, "importers")  # new py2exe
        or imp.is_frozen("__main__")  # old py2exe
    )  # tools/freeze


# the location of data files matching the source code
if mainfrozen() and getattr(sys, 'frozen', None) != 'macosx_app':
    # executable version (py2exe) doesn't support __file__
    datapath = os.path.dirname(pycompat.sysexecutable)
else:
    datapath = os.path.dirname(os.path.dirname(pycompat.fsencode(__file__)))

try:
    import importlib

    # Force loading of the resources module
    importlib.resources.open_binary  # pytype: disable=module-attr

    def open_resource(package, name):
        package = b'mercurial.' + package
        return importlib.resources.open_binary(  # pytype: disable=module-attr
            pycompat.sysstr(package), pycompat.sysstr(name)
        )


except AttributeError:

    def _package_path(package):
        return os.path.join(datapath, *package.split(b'.'))

    def open_resource(package, name):
        path = os.path.join(_package_path(package), name)
        return open(path, 'rb')