Merge pull request #108 from harshakhmk/delete-endpoint

Add delete endpoint for dynamic #108

Author: GouravSardana

main.py

@@ -50,6 +50,10 @@
                     help="Make a GET request to an API",
                     action='store_true')
 
+parser.add_argument("-DELETE",
+                     help = "Make a DELETE request to an API",
+                     action = 'store_true')
+
 parser.add_argument("-p",
                     "--playbook",
                     help="View and organise the playbook",

src/arguments/api_test.py

@@ -4,9 +4,10 @@
 class ApiTesting():
     default_url = "https://127.0.0.1:8000"
     default_headers = {}
-    # Make GET request
+    invalid_schema_message = "Check whether the URL is valid or check if" + "the localhost server is active or not"
+    # fetches the input data for making a request
     @classmethod
-    def get_request(cls):
+    def fetch_input_url(cls):
         request_url = cls.default_url
         request_headers = cls.default_headers
         input_url = input('Enter URL: ')
@@ -37,28 +38,67 @@ def get_request(cls):
             request_url += endpoint
 
         print("Trying ...\u26A1")
+        return {
+            "request_url" : request_url,
+            "request_headers" : request_headers,
+        }
 
+    #saves the json response into a file
+    @classmethod
+    def save_response_data(cls,response_data):
+        store_data = input('Store response data? (Y/N): ')
+        if(store_data.lower() == 'y'):
+                filename = input("Enter a filename (response_data.json)")
+                if filename == '':
+                    filename = 'response_data.json'
+                with open(filename, 'w') as jsonFile:
+                    json.dump(response_data, jsonFile, indent=4)
+                print(f"Response data stored in {filename}")
+        elif(store_data.lower()) == 'n':
+            print(f"You have entered {store_data}, So the response is not saved")
+        else:
+            print(f"You have entered {store_data}, please enter either Y or N")
+            cls.save_response_data(response_data)
+    # formats the response data and prints it in json on console
+    @classmethod
+    def print_response_json(cls,response):
+        print(f"Reponse Status Code: {response.status_code}")
+        response_data = json.loads(response.content)
+        parsed_json = json.dumps(response_data, indent=4)
+        output_json = highlight(parsed_json, lexers.JsonLexer(),
+                                    formatters.TerminalFormatter())
+        print(output_json)
+
+    # Make GET request
+    @classmethod
+    def get_request(cls):
+        request_data = cls.fetch_input_url()
         # Make GET request and store the response in response_data.json
         try:
-            response = requests.get(request_url, headers=request_headers)
-            print(f"Reponse Status Code: {response.status_code}")
+            response = requests.get(request_data["request_url"], headers= request_data["request_headers"])
+            cls.print_response_json(response)
             response_data = json.loads(response.content)
-            parsed_json = json.dumps(response_data, indent=4)
-            output_json = highlight(parsed_json, lexers.JsonLexer(),
-                                    formatters.TerminalFormatter())
-            print(output_json)
+            cls.save_response_data(response_data)
 
-            store_data = input('Store response data? (Y/N): ')
-            if(store_data == 'Y' or store_data == 'y'):
-                with open('response_data.json', 'w') as jsonFile:
-                    json.dump(response_data, jsonFile, indent=4)
-                print("Response data stored in response_data.json")
+        except requests.exceptions.InvalidSchema:
+            print(cls.invalid_schema_message)
+        except Exception as exception_obj:
+            print(exception_obj)
+    # Make a delete request
+    @classmethod
+    def delete_request(cls):
+        # request_data contains dictionary of inputs entered by user
+        request_data = cls.fetch_input_url()
+        try:
+            response = requests.delete(request_data["request_url"], headers= request_data["request_headers"])
+            cls.print_response_json(response)
+            response_data = json.loads(response.content)
+            cls.save_response_data(response_data)
 
         except requests.exceptions.InvalidSchema:
-            print("Check whether the URL is valid or check if " +
-                  "the localhost server is active or not")
-        except Exception as e:
-            print(e)
+            print(cls.invalid_schema_message)
+        except Exception as exception_obj:
+            print(exception_obj)
 
     @classmethod
     def __check_endpoint(cls, request_url):

src/arguments/search.py

@@ -54,6 +54,8 @@ def search_args(self):
             update.check_for_updates()
         elif self.arguments.GET:
             self.api_test_object.get_request()
+        elif self.arguments.DELETE:
+            self.api_test_object.delete_request()
         elif self.arguments.notion:
             NotionClient().get_tokenv2_cookie()