Mercurial > hg
view mercurial/utils/resourceutil.py @ 44152:a90039b07343 stable
resourceutil: ensure `_rootpath` is defined under py2exe
Can't even run `hg version` without this.
Differential Revision: https://phab.mercurial-scm.org/D7980
author | Matt Harbison <matt_harbison@yahoo.com> |
---|---|
date | Fri, 24 Jan 2020 01:11:19 -0500 |
parents | bba9149adc14 |
children | 9e367157a990 |
line wrap: on
line source
# 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) _rootpath = os.path.dirname(datapath) else: datapath = os.path.dirname(os.path.dirname(pycompat.fsencode(__file__))) _rootpath = os.path.dirname(datapath) try: from importlib import resources from .. import encoding # Force loading of the resources module resources.open_binary # pytype: disable=module-attr def open_resource(package, name): return resources.open_binary( # pytype: disable=module-attr pycompat.sysstr(package), pycompat.sysstr(name) ) def is_resource(package, name): return resources.is_resource( pycompat.sysstr(package), encoding.strfromlocal(name) ) def contents(package): for r in resources.contents(pycompat.sysstr(package)): yield encoding.strtolocal(r) except (ImportError, AttributeError): def _package_path(package): return os.path.join(_rootpath, *package.split(b'.')) def open_resource(package, name): path = os.path.join(_package_path(package), name) return open(path, 'rb') def is_resource(package, name): path = os.path.join(_package_path(package), name) try: return os.path.isfile(pycompat.fsdecode(path)) except (IOError, OSError): return False def contents(package): path = pycompat.fsdecode(_package_path(package)) for p in os.listdir(path): yield pycompat.fsencode(p)