diff options
| author | Dylan Bolger <dylan.bolger00@gmail.com> | 2023-10-15 21:35:22 -0500 |
|---|---|---|
| committer | Dylan Bolger <dylan.bolger00@gmail.com> | 2023-10-15 21:35:22 -0500 |
| commit | 7be27a3c4399e23506971fcee540176144e89da3 (patch) | |
| tree | 37901e5d0ae07f5c44ee93d0c833aac38189821c /app/energy_request | |
| parent | d8f609bf6c71e088ee0be093ea126df05d0211f5 (diff) | |
| download | city-utilities-restful-wrapper-7be27a3c4399e23506971fcee540176144e89da3.tar.xz city-utilities-restful-wrapper-7be27a3c4399e23506971fcee540176144e89da3.zip | |
Further organize modules, rename request energy request method, provide tentative data in response
Diffstat (limited to 'app/energy_request')
| -rw-r--r-- | app/energy_request/energy_request.py | 78 |
1 files changed, 0 insertions, 78 deletions
diff --git a/app/energy_request/energy_request.py b/app/energy_request/energy_request.py deleted file mode 100644 index f6aac3f..0000000 --- a/app/energy_request/energy_request.py +++ /dev/null @@ -1,78 +0,0 @@ -import json -import requests -from ..constants.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() - } - |
