def test_register_unregister_location_env(self):
user_environments_txt_file = get_user_environments_txt_file()
if (not os.path.exists(user_environments_txt_file)
or user_environments_txt_file == os.devnull):
pytest.skip('user environments.txt file {} does not exist'.format(
user_environments_txt_file))
gascon_location = join(self.prefix, 'gascon')
touch(join(gascon_location, PREFIX_MAGIC_FILE), mkdir=True)
assert gascon_location not in list_all_known_prefixes()
touch(user_environments_txt_file, mkdir=True, sudo_safe=True)
register_env(gascon_location)
assert gascon_location in yield_lines(user_environments_txt_file)
assert len(
tuple(x for x in yield_lines(user_environments_txt_file)
if paths_equal(gascon_location, x))) == 1
register_env(gascon_location) # should be completely idempotent
assert len(
tuple(x for x in yield_lines(user_environments_txt_file)
if x == gascon_location)) == 1
unregister_env(gascon_location)
assert gascon_location not in list_all_known_prefixes()
unregister_env(gascon_location) # should be idempotent
assert gascon_location not in list_all_known_prefixes()
def test_register_unregister_location_env(self):
gascon_location = join(self.prefix, 'gascon')
touch(join(gascon_location, PREFIX_MAGIC_FILE), mkdir=True)
assert gascon_location not in list_all_known_prefixes()
touch(USER_ENVIRONMENTS_TXT_FILE, mkdir=True, sudo_safe=True)
register_env(gascon_location)
assert gascon_location in yield_lines(USER_ENVIRONMENTS_TXT_FILE)
assert len(
tuple(x for x in yield_lines(USER_ENVIRONMENTS_TXT_FILE)
if paths_equal(gascon_location, x))) == 1
register_env(gascon_location) # should be completely idempotent
assert len(
tuple(x for x in yield_lines(USER_ENVIRONMENTS_TXT_FILE)
if x == gascon_location)) == 1
unregister_env(gascon_location)
assert gascon_location not in list_all_known_prefixes()
unregister_env(gascon_location) # should be idempotent
assert gascon_location not in list_all_known_prefixes()
def test_register_unregister_location_env(self):
user_environments_txt_file = get_user_environments_txt_file()
if (not os.path.exists(user_environments_txt_file)
or user_environments_txt_file == os.devnull):
pytest.skip('user environments.txt file {} does not exist'.format(user_environments_txt_file))
gascon_location = join(self.prefix, 'gascon')
touch(join(gascon_location, PREFIX_MAGIC_FILE), mkdir=True)
assert gascon_location not in list_all_known_prefixes()
touch(user_environments_txt_file, mkdir=True, sudo_safe=True)
register_env(gascon_location)
assert gascon_location in yield_lines(user_environments_txt_file)
assert len(tuple(x for x in yield_lines(user_environments_txt_file) if paths_equal(gascon_location, x))) == 1
register_env(gascon_location) # should be completely idempotent
assert len(tuple(x for x in yield_lines(user_environments_txt_file) if x == gascon_location)) == 1
unregister_env(gascon_location)
assert gascon_location not in list_all_known_prefixes()
unregister_env(gascon_location) # should be idempotent
assert gascon_location not in list_all_known_prefixes()
def remove(args, parser):
if not (args.all or args.package_names):
raise CondaValueError("no package names supplied,\n"
' try "mamba remove -h" for more details')
prefix = context.target_prefix
check_non_admin()
init_api_context()
if args.all and prefix == context.default_prefix:
raise CondaEnvironmentError("cannot remove current environment. \
deactivate and run mamba remove again")
if args.all and path_is_clean(prefix):
# full environment removal was requested, but environment doesn't exist anyway
return 0
if args.all:
if prefix == context.root_prefix:
raise CondaEnvironmentError(
"cannot remove root environment,\n"
" add -n NAME or -p PREFIX option")
print("\nRemove all packages in environment %s:\n" % prefix,
file=sys.stderr)
if "package_names" in args:
stp = PrefixSetup(
target_prefix=prefix,
unlink_precs=tuple(PrefixData(prefix).iter_records()),
link_precs=(),
remove_specs=(),
update_specs=(),
neutered_specs=(),
)
txn = UnlinkLinkTransaction(stp)
try:
handle_txn(txn, prefix, args, False, True)
except PackagesNotFoundError:
print(
"No packages found in %s. Continuing environment removal" %
prefix)
rm_rf(prefix, clean_empty_parents=True)
unregister_env(prefix)
return
else:
if args.features:
specs = tuple(
MatchSpec(track_features=f) for f in set(args.package_names))
else:
specs = [s for s in specs_from_args(args.package_names)]
if not context.quiet:
print("Removing specs: {}".format(
[s.conda_build_form() for s in specs]))
installed_json_f, installed_pkg_recs = get_installed_jsonfile(prefix)
mamba_solve_specs = [s.conda_build_form() for s in specs]
solver_options.append((api.SOLVER_FLAG_ALLOW_UNINSTALL, 1))
if context.channel_priority is ChannelPriority.STRICT:
solver_options.append((api.SOLVER_FLAG_STRICT_REPO_PRIORITY, 1))
pool = api.Pool()
repos = []
# add installed
if use_mamba_experimental:
prefix_data = api.PrefixData(context.target_prefix)
prefix_data.load()
repo = api.Repo(pool, prefix_data)
repos.append(repo)
else:
repo = api.Repo(pool, "installed", installed_json_f.name, "")
repo.set_installed()
repos.append(repo)
solver = api.Solver(pool, solver_options)
history = api.History(context.target_prefix)
history_map = history.get_requested_specs_map()
solver.add_jobs(
[ms.conda_build_form() for ms in history_map.values()],
api.SOLVER_USERINSTALLED,
)
solver.add_jobs(mamba_solve_specs,
api.SOLVER_ERASE | api.SOLVER_CLEANDEPS)
success = solver.solve()
if not success:
print(solver.problems_to_str())
exit_code = 1
return exit_code
package_cache = api.MultiPackageCache(context.pkgs_dirs)
transaction = api.Transaction(
solver, package_cache,
PackageCacheData.first_writable().pkgs_dir)
downloaded = transaction.prompt(repos)
if not downloaded:
exit(0)
mmb_specs, to_link, to_unlink = transaction.to_conda()
transaction.log_json()
specs_to_add = [MatchSpec(m) for m in mmb_specs[0]]
specs_to_remove = [MatchSpec(m) for m in mmb_specs[1]]
conda_transaction = to_txn(
specs_to_add,
specs_to_remove,
prefix,
to_link,
to_unlink,
installed_pkg_recs,
)
handle_txn(conda_transaction, prefix, args, False, True)
def remove(args, parser):
if not (args.all or args.package_names):
raise CondaValueError('no package names supplied,\n'
' try "mamba remove -h" for more details')
prefix = context.target_prefix
check_non_admin()
init_api_context()
if args.all and prefix == context.default_prefix:
raise CondaEnvironmentError("cannot remove current environment. \
deactivate and run mamba remove again")
if args.all and path_is_clean(prefix):
# full environment removal was requested, but environment doesn't exist anyway
return 0
if args.all:
if prefix == context.root_prefix:
raise CondaEnvironmentError(
'cannot remove root environment,\n'
' add -n NAME or -p PREFIX option')
print("\nRemove all packages in environment %s:\n" % prefix,
file=sys.stderr)
if 'package_names' in args:
stp = PrefixSetup(
target_prefix=prefix,
unlink_precs=tuple(PrefixData(prefix).iter_records()),
link_precs=(),
remove_specs=(),
update_specs=(),
neutered_specs=(),
)
txn = UnlinkLinkTransaction(stp)
try:
handle_txn(txn, prefix, args, False, True)
except PackagesNotFoundError:
print(
"No packages found in %s. Continuing environment removal" %
prefix)
rm_rf(prefix, clean_empty_parents=True)
unregister_env(prefix)
return
else:
if args.features:
specs = tuple(
MatchSpec(track_features=f) for f in set(args.package_names))
else:
specs = [s for s in specs_from_args(args.package_names)]
if not context.quiet:
print("Removing specs: {}".format(
[s.conda_build_form() for s in specs]))
installed_json_f = get_installed_jsonfile(prefix)
mamba_solve_specs = [s.conda_build_form() for s in specs]
solver_options.append((api.SOLVER_FLAG_ALLOW_UNINSTALL, 1))
pool = api.Pool()
repos = []
# add installed
if use_mamba_experimental:
prefix_data = api.PrefixData(context.target_prefix)
prefix_data.load()
repo = api.Repo(pool, prefix_data)
repos.append(repo)
else:
repo = api.Repo(pool, "installed", installed_json_f.name, "")
repo.set_installed()
repos.append(repo)
solver = api.Solver(pool, solver_options)
solver.add_jobs(mamba_solve_specs, api.SOLVER_ERASE)
success = solver.solve()
if not success:
print(solver.problems_to_str())
exit_code = 1
return exit_code
transaction = api.Transaction(solver)
to_link, to_unlink = transaction.to_conda()
transaction.log_json()
conda_transaction = to_txn((), specs, prefix, to_link, to_unlink)
handle_txn(conda_transaction, prefix, args, False, True)
def remove(args, parser):
if not (args.all or args.package_names):
raise CondaValueError('no package names supplied,\n'
' try "mamba remove -h" for more details')
prefix = context.target_prefix
check_non_admin()
if args.all and prefix == context.default_prefix:
raise CondaEnvironmentError("cannot remove current environment. \
deactivate and run mamba remove again")
if args.all and path_is_clean(prefix):
# full environment removal was requested, but environment doesn't exist anyway
return 0
if args.all:
if prefix == context.root_prefix:
raise CondaEnvironmentError(
'cannot remove root environment,\n'
' add -n NAME or -p PREFIX option')
print("\nRemove all packages in environment %s:\n" % prefix,
file=sys.stderr)
if 'package_names' in args:
stp = PrefixSetup(
target_prefix=prefix,
unlink_precs=tuple(PrefixData(prefix).iter_records()),
link_precs=(),
remove_specs=(),
update_specs=(),
)
txn = UnlinkLinkTransaction(stp)
handle_txn(txn, prefix, args, False, True)
rm_rf(prefix, clean_empty_parents=True)
unregister_env(prefix)
return
else:
if args.features:
specs = tuple(
MatchSpec(track_features=f) for f in set(args.package_names))
else:
specs = [s for s in specs_from_args(args.package_names)]
if not context.quiet:
print("Removing specs: {}".format(specs))
channel_urls = ()
subdirs = ()
installed_json_f = get_installed_jsonfile(prefix)
mamba_solve_specs = [s.conda_build_form() for s in specs]
solver_options.append((api.SOLVER_FLAG_ALLOW_UNINSTALL, 1))
to_link, to_unlink = api.solve([], installed_json_f.name,
mamba_solve_specs, solver_options,
api.SOLVER_ERASE, False, context.quiet)
conda_transaction = to_txn(specs, prefix, to_link, to_unlink)
handle_txn(conda_transaction, prefix, args, False, True)
