File: C:/Users/fred/anaconda3/Lib/site-packages/conda_build/jinja_context.py
# Copyright (C) 2014 Anaconda, Inc
# SPDX-License-Identifier: BSD-3-Clause
from __future__ import annotations
import datetime
import json
import os
import pathlib
import re
import time
from functools import partial
from io import StringIO, TextIOBase
from subprocess import CalledProcessError
from typing import TYPE_CHECKING
from warnings import warn
import jinja2
import yaml
from frozendict import deepfreeze
from . import _load_setup_py_data
from .environ import get_dict as get_environ
from .exceptions import CondaBuildException
from .render import get_env_dependencies
from .utils import (
apply_pin_expressions,
check_call_env,
copy_into,
ensure_valid_spec,
get_installed_packages,
get_logger,
rm_rf,
)
from .variants import DEFAULT_COMPILERS
try:
import tomllib # Python 3.11
except:
import tomli as tomllib
if TYPE_CHECKING:
from typing import IO, Any
log = get_logger(__name__)
class UndefinedNeverFail(jinja2.Undefined):
"""
A class for Undefined jinja variables.
This is even less strict than the default jinja2.Undefined class,
because it permits things like {{ MY_UNDEFINED_VAR[:2] }} and
{{ MY_UNDEFINED_VAR|int }}. This can mask lots of errors in jinja templates, so it
should only be used for a first-pass parse, when you plan on running a 'strict'
second pass later.
Note:
When using this class, any usage of an undefined variable in a jinja template is recorded
in the (global) all_undefined_names class member. Therefore, after jinja rendering,
you can detect which undefined names were used by inspecting that list.
Be sure to clear the all_undefined_names list before calling template.render().
"""
all_undefined_names = []
def __init__(
self,
hint=None,
obj=jinja2.runtime.missing,
name=None,
exc=jinja2.exceptions.UndefinedError,
):
jinja2.Undefined.__init__(self, hint, obj, name, exc)
# Using any of these methods on an Undefined variable
# results in another Undefined variable.
__add__ = __radd__ = __mul__ = __rmul__ = __div__ = __rdiv__ = __truediv__ = (
__rtruediv__
) = __floordiv__ = __rfloordiv__ = __mod__ = __rmod__ = __pos__ = __neg__ = (
__call__
) = __getitem__ = __lt__ = __le__ = __gt__ = __ge__ = __complex__ = __pow__ = (
__rpow__
) = lambda self, *args, **kwargs: self._return_undefined(self._undefined_name)
# Accessing an attribute of an Undefined variable
# results in another Undefined variable.
def __getattr__(self, k):
try:
return object.__getattr__(self, k)
except AttributeError:
self._return_undefined(self._undefined_name + "." + k)
# Unlike the methods above, Python requires that these
# few methods must always return the correct type
__str__ = __repr__ = lambda self: self._return_value("")
__unicode__ = lambda self: self._return_value("")
__int__ = lambda self: self._return_value(0)
__float__ = lambda self: self._return_value(0.0)
__nonzero__ = lambda self: self._return_value(False)
def _return_undefined(self, result_name):
# Record that this undefined variable was actually used.
UndefinedNeverFail.all_undefined_names.append(self._undefined_name)
return UndefinedNeverFail(
hint=self._undefined_hint,
obj=self._undefined_obj,
name=result_name,
exc=self._undefined_exception,
)
def _return_value(self, value=None):
# Record that this undefined variable was actually used.
UndefinedNeverFail.all_undefined_names.append(self._undefined_name)
return value
class FilteredLoader(jinja2.BaseLoader):
"""
A pass-through for the given loader, except that the loaded source is
filtered according to any metadata selectors in the source text.
"""
def __init__(self, unfiltered_loader, config):
self._unfiltered_loader = unfiltered_loader
self.list_templates = unfiltered_loader.list_templates
self.config = config
def get_source(self, environment, template):
# we have circular imports here. Do a local import
from .metadata import get_selectors, select_lines
contents, filename, uptodate = self._unfiltered_loader.get_source(
environment, template
)
return (
select_lines(
contents,
get_selectors(self.config),
variants_in_place=bool(self.config.variant),
),
filename,
uptodate,
)
def load_setup_py_data(
m,
setup_file="setup.py",
from_recipe_dir=False,
recipe_dir=None,
permit_undefined_jinja=True,
):
_setuptools_data = None
# we must copy the script into the work folder to avoid incompatible pyc files
origin_setup_script = os.path.join(
os.path.dirname(__file__), "_load_setup_py_data.py"
)
dest_setup_script = os.path.join(m.config.work_dir, "_load_setup_py_data.py")
copy_into(origin_setup_script, dest_setup_script)
env = get_environ(m)
env["CONDA_BUILD_STATE"] = "RENDER"
if os.path.isfile(m.config.build_python):
args = [m.config.build_python, dest_setup_script, m.config.work_dir, setup_file]
if from_recipe_dir:
assert recipe_dir, "recipe_dir must be set if from_recipe_dir is True"
args.append("--from-recipe-dir")
args.extend(["--recipe-dir", recipe_dir])
if permit_undefined_jinja:
args.append("--permit-undefined-jinja")
try:
check_call_env(args, env=env)
except CalledProcessError as exc:
raise CondaBuildException(
"Could not run load_setup_py_data in subprocess"
) from exc
# this is a file that the subprocess will have written
with open(
os.path.join(m.config.work_dir, "conda_build_loaded_setup_py.json")
) as f:
_setuptools_data = json.load(f)
else:
try:
_setuptools_data = _load_setup_py_data.load_setup_py_data(
setup_file,
from_recipe_dir=from_recipe_dir,
recipe_dir=recipe_dir,
work_dir=m.config.work_dir,
permit_undefined_jinja=permit_undefined_jinja,
)
except (TypeError, OSError):
# setup.py file doesn't yet exist. Will get picked up in future parsings
pass
except ImportError as e:
if permit_undefined_jinja:
log.debug(
"Reading setup.py failed due to missing modules. This is probably OK, "
"since it may succeed in later passes. Watch for incomplete recipe "
"info, though."
)
else:
raise CondaBuildException(
"Could not render recipe - need modules "
f'installed in root env. Import error was "{e}"'
)
# cleanup: we must leave the source tree empty unless the source code is already present
rm_rf(os.path.join(m.config.work_dir, "_load_setup_py_data.py"))
return _setuptools_data if _setuptools_data else {}
def load_setuptools(
m,
setup_file="setup.py",
from_recipe_dir=False,
recipe_dir=None,
permit_undefined_jinja=True,
):
warn(
"conda_build.jinja_context.load_setuptools is pending deprecation in a future release. "
"Use conda_build.jinja_context.load_setup_py_data instead.",
PendingDeprecationWarning,
)
return load_setup_py_data(
m,
setup_file=setup_file,
from_recipe_dir=from_recipe_dir,
recipe_dir=recipe_dir,
permit_undefined_jinja=permit_undefined_jinja,
)
def load_npm():
mode_dict = {"mode": "r", "encoding": "utf-8"}
with open("package.json", **mode_dict) as pkg:
return json.load(pkg)
def _find_file(file_name: str, from_recipe_dir: bool, recipe_dir: str, config) -> str:
"""Get the path to the given file which may be in the work_dir
or in the recipe_dir.
Note, the returned file name may not exist.
"""
if os.path.isabs(file_name):
path = file_name
elif from_recipe_dir and recipe_dir:
path = os.path.abspath(os.path.join(recipe_dir, file_name))
elif os.path.exists(config.work_dir):
path = os.path.join(config.work_dir, file_name)
if not os.path.isfile(path):
raise FileNotFoundError(f"No such file: {file_name!r}")
return path
def load_file_regex(
config,
load_file,
regex_pattern,
from_recipe_dir=False,
recipe_dir=None,
permit_undefined_jinja=True,
):
try:
load_file = _find_file(load_file, from_recipe_dir, recipe_dir, config)
except FileNotFoundError as e:
if permit_undefined_jinja:
log.debug(e)
return None
raise
else:
with open(load_file) as lfile:
return re.search(regex_pattern, lfile.read())
cached_env_dependencies = {}
def pin_compatible(
m,
package_name,
lower_bound=None,
upper_bound=None,
min_pin="x.x.x.x.x.x",
max_pin="x",
permit_undefined_jinja=False,
exact=False,
bypass_env_check=False,
):
"""dynamically pin based on currently installed version.
only mandatory input is package_name.
upper_bound is the authoritative upper bound, if provided. The lower bound is the the
currently installed version.
pin expressions are of the form 'x.x' - the number of pins is the number of x's separated
by ``.``.
"""
global cached_env_dependencies
compatibility = ""
# optimization: this is slow (requires solver), so better to bypass it
# until the finalization stage.
if not bypass_env_check and not permit_undefined_jinja:
# this is the version split up into its component bits.
# There are two cases considered here (so far):
# 1. Good packages that follow semver style (if not philosophy). For example, 1.2.3
# 2. Evil packages that cram everything alongside a single major version. For example, 9b
key = (m.name(), deepfreeze(m.config.variant))
if key in cached_env_dependencies:
pins = cached_env_dependencies[key]
else:
if m.is_cross and not m.build_is_host:
pins, _, _ = get_env_dependencies(m, "host", m.config.variant)
else:
pins, _, _ = get_env_dependencies(m, "build", m.config.variant)
if m.build_is_host:
host_pins, _, _ = get_env_dependencies(m, "host", m.config.variant)
pins.extend(host_pins)
cached_env_dependencies[key] = pins
versions = {p.split(" ")[0]: p.split(" ")[1:] for p in pins}
if versions:
if exact and versions.get(package_name):
compatibility = " ".join(versions[package_name])
else:
version = lower_bound or versions.get(package_name)
if version:
if hasattr(version, "__iter__") and not isinstance(version, str):
version = version[0]
else:
version = str(version)
if upper_bound:
if min_pin or lower_bound:
compatibility = ">=" + str(version) + ","
compatibility += f"<{upper_bound}"
else:
compatibility = apply_pin_expressions(version, min_pin, max_pin)
if not compatibility and not permit_undefined_jinja and not bypass_env_check:
check = re.compile(rf'pin_compatible\s*\(\s*["]{package_name}["]')
if check.search(m.extract_requirements_text()):
raise RuntimeError(
f"Could not get compatibility information for {package_name} package. "
"Is it one of your host dependencies?"
)
return (
" ".join((package_name, compatibility))
if compatibility is not None
else package_name
)
def pin_subpackage_against_outputs(
metadata,
matching_package_keys,
outputs,
min_pin,
max_pin,
exact,
permit_undefined_jinja,
skip_build_id=False,
):
pin = None
if matching_package_keys:
# two ways to match:
# 1. only one other output named the same as the subpackage_name from the key
# 2. whole key matches (both subpackage name and variant (used keys only))
if len(matching_package_keys) == 1:
key = matching_package_keys[0]
elif len(matching_package_keys) > 1:
key = None
for pkg_name, variant in matching_package_keys:
# This matches other outputs with any keys that are common to both
# metadata objects. For debugging, the keys are always the (package
# name, used vars+values). It used to be (package name, variant) -
# but that was really big and hard to look at.
shared_vars = set(variant.keys()) & set(metadata.config.variant.keys())
if not shared_vars or all(
variant[sv] == metadata.config.variant[sv] for sv in shared_vars
):
key = (pkg_name, variant)
break
if key in outputs:
sp_m = outputs[key][1]
if permit_undefined_jinja and not sp_m.version():
pin = None
else:
if exact:
pin = " ".join(
[
sp_m.name(),
sp_m.version(),
sp_m.build_id()
if not skip_build_id
else str(sp_m.build_number()),
]
)
else:
pin = f"{sp_m.name()} {apply_pin_expressions(sp_m.version(), min_pin, max_pin)}"
else:
pin = matching_package_keys[0][0]
return pin
def pin_subpackage(
metadata,
subpackage_name,
min_pin="x.x.x.x.x.x",
max_pin="x",
exact=False,
permit_undefined_jinja=False,
allow_no_other_outputs=False,
skip_build_id=False,
):
"""allow people to specify pinnings based on subpackages that are defined in the recipe.
For example, given a compiler package, allow it to specify either a compatible or exact
pinning on the runtime package that is also created by the compiler package recipe
"""
pin = None
if not hasattr(metadata, "other_outputs"):
if allow_no_other_outputs:
pin = subpackage_name
else:
raise ValueError(
"Bug in conda-build: we need to have info about other outputs in "
"order to allow pinning to them. It's not here."
)
else:
# two ways to match:
# 1. only one other output named the same as the subpackage_name from the key
# 2. whole key matches (both subpackage name and variant)
keys = list(metadata.other_outputs.keys())
matching_package_keys = [k for k in keys if k[0] == subpackage_name]
pin = pin_subpackage_against_outputs(
metadata,
matching_package_keys,
metadata.other_outputs,
min_pin,
max_pin,
exact,
permit_undefined_jinja,
skip_build_id=skip_build_id,
)
if not pin:
pin = subpackage_name
if not permit_undefined_jinja and not allow_no_other_outputs:
raise ValueError(
f"Didn't find subpackage version info for '{subpackage_name}', which is used in a"
" pin_subpackage expression. Is it actually a subpackage? If not, "
"you want pin_compatible instead."
)
return pin
def native_compiler(language, config):
compiler = language
for platform in [config.platform, config.platform.split("-")[0]]:
try:
compiler = DEFAULT_COMPILERS[platform][language]
break
except KeyError:
continue
if hasattr(compiler, "keys"):
compiler = compiler.get(config.variant.get("python", "nope"), "vs2017")
return compiler
def _target(language, config, permit_undefined_jinja=False, component="compiler"):
"""Support configuration of compilers/stdlib. This is somewhat platform specific.
Native compilers/stdlib never list their host - it is always implied. Generally, they are
metapackages, pointing at a package that does specify the host. These in turn may be
metapackages, pointing at a package where the host is the same as the target (both being the
native architecture).
"""
if component == "compiler":
package_prefix = native_compiler(language, config)
else:
package_prefix = language
version = None
if config.variant:
target_platform = config.variant.get("target_platform", config.subdir)
language_key = f"{language}_{component}"
# fall back to native if language-key is not explicitly set in variant
package_prefix = config.variant.get(language_key, package_prefix)
version = config.variant.get(language_key + "_version")
else:
target_platform = config.subdir
# support cross components. A cross package will have a name such as
# gcc_target
# gcc_linux-cos6-64
package = f"{package_prefix}_{target_platform}"
if version:
package = f"{package} {version}"
package = ensure_valid_spec(package, warn=False)
return package
# ensure we have compiler in namespace
compiler = partial(_target, component="compiler")
def ccache(method, config, permit_undefined_jinja=False):
config.ccache_method = method
return "ccache"
def cdt(package_name, config, permit_undefined_jinja=False):
"""Support configuration of Core Dependency Trees.
We should define CDTs in a single location. The current
idea is to emit parts of the following to index.json (the
bits that the solver could make use of) and parts to
about.json (the other bits).
"system": {
"os": {
"type": "windows", "linux", "bsd", "darwin",
"os_distribution": "CentOS", "FreeBSD", "Windows", "osx",
"os_version": "6.9", "10.12.3",
"os_kernel_version" : "2.6.32",
"os_libc_family": "glibc",
"os_libc_version": "2.12",
}
"cpu": {
# Whichever cpu_architecture/cpu_isa we build-out for:
# .. armv6 is compatible with and uses all CPU features of a Raspberry PI 1
# .. armv7a is compatible with and uses all CPU features of a Raspberry PI 2
# .. aarch64 is compatible with and uses all CPU features of a Raspberry PI 3
"cpu_architecture": "x86", "x86_64",
"armv6", "armv7a", "aarch32", "aarch64",
"powerpc", "powerpc64",
"s390", "s390x",
"cpu_isa": "nocona", "armv8.1-a", "armv8.3-a",
# "?" because the vfpu is specified by cpu_architecture + cpu_isa + rules.
"vfpu": "?",
"cpu_endianness": "BE", "LE",
}
"gpu ?": {
}
"compilerflags": {
# When put into a CDT these should be the base defaults.
# Package builds can and will change these frequently.
"CPPFLAGS": "-D_FORTIFY_SOURCE=2",
"CFLAGS": "-march=nocona -mtune=haswell -ftree-vectorize -fPIC -fstack-protector-strong -O2 -pipe",
"CXXFLAGS": "-fvisibility-inlines-hidden -std=c++17 -fmessage-length=0 -march=nocona -mtune=haswell -ftree-vectorize -fPIC -fstack-protector-strong -O2 -pipe",
"LDFLAGS": "-Wl,-O1,--sort-common,--as-needed,-z,relro",
"FFLAGS": "-fopenmp",
# These are appended to the non-DEBUG values:
"DEBUG_CFLAGS": "-Og -g -Wall -Wextra -fcheck=all -fbacktrace -fimplicit-none -fvar-tracking-assignments",
"DEBUG_CXXFLAGS": "-Og -g -Wall -Wextra -fcheck=all -fbacktrace -fimplicit-none -fvar-tracking-assignments",
"DEBUG_FFLAGS": "-Og -g -Wall -Wextra -fcheck=all -fbacktrace -fimplicit-none -fvar-tracking-assignments",
}
}
""" # NOQA
cdt_name = "cos6"
arch = config.host_arch or config.arch
if arch == "ppc64le" or arch == "aarch64" or arch == "ppc64" or arch == "s390x":
cdt_name = "cos7"
cdt_arch = arch
else:
cdt_arch = "x86_64" if arch == "64" else "i686"
if config.variant:
cdt_name = config.variant.get("cdt_name", cdt_name)
cdt_arch = config.variant.get("cdt_arch", cdt_arch)
if " " in package_name:
name = package_name.split(" ")[0]
ver_build = package_name.split(" ")[1:]
result = name + "-" + cdt_name + "-" + cdt_arch + " " + " ".join(ver_build)
else:
result = package_name + "-" + cdt_name + "-" + cdt_arch
return result
def resolved_packages(m, env, permit_undefined_jinja=False, bypass_env_check=False):
"""Returns the final list of packages that are listed in host or build.
This include all packages (including the indirect dependencies) that will
be installed in the host or build environment. An example usage of this
jinja function can be::
requirements:
host:
- curl 7.55.1
run_constrained:
{% for package in resolved_packages('host') %}
- {{ package }}
{% endfor %}
which will render to::
requirements:
host:
- ca-certificates 2017.08.26 h1d4fec5_0
- curl 7.55.1 h78862de_4
- libgcc-ng 7.2.0 h7cc24e2_2
- libssh2 1.8.0 h9cfc8f7_4
- openssl 1.0.2n hb7f436b_0
- zlib 1.2.11 ha838bed_2
run_constrained:
- ca-certificates 2017.08.26 h1d4fec5_0
- curl 7.55.1 h78862de_4
- libgcc-ng 7.2.0 h7cc24e2_2
- libssh2 1.8.0 h9cfc8f7_4
- openssl 1.0.2n hb7f436b_0
- zlib 1.2.11 ha838bed_2
"""
if env not in ("host", "build"):
raise ValueError("Only host and build dependencies are supported.")
package_names = []
# optimization: this is slow (requires solver), so better to bypass it
# until the finalization stage as done similarly in pin_compatible.
if not bypass_env_check and not permit_undefined_jinja:
package_names, _, _ = get_env_dependencies(m, env, m.config.variant)
return package_names
def _toml_load(stream):
"""
Load .toml from a pathname.
"""
if isinstance(stream, (TextIOBase, str)):
if isinstance(stream, TextIOBase):
data = stream.read()
else:
data = stream
return tomllib.loads(data)
# tomllib prefers binary files
return tomllib.load(stream)
_file_parsers = {
"json": json.load,
"yaml": yaml.safe_load,
"yml": yaml.safe_load,
"toml": _toml_load,
}
def _load_data(stream: IO, fmt: str, *args, **kwargs) -> Any:
try:
load = _file_parsers[fmt.lower()]
except KeyError:
raise ValueError(
f"Unknown file format: {fmt}. Allowed formats: {list(_file_parsers.keys())}"
)
else:
return load(stream, *args, **kwargs)
def load_file_data(
filename: str,
fmt: str | None = None,
*args,
config=None,
from_recipe_dir=False,
recipe_dir=None,
permit_undefined_jinja=True,
**kwargs,
):
"""Loads a file and returns the parsed data.
For example to load file data from a JSON file, you can use any of:
load_file_data("my_file.json")
load_file_data("my_json_file_without_ext", "json")
"""
try:
file_path = _find_file(filename, from_recipe_dir, recipe_dir, config)
except FileNotFoundError as e:
if permit_undefined_jinja:
log.debug(e)
return {}
raise
else:
with open(file_path) as f:
return _load_data(
f, fmt or pathlib.Path(filename).suffix.lstrip("."), *args, **kwargs
)
def load_str_data(string: str, fmt: str, *args, **kwargs):
"""Parses data from a string
For example to load string data, you can use any of:
load_str_data("[{"name": "foo", "val": "bar"}, {"name": "biz", "val": "buz"}]", "json")
load_str_data("[tool.poetry]\\nname = 'foo'", "toml")
load_str_data("name: foo\\nval: bar", "yaml")
"""
return _load_data(StringIO(string), fmt, *args, **kwargs)
def context_processor(
initial_metadata,
recipe_dir,
config,
permit_undefined_jinja,
allow_no_other_outputs=False,
bypass_env_check=False,
skip_build_id=False,
variant=None,
):
"""
Return a dictionary to use as context for jinja templates.
initial_metadata: Augment the context with values from this MetaData object.
Used to bootstrap metadata contents via multiple parsing passes.
"""
ctx = get_environ(
m=initial_metadata,
for_env=False,
skip_build_id=skip_build_id,
escape_backslash=True,
variant=variant,
)
environ = dict(os.environ)
environ.update(get_environ(m=initial_metadata, skip_build_id=skip_build_id))
ctx.update(
load_setup_py_data=partial(
load_setup_py_data,
m=initial_metadata,
recipe_dir=recipe_dir,
permit_undefined_jinja=permit_undefined_jinja,
),
# maintain old alias for backwards compatibility:
load_setuptools=partial(
load_setuptools,
m=initial_metadata,
recipe_dir=recipe_dir,
permit_undefined_jinja=permit_undefined_jinja,
),
load_npm=load_npm,
load_file_regex=partial(
load_file_regex,
config=config,
recipe_dir=recipe_dir,
permit_undefined_jinja=permit_undefined_jinja,
),
load_file_data=partial(
load_file_data,
config=config,
recipe_dir=recipe_dir,
permit_undefined_jinja=permit_undefined_jinja,
),
load_str_data=load_str_data,
installed=get_installed_packages(
os.path.join(config.host_prefix, "conda-meta")
),
pin_compatible=partial(
pin_compatible,
initial_metadata,
permit_undefined_jinja=permit_undefined_jinja,
bypass_env_check=bypass_env_check,
),
pin_subpackage=partial(
pin_subpackage,
initial_metadata,
permit_undefined_jinja=permit_undefined_jinja,
allow_no_other_outputs=allow_no_other_outputs,
skip_build_id=skip_build_id,
),
compiler=partial(
_target,
config=config,
permit_undefined_jinja=permit_undefined_jinja,
component="compiler",
),
stdlib=partial(
_target,
config=config,
permit_undefined_jinja=permit_undefined_jinja,
component="stdlib",
),
cdt=partial(cdt, config=config, permit_undefined_jinja=permit_undefined_jinja),
ccache=partial(
ccache, config=config, permit_undefined_jinja=permit_undefined_jinja
),
resolved_packages=partial(
resolved_packages,
initial_metadata,
permit_undefined_jinja=permit_undefined_jinja,
bypass_env_check=bypass_env_check,
),
time=time,
datetime=datetime,
environ=environ,
)
return ctx