Merge pull request alphagov#50 from alphagov/puppet_loadhosts_by_classes

rjw1 · rjw1 · commit f44435e08ac5 · 2014-07-14T10:22:24.000+01:00
[#43121095] Add puppet.loadhosts task
diff --git a/puppet.py b/puppet.py
@@ -1,5 +1,13 @@
 from fabric.api import *
 
+@task
+def loadhosts(*classnames):
+    """Load hosts that include given Puppet class(es)"""
+    classnames = ','.join(classnames)
+    with hide('running', 'stdout'):
+          with settings(host_string=env.gateway, gateway=None):
+              env.hosts = run('govuk_node_list -C %s' % classnames).splitlines()
+
 def puppet(*args):
     sudo('govuk_puppet %s' % ' '.join(args))
 
