diff options
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() - } - |
