##// END OF EJS Templates
policy: remove unused policynoc and policynocffi constants
Yuya Nishihara -
r32514:3f5d675f default
parent child Browse files
Show More
@@ -1,106 +1,104
1 # policy.py - module policy logic for Mercurial.
1 # policy.py - module policy logic for Mercurial.
2 #
2 #
3 # Copyright 2015 Gregory Szorc <gregory.szorc@gmail.com>
3 # Copyright 2015 Gregory Szorc <gregory.szorc@gmail.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 from __future__ import absolute_import
8 from __future__ import absolute_import
9
9
10 import os
10 import os
11 import sys
11 import sys
12
12
13 # Rules for how modules can be loaded. Values are:
13 # Rules for how modules can be loaded. Values are:
14 #
14 #
15 # c - require C extensions
15 # c - require C extensions
16 # allow - allow pure Python implementation when C loading fails
16 # allow - allow pure Python implementation when C loading fails
17 # cffi - required cffi versions (implemented within pure module)
17 # cffi - required cffi versions (implemented within pure module)
18 # cffi-allow - allow pure Python implementation if cffi version is missing
18 # cffi-allow - allow pure Python implementation if cffi version is missing
19 # py - only load pure Python modules
19 # py - only load pure Python modules
20 #
20 #
21 # By default, fall back to the pure modules so the in-place build can
21 # By default, fall back to the pure modules so the in-place build can
22 # run without recompiling the C extensions. This will be overridden by
22 # run without recompiling the C extensions. This will be overridden by
23 # __modulepolicy__ generated by setup.py.
23 # __modulepolicy__ generated by setup.py.
24 policy = b'allow'
24 policy = b'allow'
25 policynoc = (b'cffi', b'cffi-allow', b'py')
26 policynocffi = (b'c', b'py')
27 _packageprefs = {
25 _packageprefs = {
28 # policy: (versioned package, pure package)
26 # policy: (versioned package, pure package)
29 b'c': (r'cext', None),
27 b'c': (r'cext', None),
30 b'allow': (r'cext', r'pure'),
28 b'allow': (r'cext', r'pure'),
31 b'cffi': (r'cffi', None),
29 b'cffi': (r'cffi', None),
32 b'cffi-allow': (r'cffi', r'pure'),
30 b'cffi-allow': (r'cffi', r'pure'),
33 b'py': (None, r'pure'),
31 b'py': (None, r'pure'),
34 }
32 }
35
33
36 try:
34 try:
37 from . import __modulepolicy__
35 from . import __modulepolicy__
38 policy = __modulepolicy__.modulepolicy
36 policy = __modulepolicy__.modulepolicy
39 except ImportError:
37 except ImportError:
40 pass
38 pass
41
39
42 # PyPy doesn't load C extensions.
40 # PyPy doesn't load C extensions.
43 #
41 #
44 # The canonical way to do this is to test platform.python_implementation().
42 # The canonical way to do this is to test platform.python_implementation().
45 # But we don't import platform and don't bloat for it here.
43 # But we don't import platform and don't bloat for it here.
46 if r'__pypy__' in sys.builtin_module_names:
44 if r'__pypy__' in sys.builtin_module_names:
47 policy = b'cffi'
45 policy = b'cffi'
48
46
49 # Our C extensions aren't yet compatible with Python 3. So use pure Python
47 # Our C extensions aren't yet compatible with Python 3. So use pure Python
50 # on Python 3 for now.
48 # on Python 3 for now.
51 if sys.version_info[0] >= 3:
49 if sys.version_info[0] >= 3:
52 policy = b'py'
50 policy = b'py'
53
51
54 # Environment variable can always force settings.
52 # Environment variable can always force settings.
55 if sys.version_info[0] >= 3:
53 if sys.version_info[0] >= 3:
56 if r'HGMODULEPOLICY' in os.environ:
54 if r'HGMODULEPOLICY' in os.environ:
57 policy = os.environ[r'HGMODULEPOLICY'].encode(r'utf-8')
55 policy = os.environ[r'HGMODULEPOLICY'].encode(r'utf-8')
58 else:
56 else:
59 policy = os.environ.get(r'HGMODULEPOLICY', policy)
57 policy = os.environ.get(r'HGMODULEPOLICY', policy)
60
58
61 def _importfrom(pkgname, modname):
59 def _importfrom(pkgname, modname):
62 # from .<pkgname> import <modname> (where . is looked through this module)
60 # from .<pkgname> import <modname> (where . is looked through this module)
63 fakelocals = {}
61 fakelocals = {}
64 pkg = __import__(pkgname, globals(), fakelocals, [modname], level=1)
62 pkg = __import__(pkgname, globals(), fakelocals, [modname], level=1)
65 try:
63 try:
66 fakelocals[modname] = mod = getattr(pkg, modname)
64 fakelocals[modname] = mod = getattr(pkg, modname)
67 except AttributeError:
65 except AttributeError:
68 raise ImportError(r'cannot import name %s' % modname)
66 raise ImportError(r'cannot import name %s' % modname)
69 # force import; fakelocals[modname] may be replaced with the real module
67 # force import; fakelocals[modname] may be replaced with the real module
70 getattr(mod, r'__doc__', None)
68 getattr(mod, r'__doc__', None)
71 return fakelocals[modname]
69 return fakelocals[modname]
72
70
73 # keep in sync with "version" in C modules
71 # keep in sync with "version" in C modules
74 _cextversions = {
72 _cextversions = {
75 (r'cext', r'base85'): 1,
73 (r'cext', r'base85'): 1,
76 (r'cext', r'bdiff'): 1,
74 (r'cext', r'bdiff'): 1,
77 (r'cext', r'diffhelpers'): 1,
75 (r'cext', r'diffhelpers'): 1,
78 (r'cext', r'mpatch'): 1,
76 (r'cext', r'mpatch'): 1,
79 (r'cext', r'osutil'): 1,
77 (r'cext', r'osutil'): 1,
80 (r'cext', r'parsers'): 1,
78 (r'cext', r'parsers'): 1,
81 }
79 }
82
80
83 def _checkmod(pkgname, modname, mod):
81 def _checkmod(pkgname, modname, mod):
84 expected = _cextversions.get((pkgname, modname))
82 expected = _cextversions.get((pkgname, modname))
85 actual = getattr(mod, r'version', None)
83 actual = getattr(mod, r'version', None)
86 if actual != expected:
84 if actual != expected:
87 raise ImportError(r'cannot import module %s.%s '
85 raise ImportError(r'cannot import module %s.%s '
88 r'(expected version: %d, actual: %r)'
86 r'(expected version: %d, actual: %r)'
89 % (pkgname, modname, expected, actual))
87 % (pkgname, modname, expected, actual))
90
88
91 def importmod(modname):
89 def importmod(modname):
92 """Import module according to policy and check API version"""
90 """Import module according to policy and check API version"""
93 try:
91 try:
94 verpkg, purepkg = _packageprefs[policy]
92 verpkg, purepkg = _packageprefs[policy]
95 except KeyError:
93 except KeyError:
96 raise ImportError(r'invalid HGMODULEPOLICY %r' % policy)
94 raise ImportError(r'invalid HGMODULEPOLICY %r' % policy)
97 assert verpkg or purepkg
95 assert verpkg or purepkg
98 if verpkg:
96 if verpkg:
99 try:
97 try:
100 mod = _importfrom(verpkg, modname)
98 mod = _importfrom(verpkg, modname)
101 _checkmod(verpkg, modname, mod)
99 _checkmod(verpkg, modname, mod)
102 return mod
100 return mod
103 except ImportError:
101 except ImportError:
104 if not purepkg:
102 if not purepkg:
105 raise
103 raise
106 return _importfrom(purepkg, modname)
104 return _importfrom(purepkg, modname)
General Comments 0
You need to be logged in to leave comments. Login now