Edited the search() class

Author: lainq

src/arguments/search.py

@@ -26,33 +26,39 @@ def __init__(self, arguments):
 
     def search_args(self):
         if self.arguments.search:
-            queries = ["What do you want to search", "Tags"]
-            query_solutions = []
-
-            # ask quesiton
-            for each_query in queries:
-                # Be careful if there are
-                # KeyBpard Interupts or EOErrors
-                try:
-                    prompt = Prompt(str(each_query)).prompt()
-                except:
-                    sys.exit()
-
-                query_solutions.append(prompt)
-
-            question, tags = query_solutions[0], query_solutions[1]
-            json_output = self.utility_object.make_request(question, tags)
-            questions = self.utility_object.get_que(json_output)
-            if questions == []:
-                # evoke an error
-                search_error = SearchError("No answer found",
-                                           "Please try reddit")
-            else:
-                self.utility_object.get_ans(questions)
+            self.search_for_results()
 
         elif self.arguments.new:
             url = "https://stackoverflow.com/questions/ask"
             if type(self.arguments.new) == str:
                 webbrowser.open(f"{url}?title={self.arguments.new}")
             else:
                 webbrowser.open(url)
+
+    def search_for_results(self, save=False):
+        queries = ["What do you want to search", "Tags"]
+        query_solutions = []
+
+        # ask quesiton
+        for each_query in queries:
+            # Be careful if there are
+            # KeyBpard Interupts or EOErrors
+            try:
+                prompt = Prompt(str(each_query)).prompt()
+            except:
+                sys.exit()
+
+            query_solutions.append(prompt)
+
+        question, tags = query_solutions[0], query_solutions[1]
+        json_output = self.utility_object.make_request(question, tags)
+        questions = self.utility_object.get_que(json_output)
+        if questions == []:
+            # evoke an error
+            search_error = SearchError("No answer found",
+                                        "Please try reddit")
+        else:
+            data = self.utility_object.get_ans(questions)
+
+            if save:
+                SaveSearchResults(data)

src/arguments/utility.py

@@ -4,6 +4,7 @@
 import sys as sys
 
 from .error import SearchError
+from .save import SaveSearchResults
 from .markdown import MarkdownRenderer
 
 console = Console()
@@ -44,7 +45,7 @@ def get_que(self, json_data):
         return que_id
 
     def get_ans(self, questions_list):
-        # ans = []
+        ans = []
         for questions in range(1):
             try:
                 resp = requests.get(
@@ -83,3 +84,5 @@ def get_ans(self, questions_list):
                         continue
 
                     console.print(output_text)
+            ans.append(json_ans_data["items"])
+        return ans