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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
import requests
import re
from ...constants.constants import login_page_headers, login_request_json, generic_request_headers, login_page_uri, login_request_endpoint
s = requests.session()
def login():
# Grab generated session keys from viewing the webpage
aga, asi, ct = grab_required_keys()
# Perform a login request using keys found on the page and JSON data (credentials)
lt = perform_login_request(aga, asi, ct)
# Return the keys required to make endpoint calls
return {
"lt": lt,
"aga": aga,
"asi": asi,
"ct": ct
}
def grab_required_keys():
login_page_response = s.get(login_page_uri, headers=login_page_headers)
affinity_matcher = re.compile("CORS=(.+?);")
affinity_results = affinity_matcher.search(login_page_response.headers['Set-Cookie'])
app_gateway_affinity = affinity_results.group(1)
asp_net_session_id_matcher = re.compile("ASP\.NET\_SessionId=(.+?);")
asp_net_results = asp_net_session_id_matcher.search(login_page_response.headers['Set-Cookie'])
asp_net_session_id = asp_net_results.group(1)
csrf_matcher = re.compile("id=\"hdnCSRFToken\" value=\"(.+)\"")
csrf_results = csrf_matcher.search(login_page_response.text)
csrf_token = csrf_results.group(1)
s.cookies.clear()
return app_gateway_affinity, asp_net_session_id, csrf_token
def perform_login_request(app_gateway_affinity, asp_net_session_id, csrf_token):
login_request_cookies = {
'ApplicationGatewayAffinityCORS': app_gateway_affinity,
'ApplicationGatewayAffinity': app_gateway_affinity,
'ASP.NET_SessionId': asp_net_session_id,
}
generic_request_headers['csrftoken'] = csrf_token
login_response = s.post(
login_request_endpoint,
cookies=login_request_cookies,
headers=generic_request_headers,
json=login_request_json,
)
scp_matcher = re.compile("SCP=(.{36});")
scp_results = scp_matcher.search(login_response.headers['Set-Cookie'])
login_token = scp_results.group(1)
s.cookies.clear()
return login_token
|
