253 lines
11 KiB
Python
253 lines
11 KiB
Python
from typing import Dict, List
|
|
|
|
from mach_nix.data.providers import WheelDependencyProvider, SdistDependencyProvider, NixpkgsDependencyProvider
|
|
from mach_nix.data.nixpkgs import NixpkgsIndex
|
|
from mach_nix.generators import ExpressionGenerator
|
|
from mach_nix.resolver import ResolvedPkg
|
|
|
|
|
|
def unindent(text: str, remove: int):
|
|
# removes indentation of text
|
|
# also strips leading newlines
|
|
return ''.join(map(lambda l: l[remove:], text.splitlines(keepends=True)))
|
|
|
|
|
|
class OverridesGenerator(ExpressionGenerator):
|
|
|
|
def __init__(
|
|
self,
|
|
py_ver,
|
|
nixpkgs: NixpkgsIndex,
|
|
pypi_fetcher_commit,
|
|
pypi_fetcher_sha256,
|
|
disable_checks,
|
|
*args,
|
|
**kwargs):
|
|
self.nixpkgs = nixpkgs
|
|
self.disable_checks = disable_checks
|
|
self.pypi_fetcher_commit = pypi_fetcher_commit
|
|
self.pypi_fetcher_sha256 = pypi_fetcher_sha256
|
|
self.py_ver_nix = py_ver.nix()
|
|
super(OverridesGenerator, self).__init__(*args, **kwargs)
|
|
|
|
def generate(self, reqs) -> str:
|
|
pkgs = self.resolver.resolve(reqs)
|
|
pkgs = dict(sorted(((p.name, p) for p in pkgs), key=lambda x: x[1].name))
|
|
return self._gen_python_env(pkgs)
|
|
|
|
def _gen_imports(self):
|
|
out = f"""
|
|
with builtins;
|
|
let
|
|
pypi_fetcher_src = builtins.fetchTarball {{
|
|
name = "nix-pypi-fetcher";
|
|
url = "https://github.com/DavHau/nix-pypi-fetcher/tarball/{self.pypi_fetcher_commit}";
|
|
# Hash obtained using `nix-prefetch-url --unpack <url>`
|
|
sha256 = "{self.pypi_fetcher_sha256}";
|
|
}};
|
|
fetchPypi = (import pypi_fetcher_src).fetchPypi;
|
|
fetchPypiWheel = (import pypi_fetcher_src).fetchPypiWheel;
|
|
try_get = obj: name:
|
|
if hasAttr name obj
|
|
then obj."${{name}}"
|
|
else [];
|
|
is_py_module = pkg:
|
|
isAttrs pkg && hasAttr "pythonModule" pkg;
|
|
filter_deps = oldAttrs: inputs_type:
|
|
filter (pkg: ! is_py_module pkg) (try_get oldAttrs inputs_type);
|
|
override = pkg:
|
|
if hasAttr "overridePythonAttrs" pkg then
|
|
pkg.overridePythonAttrs
|
|
else
|
|
pkg.overrideAttrs;
|
|
get_passthru = python: pname:
|
|
if hasAttr "${{pname}}" python then
|
|
let result = (tryEval
|
|
(if isNull python."${{pname}}" then
|
|
{{}}
|
|
else
|
|
python."${{pname}}".passthru));
|
|
in
|
|
if result.success then result.value else {{}}
|
|
else {{}};
|
|
"""
|
|
return unindent(out, 12)
|
|
|
|
def _gen_build_inputs(self, build_inputs_local, build_inputs_nixpkgs) -> str:
|
|
name = lambda n: f'python-self."{n}"' if '.' in n else n
|
|
build_inputs_str = ' '.join(
|
|
name(b) for b in sorted(build_inputs_local | build_inputs_nixpkgs))
|
|
return build_inputs_str
|
|
|
|
def _gen_prop_build_inputs(self, prop_build_inputs_local, prop_build_inputs_nixpkgs) -> str:
|
|
name = lambda n: f'python-self."{n}"' if '.' in n else n
|
|
prop_build_inputs_str = ' '.join(
|
|
name(b) for b in sorted(prop_build_inputs_local | prop_build_inputs_nixpkgs))
|
|
return prop_build_inputs_str
|
|
|
|
def _gen_overrideAttrs(
|
|
self, name, ver, circular_deps, nix_name, provider, build_inputs_str, prop_build_inputs_str,
|
|
keep_src=False):
|
|
out = f"""
|
|
"{nix_name}" = override python-super.{nix_name} ( oldAttrs: {{
|
|
pname = "{name}";
|
|
version = "{ver}";
|
|
passthru = (get_passthru python-super "{nix_name}") // {{ provider = "{provider}"; }};
|
|
buildInputs = with python-self; (filter_deps oldAttrs "buildInputs") ++ [ {build_inputs_str} ];
|
|
propagatedBuildInputs = with python-self; (filter_deps oldAttrs "propagatedBuildInputs") ++ [ {prop_build_inputs_str} ];"""
|
|
if not keep_src:
|
|
out += f"""
|
|
src = fetchPypi "{name}" "{ver}";"""
|
|
if circular_deps:
|
|
out += f"""
|
|
pipInstallFlags = "--no-dependencies";"""
|
|
if self.disable_checks:
|
|
out += """
|
|
doCheck = false;
|
|
doInstallCheck = false;"""
|
|
out += """
|
|
});\n"""
|
|
return unindent(out, 8)
|
|
|
|
def _gen_builPythonPackage(self, name, ver, circular_deps, nix_name, build_inputs_str, prop_build_inputs_str):
|
|
out = f"""
|
|
"{nix_name}" = python-self.buildPythonPackage {{
|
|
pname = "{name}";
|
|
version = "{ver}";
|
|
src = fetchPypi "{name}" "{ver}";
|
|
passthru = (get_passthru python-super "{nix_name}") // {{ provider = "sdist"; }};"""
|
|
if circular_deps:
|
|
out += f"""
|
|
pipInstallFlags = "--no-dependencies";"""
|
|
if build_inputs_str.strip():
|
|
out += f"""
|
|
buildInputs = with python-self; [ {build_inputs_str} ];"""
|
|
if prop_build_inputs_str.strip():
|
|
out += f"""
|
|
propagatedBuildInputs = with python-self; [ {prop_build_inputs_str} ];"""
|
|
if self.disable_checks:
|
|
out += """
|
|
doCheck = false;
|
|
doInstallCheck = false;"""
|
|
out += """
|
|
};\n"""
|
|
return unindent(out, 8)
|
|
|
|
def _gen_wheel_buildPythonPackage(self, name, ver, circular_deps, nix_name, prop_build_inputs_str, fname):
|
|
manylinux = "manylinux1 ++ " if 'manylinux' in fname else ''
|
|
|
|
# dontStrip added due to this bug - https://github.com/pypa/manylinux/issues/119
|
|
out = f"""
|
|
"{nix_name}" = python-self.buildPythonPackage {{
|
|
pname = "{name}";
|
|
version = "{ver}";
|
|
src = fetchPypiWheel "{name}" "{ver}" "{fname}";
|
|
format = "wheel";
|
|
doCheck = false;
|
|
doInstallCheck = false;
|
|
dontStrip = true;
|
|
passthru = (get_passthru python-super "{nix_name}") // {{ provider = "wheel"; }};"""
|
|
if circular_deps:
|
|
out += f"""
|
|
pipInstallFlags = "--no-dependencies";"""
|
|
if manylinux:
|
|
out += f"""
|
|
nativeBuildInputs = [ autoPatchelfHook ];
|
|
autoPatchelfIgnoreNotFound = true;"""
|
|
if prop_build_inputs_str.strip() or manylinux:
|
|
out += f"""
|
|
propagatedBuildInputs = with python-self; {manylinux}[ {prop_build_inputs_str} ];"""
|
|
out += """
|
|
};\n"""
|
|
return unindent(out, 8)
|
|
|
|
def _gen_overrides(self, pkgs: Dict[str, ResolvedPkg], overrides_keys):
|
|
pkg_names_str = "".join(
|
|
(f"ps.\"{self._get_ref_name(name, pkgs[name].ver)}\"\n{' ' * 14}"
|
|
for (name, pkg) in pkgs.items() if pkg.is_root))
|
|
out = f"""
|
|
select_pkgs = ps: [
|
|
{pkg_names_str.strip()}
|
|
];
|
|
overrides = manylinux1: autoPatchelfHook: python-self: python-super: {{
|
|
"""
|
|
out = unindent(out, 10)
|
|
for pkg in pkgs.values():
|
|
if pkg.name not in overrides_keys:
|
|
continue
|
|
overlays_required = True
|
|
# get correct build input names
|
|
_build_inputs = [self._get_ref_name(b, pkgs[b].ver) for b in pkg.build_inputs]
|
|
build_inputs_local = {b for b in _build_inputs if b in overrides_keys}
|
|
build_inputs_nixpkgs = set(_build_inputs) - build_inputs_local
|
|
# get correct propagated build input names
|
|
_prop_build_inputs = [self._get_ref_name(b, pkgs[b].ver) for b in pkg.prop_build_inputs]
|
|
prop_build_inputs_local = {b for b in _prop_build_inputs if b in overrides_keys}
|
|
prop_build_inputs_nixpkgs = set(_prop_build_inputs) - prop_build_inputs_local
|
|
# convert build inputs to string
|
|
build_inputs_str = self._gen_build_inputs(build_inputs_local, build_inputs_nixpkgs, ).strip()
|
|
# convert prop build inputs to string
|
|
prop_build_inputs_str = self._gen_prop_build_inputs(
|
|
prop_build_inputs_local, prop_build_inputs_nixpkgs).strip()
|
|
|
|
# SDIST
|
|
if isinstance(pkg.provider_info.provider, SdistDependencyProvider):
|
|
# generate package overlays either via `overrideAttrs` if package already exists in nixpkgs,
|
|
# or by creating it from scratch using `buildPythonPackage`
|
|
nix_name = self._get_ref_name(pkg.name, pkg.ver)
|
|
if self.nixpkgs.exists(pkg.name):
|
|
out += self._gen_overrideAttrs(
|
|
pkg.name,
|
|
pkg.provider_info.provider.deviated_version(pkg.name, pkg.ver),
|
|
pkg.removed_circular_deps,
|
|
nix_name,
|
|
'sdist',
|
|
build_inputs_str,
|
|
prop_build_inputs_str)
|
|
else:
|
|
out += self._gen_builPythonPackage(
|
|
pkg.name,
|
|
pkg.provider_info.provider.deviated_version(pkg.name, pkg.ver),
|
|
pkg.removed_circular_deps,
|
|
nix_name,
|
|
build_inputs_str,
|
|
prop_build_inputs_str)
|
|
# WHEEL
|
|
elif isinstance(pkg.provider_info.provider, WheelDependencyProvider):
|
|
out += self._gen_wheel_buildPythonPackage(
|
|
pkg.name,
|
|
pkg.provider_info.provider.deviated_version(pkg.name, pkg.ver),
|
|
pkg.removed_circular_deps,
|
|
self._get_ref_name(pkg.name, pkg.ver),
|
|
prop_build_inputs_str,
|
|
pkg.provider_info.wheel_fname)
|
|
# NIXPKGS
|
|
elif isinstance(pkg.provider_info.provider, NixpkgsDependencyProvider):
|
|
nix_name = self.nixpkgs.find_best_nixpkgs_candidate(pkg.name, pkg.ver)
|
|
out += self._gen_overrideAttrs(
|
|
pkg.name,
|
|
pkg.ver,
|
|
pkg.removed_circular_deps,
|
|
nix_name,
|
|
'nixpkgs',
|
|
build_inputs_str,
|
|
prop_build_inputs_str,
|
|
keep_src=True)
|
|
end_overlay_section = f"""
|
|
}};
|
|
"""
|
|
return out + unindent(end_overlay_section, 14)
|
|
|
|
def _get_ref_name(self, name, ver) -> str:
|
|
if self.nixpkgs.exists(name):
|
|
return self.nixpkgs.find_best_nixpkgs_candidate(name, ver)
|
|
return name
|
|
|
|
def _gen_python_env(self, pkgs: Dict[str, ResolvedPkg]):
|
|
overrides_keys = {p.name for p in pkgs.values()}
|
|
out = self._gen_imports() + self._gen_overrides(pkgs, overrides_keys)
|
|
python_with_packages = f"""
|
|
in
|
|
{{ inherit overrides select_pkgs; }}
|
|
"""
|
|
return out + unindent(python_with_packages, 12)
|