Skip to content

Commit c97e707

Browse files
committed
needed to add session handling to auth_request test to bypass netrc file
1 parent 54156b9 commit c97e707

File tree

1 file changed

+38
-5
lines changed

1 file changed

+38
-5
lines changed

utils/auth_request.py

Lines changed: 38 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import sys
66
import json
77
import requests
8+
import logging
89

910
###############################################################################
1011
# MAIN
@@ -16,29 +17,61 @@
1617
password = sys.argv[2]
1718
organization = sys.argv[3]
1819
url = "slideruleearth.io"
20+
verbose = False
1921

2022
# Set URL
2123
if len(sys.argv) > 4:
2224
url = sys.argv[4]
2325

26+
# Set Verbose
27+
if len(sys.argv) > 5:
28+
verbose = sys.argv[5] == "verbose"
29+
30+
# Configure Debug Logging
31+
if verbose:
32+
import http.client as http_client
33+
http_client.HTTPConnection.debuglevel = 1
34+
35+
# You must initialize logging, otherwise you'll not see debug output.
36+
logging.basicConfig()
37+
logging.getLogger().setLevel(logging.DEBUG)
38+
requests_log = logging.getLogger("requests.packages.urllib3")
39+
requests_log.setLevel(logging.DEBUG)
40+
requests_log.propagate = True
41+
42+
# Create Session
43+
session = requests.Session()
44+
session.trust_env = False
45+
2446
# Authentication Request
2547
host = "https://ps." + url + "/api/org_token/"
2648
rqst = {"username": username, "password": password, "org_name": organization}
2749
headers = {'Content-Type': 'application/json'}
28-
rsps = requests.post(host, data=json.dumps(rqst), headers=headers, timeout=(60,10)).json()
50+
rsps = session.post(host, data=json.dumps(rqst), headers=headers, timeout=(60,10)).json()
2951
refresh = rsps["refresh"]
3052
access = rsps["access"]
3153
print("Login Response: ", rsps)
3254

3355
# Organization Access Request
3456
host = "https://ps." + url + "/api/membership_status/" + organization + "/"
3557
headers = {'Authorization': 'Bearer ' + access}
36-
rsps = requests.get(host, headers=headers, timeout=(60,10)).json()
58+
rsps = session.get(host, headers=headers, timeout=(60,10)).json()
3759
print("Validation Response: ", rsps)
3860

39-
# Refresh Request
61+
# Refresh Request 1
62+
host = "https://ps." + url + "/api/org_token/refresh/"
63+
rqst = {"refresh": refresh}
64+
headers = {'Content-Type': 'application/json', 'Authorization': 'Bearer ' + access}
65+
rsps = session.post(host, data=json.dumps(rqst), headers=headers, timeout=(60,10)).json()
66+
refresh = rsps["refresh"]
67+
access = rsps["access"]
68+
print("Refresh 1 Response: ", rsps)
69+
70+
# Refresh Request 2
4071
host = "https://ps." + url + "/api/org_token/refresh/"
4172
rqst = {"refresh": refresh}
4273
headers = {'Content-Type': 'application/json', 'Authorization': 'Bearer ' + access}
43-
rsps = requests.post(host, data=json.dumps(rqst), headers=headers, timeout=(60,10)).json()
44-
print("Refresh Response: ", rsps)
74+
rsps = session.post(host, data=json.dumps(rqst), headers=headers, timeout=(60,10)).json()
75+
refresh = rsps["refresh"]
76+
access = rsps["access"]
77+
print("Refresh 2 Response: ", rsps)

0 commit comments

Comments
 (0)