|
| 1 | +from __future__ import annotations |
| 2 | + |
1 | 3 | import functools |
2 | 4 | import importlib.abc |
| 5 | +import importlib.machinery |
| 6 | +import importlib.util |
| 7 | +import json |
3 | 8 | import os |
| 9 | +import pathlib |
| 10 | +import re |
4 | 11 | import subprocess |
5 | 12 | import sys |
6 | | -import warnings |
7 | 13 |
|
8 | 14 | from types import ModuleType |
9 | | -from typing import List, Mapping, Optional, Union |
10 | | - |
11 | | - |
12 | | -if sys.version_info >= (3, 9): |
13 | | - from collections.abc import Sequence |
14 | | -else: |
15 | | - from typing import Sequence |
16 | | - |
17 | | - |
18 | | -# This file should be standalone! |
19 | | -# It is copied during the editable hook installation. |
20 | | - |
21 | | - |
22 | | -_COLORS = { |
23 | | - 'cyan': '\33[36m', |
24 | | - 'yellow': '\33[93m', |
25 | | - 'light_blue': '\33[94m', |
26 | | - 'bold': '\33[1m', |
27 | | - 'dim': '\33[2m', |
28 | | - 'underline': '\33[4m', |
29 | | - 'reset': '\33[0m', |
30 | | -} |
31 | | -_NO_COLORS = {color: '' for color in _COLORS} |
32 | | - |
33 | | - |
34 | | -def _init_colors() -> Mapping[str, str]: |
35 | | - """Detect if we should be using colors in the output. We will enable colors |
36 | | - if running in a TTY, and no environment variable overrides it. Setting the |
37 | | - NO_COLOR (https://no-color.org/) environment variable force-disables colors, |
38 | | - and FORCE_COLOR forces color to be used, which is useful for thing like |
39 | | - Github actions. |
40 | | - """ |
41 | | - if 'NO_COLOR' in os.environ: |
42 | | - if 'FORCE_COLOR' in os.environ: |
43 | | - warnings.warn('Both NO_COLOR and FORCE_COLOR environment variables are set, disabling color') |
44 | | - return _NO_COLORS |
45 | | - elif 'FORCE_COLOR' in os.environ or sys.stdout.isatty(): |
46 | | - return _COLORS |
47 | | - return _NO_COLORS |
48 | | - |
49 | | - |
50 | | -_STYLES = _init_colors() # holds the color values, should be _COLORS or _NO_COLORS |
51 | | - |
52 | | - |
53 | | -class MesonpyFinder(importlib.abc.MetaPathFinder): |
54 | | - """Custom loader that whose purpose is to detect when the import system is |
55 | | - trying to load our modules, and trigger a rebuild. After triggering a |
56 | | - rebuild, we return None in find_spec, letting the normal finders pick up the |
57 | | - modules. |
58 | | - """ |
59 | | - |
60 | | - def __init__( |
61 | | - self, |
62 | | - project_name: str, |
63 | | - hook_name: str, |
64 | | - project_path: str, |
65 | | - build_path: str, |
66 | | - import_paths: List[str], |
67 | | - top_level_modules: List[str], |
68 | | - rebuild_commands: List[List[str]], |
69 | | - verbose: bool = False, |
70 | | - ) -> None: |
71 | | - self._project_name = project_name |
72 | | - self._hook_name = hook_name |
73 | | - self._project_path = project_path |
74 | | - self._build_path = build_path |
75 | | - self._import_paths = import_paths |
76 | | - self._top_level_modules = top_level_modules |
77 | | - self._rebuild_commands = rebuild_commands |
78 | | - self._verbose = verbose |
79 | | - |
80 | | - for path in (self._project_path, self._build_path): |
81 | | - if not os.path.isdir(path): |
82 | | - raise ImportError( |
83 | | - f'{path} is not a directory, but it is required to rebuild ' |
84 | | - f'"{self._project_name}", which is installed in editable ' |
85 | | - 'mode. Please reinstall the project to get it back to ' |
86 | | - 'working condition. If there are any issues uninstalling ' |
87 | | - 'this installation, you can manually remove ' |
88 | | - f'{self._hook_name} and {os.path.basename(__file__)}, ' |
89 | | - f'located in {os.path.dirname(__file__)}.' |
90 | | - ) |
91 | | - |
92 | | - def __repr__(self) -> str: |
93 | | - return f'{self.__class__.__name__}({self._project_path})' |
94 | | - |
95 | | - def _debug(self, msg: str) -> None: |
96 | | - if self._verbose: |
97 | | - print(msg.format(**_STYLES)) |
98 | | - |
99 | | - def _proc(self, command: List[str]) -> None: |
100 | | - # skip editable hook installation in subprocesses, as during the build |
101 | | - # commands the module we are rebuilding might be imported, causing a |
102 | | - # rebuild loop |
103 | | - # see https://github.com/mesonbuild/meson-python/pull/87#issuecomment-1342548894 |
104 | | - env = os.environ.copy() |
105 | | - env['_MESONPY_EDITABLE_SKIP'] = os.pathsep.join(( |
106 | | - env.get('_MESONPY_EDITABLE_SKIP', ''), |
107 | | - self._project_path, |
108 | | - )) |
109 | | - |
110 | | - if self._verbose: |
111 | | - subprocess.check_call(command, cwd=self._build_path, env=env) |
112 | | - else: |
113 | | - subprocess.check_output(command, cwd=self._build_path, env=env) |
114 | | - |
115 | | - @functools.lru_cache(maxsize=1) |
116 | | - def rebuild(self) -> None: |
117 | | - self._debug(f'{{cyan}}{{bold}}+ rebuilding {self._project_path}{{reset}}') |
118 | | - for command in self._rebuild_commands: |
119 | | - self._proc(command) |
120 | | - self._debug('{cyan}{bold}+ successfully rebuilt{reset}') |
| 15 | +from typing import Any, Dict, Optional, Sequence, Set, Union |
| 16 | + |
| 17 | + |
| 18 | +MARKER = 'MESON_PYTHON_EDITABLE_SKIP' |
| 19 | +VERBOSE = 'MESON_PYTHON_EDITABLE_VERBOSE' |
| 20 | +SUFFIXES = frozenset(importlib.machinery.all_suffixes()) |
| 21 | + |
| 22 | + |
| 23 | +def collect(install_plan: Dict[str, Dict[str, Any]]) -> Dict[str, str]: |
| 24 | + modules = {} |
| 25 | + for group in install_plan.values(): |
| 26 | + for src, target in group.items(): |
| 27 | + path = pathlib.Path(target['destination']) |
| 28 | + if path.parts[0] in {'{py_platlib}', '{py_purelib}'}: |
| 29 | + if path.parts[-1] == '__init__.py': |
| 30 | + module = '.'.join(path.parts[1:-1]) |
| 31 | + modules[module] = src |
| 32 | + else: |
| 33 | + match = re.match(r'^([^.]+)(.*)$', path.name) |
| 34 | + assert match is not None |
| 35 | + name, suffix = match.groups() |
| 36 | + if suffix in SUFFIXES: |
| 37 | + module = '.'.join(path.parts[1:-1] + (name, )) |
| 38 | + modules[module] = src |
| 39 | + return modules |
| 40 | + |
| 41 | + |
| 42 | +class MesonpyMetaFinder(importlib.abc.MetaPathFinder): |
| 43 | + def __init__(self, names: Set[str], path: str): |
| 44 | + self._top_level_modules = names |
| 45 | + self._build_path = path |
| 46 | + self._verbose = False |
| 47 | + |
| 48 | + @property |
| 49 | + def verbose(self) -> bool: |
| 50 | + return self._verbose or os.environ.get(VERBOSE, '') |
121 | 51 |
|
122 | 52 | def find_spec( |
123 | | - self, |
124 | | - fullname: str, |
125 | | - path: Optional[Sequence[Union[str, bytes]]], |
126 | | - target: Optional[ModuleType] = None, |
127 | | - ) -> None: |
128 | | - # if it's one of our modules, trigger a rebuild |
| 53 | + self, |
| 54 | + fullname: str, |
| 55 | + path: Optional[Sequence[Union[bytes, str]]] = None, |
| 56 | + target: Optional[ModuleType] = None |
| 57 | + ) -> Optional[importlib.machinery.ModuleSpec]: |
129 | 58 | if fullname.split('.', maxsplit=1)[0] in self._top_level_modules: |
| 59 | + if self._build_path in os.environ.get(MARKER, '').split(os.pathsep): |
| 60 | + return None |
130 | 61 | self.rebuild() |
131 | | - # prepend the project path to sys.path, so that the normal finder |
132 | | - # can find our modules |
133 | | - # we prepend so that our path comes before the current path (if |
134 | | - # the interpreter is run with -m), see gh-239 |
135 | | - if sys.path[:len(self._import_paths)] != self._import_paths: |
136 | | - for path in self._import_paths: |
137 | | - if path in sys.path: |
138 | | - sys.path.remove(path) |
139 | | - sys.path = self._import_paths + sys.path |
140 | | - # return none (meaning we "didn't find" the module) and let the normal |
141 | | - # finders find/import it |
| 62 | + install_plan_path = os.path.join(self._build_path, 'meson-info', 'intro-install_plan.json') |
| 63 | + with open(install_plan_path, 'r', encoding='utf8') as f: |
| 64 | + install_plan = json.load(f) |
| 65 | + modules = collect(install_plan) |
| 66 | + path = modules.get(fullname) |
| 67 | + if path is not None: |
| 68 | + return importlib.util.spec_from_file_location(fullname, path) |
142 | 69 | return None |
143 | 70 |
|
144 | | - @classmethod |
145 | | - def install( |
146 | | - cls, |
147 | | - project_name: str, |
148 | | - hook_name: str, |
149 | | - project_path: str, |
150 | | - build_path: str, |
151 | | - import_paths: List[str], |
152 | | - top_level_modules: List[str], |
153 | | - rebuild_commands: List[List[str]], |
154 | | - verbose: bool = False, |
155 | | - ) -> None: |
156 | | - if project_path in os.environ.get('_MESONPY_EDITABLE_SKIP', '').split(os.pathsep): |
157 | | - return |
158 | | - if os.environ.get('MESONPY_EDITABLE_VERBOSE', ''): |
159 | | - verbose = True |
160 | | - # install our finder |
161 | | - finder = cls( |
162 | | - project_name, |
163 | | - hook_name, |
164 | | - project_path, |
165 | | - build_path, |
166 | | - import_paths, |
167 | | - top_level_modules, |
168 | | - rebuild_commands, |
169 | | - verbose, |
170 | | - ) |
171 | | - if finder not in sys.meta_path: |
172 | | - # prepend our finder to sys.meta_path, so that it is queried before |
173 | | - # the normal finders, and can trigger a project rebuild |
174 | | - sys.meta_path.insert(0, finder) |
175 | | - # we add the project path to sys.path later, so that we can prepend |
176 | | - # after the current directory is prepended (when -m is used) |
177 | | - # see gh-239 |
| 71 | + @functools.lru_cache(maxsize=1) |
| 72 | + def rebuild(self) -> None: |
| 73 | + # skip editable wheel lookup during rebuild: during the build |
| 74 | + # the module we are rebuilding might be imported causing a |
| 75 | + # rebuild loop. |
| 76 | + env = os.environ.copy() |
| 77 | + env[MARKER] = os.pathsep.join((env.get(MARKER, ''), self._build_path)) |
| 78 | + stdout = None if self.verbose else subprocess.DEVNULL |
| 79 | + subprocess.run(['meson', 'compile'], cwd=self._build_path, env=env, stdout=stdout, check=True) |
178 | 80 |
|
179 | 81 |
|
180 | | -# generated hook install below |
| 82 | +def install(names: Set[str], path: str) -> None: |
| 83 | + sys.meta_path.insert(0, MesonpyMetaFinder(names, path)) |
0 commit comments