Update puppet_class to take multiple classes

alexmuller · alexmuller · commit e222520ab888 · 2015-08-24T16:58:25.000+01:00
Similar to b2f93a9, this commit uses the same logic when selecting `puppet_class`.
diff --git a/fabfile.py b/fabfile.py
@@ -271,10 +271,11 @@ def klass(*class_names):
 @task
 @serial
 @hosts('localhost')
-def puppet_class(class_name):
+def puppet_class(*class_names):
     """Select all machines which include a given puppet class"""
-    env.roledefs.fetch_puppet_class(class_name)
-    env.hosts.extend(env.roledefs['puppet_class-%s' % class_name]())
+    for class_name in class_names:
+        env.roledefs.fetch_puppet_class(class_name)
+        env.hosts.extend(env.roledefs['puppet_class-%s' % class_name]())
 
 @task
 @hosts('localhost')
