Add POST endpoint

Author: harshakhmk

main.py

@@ -50,6 +50,10 @@
                     help="Make a GET request to an API",
                     action='store_true')
 
+parser.add_argument("-POST",
+                    help="Make a POST request to an API",
+                    action='store_true')
+
 parser.add_argument("-DELETE",
                      help = "Make a DELETE request to an API",
                      action = 'store_true')

src/arguments/api_test.py

@@ -42,7 +42,39 @@ def fetch_input_url(cls):
             "request_url" : request_url,
             "request_headers" : request_headers,
         }
-
+    @classmethod
+    def read_data_from_file(cls):
+        filename = input("Enter a filename (response_data.json)")
+        if filename == '':
+            print("filename empty, so default file (response_data.json) is used ")
+        with open(filename, "r") as reader:
+            file_content = reader.read()
+            try:
+                json_data = json.loads(file_content)
+                data = json_data.get("data")
+                return data
+                # Make sure the data is not None and send
+            except json.JSONDecodeError:
+                 print("Unable to parse the file, Please try again")
+                 cls.read_data_from_file()
+    @classmethod
+    def fetch_payload_data(cls):
+        store = int(input("Please choose the below options? (1/2/3)"))
+        print("Option 1: For sending data payload from terminal\n")
+        print("Option 2: For sending data payload by reading from json file\n")
+        print("Option 3: For not sending any data to POST request")
+        data = None
+        if(store == 1):
+            data = input("Enter data as key value pairs")
+            data = eval(data)
+        elif(store == 2):
+            data = cls.read_data_from_file()
+        elif(store == 3):
+            print(f"you have entered {store}, so sending POST request without any data")
+        else:
+            print(f"You have entered {store_data}, please enter from the above options")
+            cls.fetch_payload_data()
+        return data
     #saves the json response into a file
     @classmethod
     def save_response_data(cls,response_data):
@@ -84,6 +116,20 @@ def get_request(cls):
             print(cls.invalid_schema_message)
         except Exception as exception_obj:
             print(exception_obj)
+    #Make a POST request
+    @classmethod
+    def post_request(cls):
+        request_data = cls.fetch_input_url()
+        data = cls.fetch_payload_data()
+        try:
+            response= requests.post(url = request_data["request_url"], headers= request_data["request_headers"], data = data)
+            cls.print_response_json(response)
+            response_data = json.loads(response.content)
+            cls.save_response_data(response_data)
+        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):

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.POST:
+            self.api_test_object.post_request()
         elif self.arguments.DELETE:
             self.api_test_object.delete_request()
         elif self.arguments.notion: