diff --git a/bin/bump_kernel_tag_mods b/bin/bump_kernel_tag_mods index 898e540..693782d 100755 --- a/bin/bump_kernel_tag_mods +++ b/bin/bump_kernel_tag_mods @@ -12,9 +12,12 @@ if only_injected: do_spm_repo = "--no-spm-repo" not in args if not do_spm_repo: args.remove("--no-spm-repo") +keyslot = "--keyslot" in args +if keyslot: + args.remove("--keyslot") if len(args) < 3: - print("bump_kernel_tag_mods [--only-injected] [--no-spm-repo]") + print("bump_kernel_tag_mods [--only-injected] [--no-spm-repo] [--keyslot]") raise SystemExit(1) kernel_tag = args.pop(0) @@ -48,7 +51,10 @@ try: injected_atoms = [] for lst in (normal_pkgs, injected_pkgs): for pkg_id in lst: - atom_str = entropy.dep.remove_tag(repo.retrieveAtom(pkg_id)) + if keyslot: + atom_str = repo.retrieveKeySlotAggregated(pkg_id) + else: + atom_str = entropy.dep.remove_tag(repo.retrieveAtom(pkg_id)) if do_spm_repo: spm_repo = repo.retrieveSpmRepository(pkg_id) if spm_repo is not None: diff --git a/bin/bump_kernel_tag_mods_inject b/bin/bump_kernel_tag_mods_inject index 9828f95..304d20a 100755 --- a/bin/bump_kernel_tag_mods_inject +++ b/bin/bump_kernel_tag_mods_inject @@ -8,9 +8,12 @@ args = sys.argv[1:] do_spm_repo = "--no-spm-repo" not in args if not do_spm_repo: args.remove("--no-spm-repo") +keyslot = "--keyslot" in args +if keyslot: + args.remove("--keyslot") if len(args) < 3: - print("bump_kernel_tag_mods [--no-spm-repo]") + print("bump_kernel_tag_mods [--no-spm-repo] [--keyslot]") raise SystemExit(1) kernel_tag = args.pop(0) @@ -40,7 +43,10 @@ try: if spm_repo is not None: atom_str += "::" + spm_repo injected_atoms.append(atom_str) - injected_atoms = [entropy.dep.remove_tag(repo.retrieveAtom(pkg_id)) for pkg_id in pkg_ids] + if keyslot: + injected_atoms = [repo.retrieveKeySlotAggregated(pkg_id) for pkg_id in pkg_ids] + else: + injected_atoms = [entropy.dep.remove_tag(repo.retrieveAtom(pkg_id)) for pkg_id in pkg_ids] finally: srv.shutdown()