def do_test(dirname):
envdir = os.path.join(dirname, "myenv")
print('CONDA_EXE: {}'.format(os.environ.get('CONDA_EXE')))
# originally we did not specify a python version here, but we
# needed to add it with python 3.8 was released because a compatible
# version of ipython had not been created yet.
conda_api.create(prefix=envdir, pkgs=['python<3.8'])
assert os.path.isdir(envdir)
assert os.path.isdir(os.path.join(envdir, "conda-meta"))
assert os.path.exists(os.path.join(envdir, PYTHON_BINARY))
# test that if it exists we can't create it again
with pytest.raises(conda_api.CondaEnvExistsError) as excinfo:
conda_api.create(prefix=envdir, pkgs=['python'])
assert 'already exists' in repr(excinfo.value)
# test that we can install a package
assert not os.path.exists(os.path.join(envdir, IPYTHON_BINARY))
conda_api.install(prefix=envdir, pkgs=['ipython'])
assert os.path.exists(os.path.join(envdir, IPYTHON_BINARY))
# test that we can remove it again
conda_api.remove(prefix=envdir, pkgs=['ipython'])
assert not os.path.exists(os.path.join(envdir, IPYTHON_BINARY))
def do_test(dirname):
envdir = os.path.join(dirname, "myenv")
conda_api.create(prefix=envdir, pkgs=['python'])
with pytest.raises(TypeError) as excinfo:
conda_api.install(prefix=envdir, pkgs=[])
assert 'must specify a list' in repr(excinfo.value)
def fix_environment_deviations(self, prefix, spec, deviations=None, create=True):
if deviations is None:
deviations = self.find_environment_deviations(prefix, spec)
if deviations.unfixable:
raise CondaManagerError("Unable to update environment at %s" % prefix)
if os.path.isdir(os.path.join(prefix, 'conda-meta')):
to_update = list(set(deviations.missing_packages + deviations.wrong_version_packages))
if len(to_update) > 0:
specs = spec.specs_for_conda_package_names(to_update)
assert len(specs) == len(to_update)
spec.apply_pins(prefix, specs)
try:
conda_api.install(
prefix=prefix,
pkgs=specs,
channels=spec.channels,
stdout_callback=self._on_stdout,
stderr_callback=self._on_stderr)
except conda_api.CondaError as e:
raise CondaManagerError("Failed to install packages: {}: {}".format(", ".join(specs), str(e)))
finally:
spec.remove_pins(prefix)
elif create:
# Create environment from scratch
command_line_packages = set(spec.conda_packages_for_create)
# conda won't let us create a completely empty environment
if len(command_line_packages) == 0:
command_line_packages = set(['python'])
try:
conda_api.create(
prefix=prefix,
pkgs=list(command_line_packages),
channels=spec.channels,
stdout_callback=self._on_stdout,
stderr_callback=self._on_stderr)
except conda_api.CondaError as e:
raise CondaManagerError("Failed to create environment at %s: %s" % (prefix, str(e)))
else:
raise CondaManagerError("Conda environment at %s does not exist" % (prefix))
# now add pip if needed
missing = list(deviations.missing_pip_packages)
if len(missing) > 0:
specs = spec.specs_for_pip_package_names(missing)
assert len(specs) == len(missing)
try:
pip_api.install(prefix=prefix, pkgs=specs)
except pip_api.PipError as e:
raise CondaManagerError("Failed to install missing pip packages: {}: {}".format(
", ".join(missing), str(e)))
# write a file to tell us we can short-circuit next time
self._write_timestamp_file(prefix, spec)
def test_conda_install_gets_channels(monkeypatch):
def mock_call_conda(extra_args):
assert [
'install', '--yes', '--quiet', '--prefix', '/prefix', '--channel',
'foo', 'python'
] == extra_args
monkeypatch.setattr('anaconda_project.internal.conda_api._call_conda',
mock_call_conda)
conda_api.install(prefix='/prefix', pkgs=['python'], channels=['foo'])
def test_conda_install_gets_channels(monkeypatch):
def mock_call_conda(extra_args,
json_mode=False,
platform=None,
stdout_callback=None,
stderr_callback=None):
assert [
'install', '--yes', '--prefix', '/prefix', '--channel', 'foo',
'python'
] == extra_args
monkeypatch.setattr('anaconda_project.internal.conda_api._call_conda',
mock_call_conda)
conda_api.install(prefix='/prefix', pkgs=['python'], channels=['foo'])
def do_test(dirname):
envdir = os.path.join(dirname, "myenv")
# don't specify a python version so we use the one we already have
# in the root env, otherwise this might take forever.
conda_api.create(prefix=envdir, pkgs=['python'])
assert os.path.isdir(envdir)
assert os.path.isdir(os.path.join(envdir, "conda-meta"))
assert os.path.exists(os.path.join(envdir, PYTHON_BINARY))
# test that if it exists we can't create it again
with pytest.raises(conda_api.CondaEnvExistsError) as excinfo:
conda_api.create(prefix=envdir, pkgs=['python'])
assert 'already exists' in repr(excinfo.value)
# test that we can install a package
assert not os.path.exists(os.path.join(envdir, IPYTHON_BINARY))
conda_api.install(prefix=envdir, pkgs=['ipython'])
assert os.path.exists(os.path.join(envdir, IPYTHON_BINARY))
# test that we can remove it again
conda_api.remove(prefix=envdir, pkgs=['ipython'])
assert not os.path.exists(os.path.join(envdir, IPYTHON_BINARY))
def fix_environment_deviations(self,
prefix,
spec,
deviations=None,
create=True):
if deviations is None:
deviations = self.find_environment_deviations(prefix, spec)
if deviations.unfixable:
raise CondaManagerError("Unable to update environment at %s" %
prefix)
conda_meta = os.path.join(prefix, 'conda-meta')
packed = os.path.join(conda_meta, '.packed')
install_pip = True
if os.path.isdir(conda_meta) and os.path.exists(packed):
with open(packed) as f:
packed_arch = f.read().strip()
matched = packed_arch == conda_api.current_platform()
if matched:
if 'win' in conda_api.current_platform():
unpack_script = [
'python',
os.path.join(prefix, 'Scripts',
'conda-unpack-script.py')
]
else:
unpack_script = os.path.join(prefix, 'bin', 'conda-unpack')
try:
subprocess.check_call(unpack_script)
os.remove(packed)
install_pip = False
except (subprocess.CalledProcessError, OSError) as e:
self._log_info(
'Warning: conda-unpack could not be run: \n{}\n'
'The environment will be recreated.'.format(str(e)))
create = True
shutil.rmtree(prefix)
else:
self._log_info(
'Warning: The unpacked env does not match the current architecture. '
'It will be recreated.')
create = True
shutil.rmtree(prefix)
if os.path.isdir(conda_meta):
to_update = list(
set(deviations.missing_packages +
deviations.wrong_version_packages))
if len(to_update) > 0:
specs = spec.specs_for_conda_package_names(to_update)
assert len(specs) == len(to_update)
spec.apply_pins(prefix, specs)
try:
conda_api.install(prefix=prefix,
pkgs=specs,
channels=spec.channels,
stdout_callback=self._on_stdout,
stderr_callback=self._on_stderr)
except conda_api.CondaError as e:
raise CondaManagerError(
"Failed to install packages: {}: {}".format(
", ".join(specs), str(e)))
finally:
spec.remove_pins(prefix)
elif create:
# Create environment from scratch
command_line_packages = set(spec.conda_packages_for_create)
try:
conda_api.create(prefix=prefix,
pkgs=list(command_line_packages),
channels=spec.channels,
stdout_callback=self._on_stdout,
stderr_callback=self._on_stderr)
except conda_api.CondaError as e:
raise CondaManagerError(
"Failed to create environment at %s: %s" %
(prefix, str(e)))
else:
raise CondaManagerError("Conda environment at %s does not exist" %
(prefix))
# now add pip if needed
missing = list(deviations.missing_pip_packages)
if (len(missing) > 0) and install_pip:
specs = spec.specs_for_pip_package_names(missing)
assert len(specs) == len(missing)
try:
pip_api.install(prefix=prefix,
pkgs=specs,
stdout_callback=self._on_stdout,
stderr_callback=self._on_stderr)
except pip_api.PipError as e:
raise CondaManagerError(
"Failed to install missing pip packages: {}: {}".format(
", ".join(missing), str(e)))
# write a file to tell us we can short-circuit next time
self._write_timestamp_file(prefix, spec)
