Include collector in network-config-analyzer package (#145)

Signed-off-by: Ziv Nevo <[email protected]>

collector/collect.py

@@ -1,21 +1,29 @@
 import subprocess
 
 
-get_ctx = 'kubectl config current-context'
-print(get_ctx)
-cmdline_process = subprocess.Popen(get_ctx.split(' '), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-ctx, err = cmdline_process.communicate()
-ctx_cluster = ctx.decode("utf-8").replace('/', '-').strip()
-print(ctx_cluster)
+def collect():
+    """
+    A small utility to collect all the network-related resources in the current cluster context
+    """
+    get_ctx = 'kubectl config current-context'
+    print(get_ctx)
+    cmdline_process = subprocess.Popen(get_ctx.split(' '), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    ctx, err = cmdline_process.communicate()
+    ctx_cluster = ctx.decode("utf-8").replace('/', '-').strip()
+    print(ctx_cluster)
 
-resources = ['pods', 'namespaces', 'services', 'networkpolicies', 'ingresses', 'ingressclasses', 'authorizationpolicies', 'destinationrules','gateways', 'serviceentries','sidecars','virtualservices' ]
+    resources = ['pods', 'namespaces', 'services', 'networkpolicies', 'ingresses', 'ingressclasses',
+                 'authorizationpolicies', 'destinationrules','gateways', 'serviceentries','sidecars','virtualservices' ]
 
-for resource in resources:
-    cmd = 'kubectl get ' + resource + ' -A -o yaml'
-    cmdline_list = cmd.split(' ')
-    print(cmd)
-    log = open(ctx_cluster + '_' + resource + '.yaml', 'a')
-    cmdline_process = subprocess.Popen(cmdline_list, stdout=log, stderr=subprocess.PIPE)
-    out, err = cmdline_process.communicate()
-    if len(err) > 0:
-        print(err)
+    for resource in resources:
+        cmd = 'kubectl get ' + resource + ' -A -o yaml'
+        cmdline_list = cmd.split(' ')
+        print(cmd)
+        log = open(ctx_cluster + '_' + resource + '.yaml', 'a')
+        cmdline_process = subprocess.Popen(cmdline_list, stdout=log, stderr=subprocess.PIPE)
+        out, err = cmdline_process.communicate()
+        if len(err) > 0:
+            print(err)
+
+if __name__ == "__main__":
+    collect()

setup.cfg

@@ -29,11 +29,14 @@ install_requires =
 python_requires = >=3.9
 
 [options.packages.find]
-include = nca*
+include = 
+    nca*
+    collector*
 
 [options.package_data]
 * = VERSION.txt
 
 [options.entry_points]
 console_scripts =
     nca = nca.nca_cli:nca_main
+    nca_collect = collector.collect:collect