From 8b15a16c0d815b1ead01423fcccbc20403704529 Mon Sep 17 00:00:00 2001
From: jyejare <jyejare@redhat.com>
Date: Wed, 28 Feb 2024 18:45:29 +0530
Subject: [PATCH] Finder/Reader to read the specific path from json file

---
 candore/__init__.py       |  8 ++++++++
 candore/cli.py            | 18 +++++++++++++++++
 candore/modules/finder.py | 41 +++++++++++++++++++++++++++++++++++++++
 3 files changed, 67 insertions(+)
 create mode 100644 candore/modules/finder.py

diff --git a/candore/__init__.py b/candore/__init__.py
index d8ac851..11bd0fa 100644
--- a/candore/__init__.py
+++ b/candore/__init__.py
@@ -1,6 +1,7 @@
 import asyncio  # noqa: F401
 import json
 from pathlib import Path
+from pprint import pprint
 
 import click
 
@@ -8,6 +9,7 @@
 from candore.modules.api_lister import APILister
 from candore.modules.comparator import Comparator
 from candore.modules.extractor import Extractor
+from candore.modules.finder import Finder
 from candore.modules.report import Reporting
 
 
@@ -58,3 +60,9 @@ def compare_entities(
         results = comp.compare_json(pre_file=pre_file, post_file=post_file)
         reporter = Reporting(results=results)
         reporter.generate_report(output_file=output, output_type=report_type)
+
+    def find_path(self, path, json_file, delimiter):
+        finder = Finder()
+        data = finder.find(path=path, json_file=json_file, delimiter=delimiter)
+        if data:
+            pprint(data)
diff --git a/candore/cli.py b/candore/cli.py
index 8898a6a..b8d1f1d 100644
--- a/candore/cli.py
+++ b/candore/cli.py
@@ -75,5 +75,23 @@ def compare(ctx, pre, post, output, report_type, record_evs):
     )
 
 
+@candore.command(help="JSON Reader for reading the specific path data from entities data file")
+@click.option(
+    "--path",
+    type=str,
+    help="The path to search the data from.\n"
+    "Path contents could divided by some delimiter.\n"
+    "e.g entity/5/description",
+)
+@click.option(
+    "--data-file", type=str, help="The data file from which to search the data on a given path"
+)
+@click.option("--delimiter", type=str, default='/', help="Settings file path. Default is '/'")
+@click.pass_context
+def reader(ctx, path, data_file, delimiter):
+    candore_obj = ctx.parent.candore
+    candore_obj.find_path(path=path, json_file=data_file, delimiter=delimiter)
+
+
 if __name__ == "__main__":
     candore()
diff --git a/candore/modules/finder.py b/candore/modules/finder.py
new file mode 100644
index 0000000..f475280
--- /dev/null
+++ b/candore/modules/finder.py
@@ -0,0 +1,41 @@
+import json
+
+
+class Finder:
+    def read_json(self, jsonfile=None):
+        jdata = None
+        with open(jsonfile) as file:
+            jdata = json.load(file)
+        return jdata
+
+    def json_iterator(self, data=None, path=None, delimiter=None):
+        if not data:
+            return data
+        if path:
+            path_contents = path.strip().split(delimiter)
+            first_path = path_contents[0]
+            remaining_path = f'{delimiter}'.join(path_contents[1:])
+
+            if isinstance(data, dict):
+                if first_path in data:
+                    return self.json_iterator(data.get(first_path, None), remaining_path, delimiter)
+                else:
+                    print(
+                        f'Oh!O! {first_path} is not found in given path, please correct the path!'
+                    )
+            elif isinstance(data, list):
+                for index, element_data in enumerate(data):
+                    if isinstance(element_data, dict):
+                        if 'id' in element_data and str(element_data['id']) == first_path:
+                            return self.json_iterator(element_data, remaining_path, delimiter)
+                    elif isinstance(element_data, list):
+                        if first_path in element_data:
+                            return self.json_iterator(
+                                element_data[index], remaining_path, delimiter
+                            )
+        else:
+            return data
+
+    def find(self, path=None, json_file=None, delimiter='/'):
+        jdata = self.read_json(json_file)
+        return self.json_iterator(jdata, path=path, delimiter=delimiter)