forked from icann/czds-api-client-python
-
Notifications
You must be signed in to change notification settings - Fork 0
/
do_authentication.py
36 lines (30 loc) · 1.25 KB
/
do_authentication.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import json
import requests
import sys
import datetime
def authenticate(username, password, authen_base_url):
authen_headers = {'Content-Type': 'application/json',
'Accept': 'application/json'}
credential = {'username': username,
'password': password}
authen_url = authen_base_url + '/api/authenticate'
response = requests.post(authen_url, data=json.dumps(credential), headers=authen_headers)
status_code = response.status_code
# Return the access_token on status code 200. Otherwise, terminate the program.
if status_code == 200:
access_token = response.json()['accessToken']
print('{0}: Received access_token:'.format(datetime.datetime.now()))
# print(access_token)
return access_token
elif status_code == 404:
sys.stderr.write("Invalid url " + authen_url)
exit(1)
elif status_code == 401:
sys.stderr.write("Invalid username/password. Please reset your password via web")
exit(1)
elif status_code == 500:
sys.stderr.write("Internal server error. Please try again later")
exit(1)
else:
sys.stderr.write("Failed to authenticate user {0} with error code {1}".format(username, status_code))
exit(1)