diff options
| author | Dylan Bolger <dylan.bolger00@gmail.com> | 2023-10-15 12:33:52 -0500 |
|---|---|---|
| committer | Dylan Bolger <dylan.bolger00@gmail.com> | 2023-10-15 12:33:52 -0500 |
| commit | 8cf01e1e4499cfc4afa0a051d7940e4eccb0debe (patch) | |
| tree | c60ea2526dc6af595d764c93303ed8945faee3c3 /energy_request.py | |
| parent | ff1777904e1bbf99d56da5b5f32e29b045406965 (diff) | |
| download | city-utilities-restful-wrapper-8cf01e1e4499cfc4afa0a051d7940e4eccb0debe.tar.xz city-utilities-restful-wrapper-8cf01e1e4499cfc4afa0a051d7940e4eccb0debe.zip | |
Add Dockerfile and organize structure for running modules
Diffstat (limited to 'energy_request.py')
| -rw-r--r-- | energy_request.py | 78 |
1 files changed, 0 insertions, 78 deletions
diff --git a/energy_request.py b/energy_request.py deleted file mode 100644 index 9b8542a..0000000 --- a/energy_request.py +++ /dev/null @@ -1,78 +0,0 @@ -import json -import requests -from constants import usageRequestCookies, genericRequestHeaders, electricUsageRequestJson, waterUsageRequestJson, waterRequestEndpoint, electricRequestEndpoint - -# Electric - -def dayElectricRequest(): - electricUsageRequestJson['Mode'] = 'D' - return performElectricRequest() - -def monthElectricRequest(): - electricUsageRequestJson['Mode'] = 'M' - return performElectricRequest() - -def performElectricRequest(): - electricUsageResponse = requests.post( - electricRequestEndpoint, - cookies=usageRequestCookies, - headers=genericRequestHeaders, - json=electricUsageRequestJson - ) - return parseResponse(electricUsageResponse) - -def requestElectric(): - return { - "day": dayElectricRequest(), - "month": monthElectricRequest() - } - -# Water - -def dayWaterRequest(): - waterUsageRequestJson['Mode'] = 'D' - return performWaterRequest() - -def monthWaterRequest(): - waterUsageRequestJson['Mode'] = 'M' - return performWaterRequest() - -def performWaterRequest(): - waterUsageResponse = requests.post( - waterRequestEndpoint, - cookies=usageRequestCookies, - headers=genericRequestHeaders, - json=waterUsageRequestJson - ) - return parseResponse(waterUsageResponse) - -def requestWater(): - return { - "day": dayWaterRequest(), - "month": monthWaterRequest() - } - -# Utility methods - -def parseResponse(response): - # such an icky response from an endpoint - # TODO: Remove unneeded JSON entries? - return json.loads(response.text.replace("\\\"", "\"").replace("\\\"", "\"").replace("\"{\"", "{\"").replace("}\"}", "}}"))['d']['objUsageGenerationResultSetTwo'] - -def setupRequestParameters(parameters): - # Setup cookies and csrftoken to perform requests - usageRequestCookies['ApplicationGatewayAffinityCORS'] = parameters['aga'] - usageRequestCookies['ApplicationGatewayAffinity'] = parameters['aga'] - usageRequestCookies['ASP.NET_SessionId'] = parameters['asi'] - usageRequestCookies['SCP'] = parameters['lt'] - genericRequestHeaders['csrftoken'] = parameters['ct'] - -# Service calling method - -def request(requestParameters): - setupRequestParameters(requestParameters) - return { - "electric": requestElectric(), - "water": requestWater() - } - |
