From 7be27a3c4399e23506971fcee540176144e89da3 Mon Sep 17 00:00:00 2001
From: Dylan Bolger <dylan.bolger00@gmail.com>
Date: Sun, 15 Oct 2023 21:35:22 -0500
Subject: Further organize modules, rename request energy request method,
 provide tentative data in response

---
 app/request/login/login_request.py | 53 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 app/request/login/login_request.py

(limited to 'app/request/login')

diff --git a/app/request/login/login_request.py b/app/request/login/login_request.py
new file mode 100644
index 0000000..c74986e
--- /dev/null
+++ b/app/request/login/login_request.py
@@ -0,0 +1,53 @@
+import requests
+import re
+from ...constants.constants import loginPageHeaders, loginRequestJson, genericRequestHeaders, loginPageUri, loginRequestEndpoint
+
+def login():
+    # Grab generated session keys from viewing the webpage
+    aga, asi, ct = grabRequiredKeys()
+    # Perform a login request using keys found on the page and JSON data (credentials)
+    lt = performLoginRequest(aga, asi, ct)
+    # Return the keys required to make endpoint calls
+    return {
+        "lt": lt,
+        "aga": aga,
+        "asi": asi,
+        "ct": ct
+    }
+
+
+def grabRequiredKeys():
+    loginPageResponse = requests.get(loginPageUri, headers=loginPageHeaders)
+    affinityMatcher = re.compile("CORS=(.+?);")
+    affinityResults = affinityMatcher.search(loginPageResponse.headers['Set-Cookie'])
+    appGatewayAffinity = affinityResults.group(1)
+
+    aspNetSessionIdMatcher = re.compile("ASP\.NET\_SessionId=(.+?);")
+    aspnetResults = aspNetSessionIdMatcher.search(loginPageResponse.headers['Set-Cookie'])
+    aspNetSessionId = aspnetResults.group(1)
+
+    csrfMatcher = re.compile("id=\"hdnCSRFToken\" value=\"(.+)\"")
+    csrfResults = csrfMatcher.search(loginPageResponse.text)
+    csrfToken = csrfResults.group(1)
+    return appGatewayAffinity, aspNetSessionId, csrfToken
+
+def performLoginRequest(appGatewayAffinity, aspNetSessionId, csrfToken):
+    loginRequestCookies = {
+        'ApplicationGatewayAffinityCORS': appGatewayAffinity,
+        'ApplicationGatewayAffinity': appGatewayAffinity,
+        'ASP.NET_SessionId': aspNetSessionId,
+    }
+
+    genericRequestHeaders['csrftoken'] = csrfToken
+
+    loginResponse = requests.post(
+        loginRequestEndpoint,
+        cookies=loginRequestCookies,
+        headers=genericRequestHeaders,
+        json=loginRequestJson,
+    )
+
+    scpMatcher = re.compile("SCP=(.{36});")
+    scpSearchResults = scpMatcher.search(loginResponse.headers['Set-Cookie'])
+    loginToken = scpSearchResults.group(1)
+    return loginToken
-- 
cgit v1.2.3