stacks: update environment add/remove algorithms

lib/spack/spack/environment.py

@@ -674,39 +674,29 @@ def add(self, user_spec, list_name='specs'):
             elif not spack.repo.path.exists(spec.name):
                 raise SpackEnvironmentError('no such package: %s' % spec.name)
 
-        added = False
-        existing = False
-        for i, (name, speclist) in enumerate(self.read_specs.items()):
-            # Iterate over all named lists from an OrderedDict()
-            if name == list_name:
-                # We need to modify this list
-                # TODO: Add conditional which reimplements name-level checking
-                existing = str(spec) in speclist.yaml_list
+        list_to_change = self.read_specs[list_name]
+        existing = str(spec) in list_to_change.yaml_list
         if not existing:
-                    speclist.add(str(spec))
-                    added = True
-            elif added:
-                # We've already modified a list, so all later lists need to
-                # have their references updated.
+            list_to_change.add(str(spec))
+            index = list(self.read_specs.keys()).index(list_name)
+
+            for i, (name, speclist) in enumerate(
+                    list(self.read_specs.items())[index + 1:], index + 1):
                 new_reference = dict((n, self.read_specs[n])
                                      for n in list(self.read_specs.keys())[:i])
                 speclist.update_reference(new_reference)
+
         return bool(not existing)
 
     def remove(self, query_spec, list_name='specs', force=False):
         """Remove specs from an environment that match a query_spec"""
         query_spec = Spec(query_spec)
 
-        removed = False
-        for i, (name, speclist) in enumerate(self.read_specs.items()):
-            # Iterate over all named lists from an OrderedDict()
-            if name == list_name:
-                # We need to modify this list
-                # try abstract specs first
+        list_to_change = self.read_specs[list_name]
         matches = []
 
         if not query_spec.concrete:
-                    matches = [s for s in speclist if s.satisfies(query_spec)]
+            matches = [s for s in list_to_change if s.satisfies(query_spec)]
 
         if not matches:
             # concrete specs match against concrete specs in the env
@@ -722,9 +712,8 @@ def remove(self, query_spec, list_name='specs', force=False):
                 "Not found: {0}".format(query_spec))
 
         for spec in matches:
-                    if spec in speclist:
-                        speclist.remove(spec)
-                        removed = True
+            if spec in list_to_change:
+                list_to_change.remove(spec)
 
             if force and spec in self.concretized_user_specs:
                 i = self.concretized_user_specs.index(spec)
@@ -733,11 +722,10 @@ def remove(self, query_spec, list_name='specs', force=False):
                 dag_hash = self.concretized_order[i]
                 del self.concretized_order[i]
                 del self.specs_by_hash[dag_hash]
-                        removed = True
 
-            elif removed:
-                # We've already modified one list, so all later lists need
-                # their references updated.
+        index = list(self.read_specs.keys()).index(list_name)
+        for i, (name, speclist) in enumerate(
+                list(self.read_specs.items())[index + 1:], index + 1):
             new_reference = dict((n, self.read_specs[n])
                                  for n in list(self.read_specs.keys())[:i])
             speclist.update_reference(new_reference)

lib/spack/spack/spec_list.py

@@ -136,7 +136,7 @@ def _expand_references(self, yaml):
                         return ret
                     else:
                         msg = 'SpecList %s refers to ' % self.name
-                        msg = 'named list %s ' % name
+                        msg += 'named list %s ' % name
                         msg += 'which does not appear in its reference dict'
                         raise UndefinedReferenceError(msg)
             # No references in this

lib/spack/spack/test/cmd/env.py

@@ -827,6 +827,33 @@ def test_stack_yaml_remove_from_list(tmpdir):
         assert Spec('callpath') in test.user_specs
 
 
+def test_stack_yaml_remove_from_list_force(tmpdir):
+    filename = str(tmpdir.join('spack.yaml'))
+    with open(filename, 'w') as f:
+        f.write("""\
+env:
+  definitions:
+    - packages: [mpileaks, callpath]
+  specs:
+    - matrix:
+        - [$packages]
+        - [^mpich, ^zmpi]
+""")
+    with tmpdir.as_cwd():
+        env('create', 'test', './spack.yaml')
+        with ev.read('test'):
+            concretize()
+            remove('-f', '-l', 'packages', 'mpileaks')
+            find_output = find('-c')
+
+        print find_output
+        assert False
+        test = ev.read('test')
+
+        assert Spec('mpileaks') not in test.user_specs
+        assert Spec('callpath') in test.user_specs
+
+
 def test_stack_yaml_attempt_remove_from_matrix(tmpdir):
     filename = str(tmpdir.join('spack.yaml'))
     with open(filename, 'w') as f: