diff options
| -rw-r--r-- | pym/portage/sets/__init__.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/pym/portage/sets/__init__.py b/pym/portage/sets/__init__.py index 29dc60c07..3e1fbdb2e 100644 --- a/pym/portage/sets/__init__.py +++ b/pym/portage/sets/__init__.py @@ -77,9 +77,12 @@ class PackageSet(object): else: return "" - def _updateAtomMap(self): - self._atommap.clear() - for a in self._atoms: + def _updateAtomMap(self, atoms=None): + """Update self._atommap for specific atoms or all atoms.""" + if not atoms: + self._atommap.clear() + atoms = self._atoms + for a in atoms: cp = dep_getkey(a) self._atommap.setdefault(cp, set()) self._atommap[cp].add(a) @@ -118,7 +121,7 @@ class EditablePackageSet(PackageSet): def update(self, atoms): self.getAtoms() self._atoms.update(atoms) - self._updateAtomMap() + self._updateAtomMap(atoms=atoms) self.write() def add(self, atom): |