Skip to content

Commit fd8bc08

Browse files
committed
added support for header for api calls
1 parent c3d2af3 commit fd8bc08

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

src/arguments/api_test.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,26 @@
33

44
class ApiTesting():
55
default_url = "https://127.0.0.1:8000"
6-
6+
default_headers = {}
77
# Make GET request
88
@classmethod
99
def get_request(cls):
1010
request_url = cls.default_url
11+
request_headers = cls.default_headers
1112
input_url = input('Enter URL: ')
13+
input_headers = input('Enter Headers: ')
14+
1215
if input_url != '':
1316
request_url = input_url
1417

18+
if input_headers != '':
19+
try:
20+
request_headers = json.loads(input_headers)
21+
# If Exception Occurs while Parsing Headers Printing info
22+
except Exception as ex:
23+
print("Failed to parse Input Headers")
24+
25+
1526
# Check whether the request_url has an endpoint or not
1627
has_endpoint = cls.__check_endpoint(request_url)
1728

@@ -33,7 +44,7 @@ def get_request(cls):
3344

3445
# Make GET request and store the response in response_data.json
3546
try:
36-
response = requests.get(request_url)
47+
response = requests.get(request_url, headers=request_headers)
3748
print(f"Reponse Status Code: {response.status_code}")
3849
response_data = json.loads(response.content)
3950
parsed_json = json.dumps(response_data, indent=4)

0 commit comments

Comments
 (0)