Add kubernetes inventory.

Author: leahneukirchen

pyinfra/api/inventories/__init__.py

@@ -0,0 +1,26 @@
+from pyinfra.api.exceptions import InventoryError
+from pyinfra.api.connectors.util import run_local_process, split_combined_output
+
+def get_pods(selector, namespace='default', all_namespaces=False, container=None):
+    if all_namespaces:
+        ns = ['-A']
+    else:
+        ns = ['-n', namespace]
+
+    return_code, combined_output = run_local_process(
+        ['"$@"', '-',
+         'kubectl', 'get', 'pods', *ns, '-l', selector, '--template',
+         r'{{range .items}}@kubernetes/' +
+         r'{{.metadata.namespace}}/{{.metadata.name}}{{"\n"}}{{end}}'
+        ]
+    )
+    stdout, stderr = split_combined_output(combined_output)
+
+    if return_code == 0:
+        data = {}
+        if container:
+            data['container'] = container
+        return list(map(lambda s: (s, data), stdout))
+    else:
+        raise InventoryError('kubectl failed (status {0}): {1}'.
+                             format(return_code, '\n'.join(stderr)))