##// END OF EJS Templates
resourceutil: ensure `_rootpath` is defined under py2exe...
Matt Harbison -
r44613:a90039b0 stable
parent child Browse files
Show More
@@ -1,84 +1,85 b''
1 # resourceutil.py - utility for looking up resources
1 # resourceutil.py - utility for looking up resources
2 #
2 #
3 # Copyright 2005 K. Thananchayan <thananck@yahoo.com>
3 # Copyright 2005 K. Thananchayan <thananck@yahoo.com>
4 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
5 # Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
5 # Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
6 #
6 #
7 # This software may be used and distributed according to the terms of the
7 # This software may be used and distributed according to the terms of the
8 # GNU General Public License version 2 or any later version.
8 # GNU General Public License version 2 or any later version.
9
9
10 from __future__ import absolute_import
10 from __future__ import absolute_import
11
11
12 import imp
12 import imp
13 import os
13 import os
14 import sys
14 import sys
15
15
16 from .. import pycompat
16 from .. import pycompat
17
17
18
18
19 def mainfrozen():
19 def mainfrozen():
20 """return True if we are a frozen executable.
20 """return True if we are a frozen executable.
21
21
22 The code supports py2exe (most common, Windows only) and tools/freeze
22 The code supports py2exe (most common, Windows only) and tools/freeze
23 (portable, not much used).
23 (portable, not much used).
24 """
24 """
25 return (
25 return (
26 pycompat.safehasattr(sys, "frozen")
26 pycompat.safehasattr(sys, "frozen")
27 or pycompat.safehasattr(sys, "importers") # new py2exe
27 or pycompat.safehasattr(sys, "importers") # new py2exe
28 or imp.is_frozen("__main__") # old py2exe
28 or imp.is_frozen("__main__") # old py2exe
29 ) # tools/freeze
29 ) # tools/freeze
30
30
31
31
32 # the location of data files matching the source code
32 # the location of data files matching the source code
33 if mainfrozen() and getattr(sys, 'frozen', None) != 'macosx_app':
33 if mainfrozen() and getattr(sys, 'frozen', None) != 'macosx_app':
34 # executable version (py2exe) doesn't support __file__
34 # executable version (py2exe) doesn't support __file__
35 datapath = os.path.dirname(pycompat.sysexecutable)
35 datapath = os.path.dirname(pycompat.sysexecutable)
36 _rootpath = os.path.dirname(datapath)
36 else:
37 else:
37 datapath = os.path.dirname(os.path.dirname(pycompat.fsencode(__file__)))
38 datapath = os.path.dirname(os.path.dirname(pycompat.fsencode(__file__)))
38 _rootpath = os.path.dirname(datapath)
39 _rootpath = os.path.dirname(datapath)
39
40
40 try:
41 try:
41 from importlib import resources
42 from importlib import resources
42
43
43 from .. import encoding
44 from .. import encoding
44
45
45 # Force loading of the resources module
46 # Force loading of the resources module
46 resources.open_binary # pytype: disable=module-attr
47 resources.open_binary # pytype: disable=module-attr
47
48
48 def open_resource(package, name):
49 def open_resource(package, name):
49 return resources.open_binary( # pytype: disable=module-attr
50 return resources.open_binary( # pytype: disable=module-attr
50 pycompat.sysstr(package), pycompat.sysstr(name)
51 pycompat.sysstr(package), pycompat.sysstr(name)
51 )
52 )
52
53
53 def is_resource(package, name):
54 def is_resource(package, name):
54 return resources.is_resource(
55 return resources.is_resource(
55 pycompat.sysstr(package), encoding.strfromlocal(name)
56 pycompat.sysstr(package), encoding.strfromlocal(name)
56 )
57 )
57
58
58 def contents(package):
59 def contents(package):
59 for r in resources.contents(pycompat.sysstr(package)):
60 for r in resources.contents(pycompat.sysstr(package)):
60 yield encoding.strtolocal(r)
61 yield encoding.strtolocal(r)
61
62
62
63
63 except (ImportError, AttributeError):
64 except (ImportError, AttributeError):
64
65
65 def _package_path(package):
66 def _package_path(package):
66 return os.path.join(_rootpath, *package.split(b'.'))
67 return os.path.join(_rootpath, *package.split(b'.'))
67
68
68 def open_resource(package, name):
69 def open_resource(package, name):
69 path = os.path.join(_package_path(package), name)
70 path = os.path.join(_package_path(package), name)
70 return open(path, 'rb')
71 return open(path, 'rb')
71
72
72 def is_resource(package, name):
73 def is_resource(package, name):
73 path = os.path.join(_package_path(package), name)
74 path = os.path.join(_package_path(package), name)
74
75
75 try:
76 try:
76 return os.path.isfile(pycompat.fsdecode(path))
77 return os.path.isfile(pycompat.fsdecode(path))
77 except (IOError, OSError):
78 except (IOError, OSError):
78 return False
79 return False
79
80
80 def contents(package):
81 def contents(package):
81 path = pycompat.fsdecode(_package_path(package))
82 path = pycompat.fsdecode(_package_path(package))
82
83
83 for p in os.listdir(path):
84 for p in os.listdir(path):
84 yield pycompat.fsencode(p)
85 yield pycompat.fsencode(p)
General Comments 0
You need to be logged in to leave comments. Login now