Support for only-cross-site

Author: Carlos Konstanski

examples/nagios_.jinja2

@@ -4,8 +4,7 @@
 {% set named_object = object_name in named_objects %}
 {% for resource in resources %}
 {% if resource.exported %}
-{% if "no-cross-site" not in resource.tags %}
-{% if "garb" not in resource.name %}
+{% if "only-cross-site" not in resource.tags or localsite == "false" %}
 define {{object_name}} {
     {% if named_object %}
     {{object_name}}_name {{resource.name}}
@@ -29,5 +28,4 @@ define {{object_name}} {
 }
 {% endif %}
 {% endif %}
-{% endif %}
 {% endfor %}

puppetdb_stencil.py

@@ -24,7 +24,7 @@
                                  loader=LOADER, extensions=EXTENSIONS)
 
 
-def render_resources(database, resource_type, template_names):
+def render_resources(database, resource_type, localsite, template_names):
     """
     Render resources of the given type. They are queried from the given
     database and rendered using the first template from template_names that can
@@ -38,7 +38,7 @@ def render_resources(database, resource_type, template_names):
     else:
         return template.render(resource_type=resource_type,
                                resources=resources, metaparams=METAPARAMS,
-                               env=os.environ)
+                               env=os.environ, localsite=localsite)
 
 
 def main():
@@ -51,6 +51,7 @@ def main():
     parser.add_argument('--debug', '-d', action='store_true')
     parser.add_argument('--host', '-H', default='localhost')
     parser.add_argument('--port', '-p', default='8080')
+    parser.add_argument('--localsite', '-l', default='true')
 
     args = parser.parse_args()
     logging.basicConfig(level=logging.DEBUG if args.debug else logging.WARN)
@@ -61,7 +62,7 @@ def main():
         templates = ['{0}.jinja2'.format(resource_type)]
         if args.templates:
             templates += args.templates
-        print(render_resources(database, resource_type, templates))
+        print(render_resources(database, resource_type, args.localsite, templates))
 
 
 if __name__ == '__main__':