volvo2mqtt/volvo.py

import requests
from datetime import datetime, timedelta
import mqtt
from config import settings
from babel.dates import format_datetime
from const import charging_system_states, CLIMATE_START_URL, \
    OAUTH_URL, VEHICLES_URL, VEHICLE_DETAILS_URL, RECHARGE_STATUS_URL

session = requests.Session()
session.headers = {
                "vcc-api-key": settings["volvoData"]["vccapikey"],
                "content-type": "application/json",
                "accept": "*/*"
}

token_expires_at: datetime
refresh_token = None
vins = []
recharge_response = {}
recharge_last_update = None


def authorize():
    headers = {
                "authorization": "Basic aDRZZjBiOlU4WWtTYlZsNnh3c2c1WVFxWmZyZ1ZtSWFEcGhPc3kxUENhVXNpY1F0bzNUUjVrd2FKc2U0QVpkZ2ZJZmNMeXc=",
                "content-type": "application/x-www-form-urlencoded",
                "accept": "application/json"
    }

    body = {
            "username": settings.volvoData["username"],
            "password": settings.volvoData["password"],
            "grant_type": "password",
            "scope": "openid email profile care_by_volvo:financial_information:invoice:read care_by_volvo:financial_information:payment_method care_by_volvo:subscription:read customer:attributes customer:attributes:write order:attributes vehicle:attributes tsp_customer_api:all conve:brake_status conve:climatization_start_stop conve:command_accessibility conve:commands conve:diagnostics_engine_status conve:diagnostics_workshop conve:doors_status conve:engine_status conve:environment conve:fuel_status conve:honk_flash conve:lock conve:lock_status conve:navigation conve:odometer_status conve:trip_statistics conve:tyre_status conve:unlock conve:vehicle_relation conve:warnings conve:windows_status energy:battery_charge_level energy:charging_connection_status energy:charging_system_status energy:electric_range energy:estimated_charging_time energy:recharge_status vehicle:attributes"
    }
    auth = requests.post(OAUTH_URL, data=body, headers=headers)
    if auth.status_code == 200:
        data = auth.json()
        session.headers.update({'authorization': "Bearer " + data["access_token"]})

        global token_expires_at, refresh_token
        token_expires_at = datetime.now() + timedelta(seconds=(data["expires_in"] - 30))
        refresh_token = data["refresh_token"]

        get_vehicles()
    else:
        message = auth.json()
        raise Exception(message["error_description"])


def refresh_auth():
    print("Refreshing credentials")
    global refresh_token
    headers = {
                "authorization": "Basic aDRZZjBiOlU4WWtTYlZsNnh3c2c1WVFxWmZyZ1ZtSWFEcGhPc3kxUENhVXNpY1F0bzNUUjVrd2FKc2U0QVpkZ2ZJZmNMeXc=",
                "content-type": "application/x-www-form-urlencoded",
                "accept": "application/json"
    }

    body = {
            "grant_type": "refresh_token",
            "refresh_token": refresh_token
    }
    auth = requests.post(OAUTH_URL, data=body, headers=headers)
    if auth.status_code == 200:
        data = auth.json()
        session.headers.update({'authorization': "Bearer " + data["access_token"]})

        global token_expires_at
        token_expires_at = datetime.now() + timedelta(seconds=(data["expires_in"] - 30))
        refresh_token = data["refresh_token"]


def get_vehicles():
    global vins
    if not settings.volvoData["vin"]:
        vehicles = session.get(VEHICLES_URL)
        if vehicles.status_code == 200:
            data = vehicles.json()
            if len(data["data"]) > 0:
                for vehicle in data["data"]:
                    vins.append(vehicle["vin"])
            else:
                raise Exception("No vehicle in account " + settings.volvoData["username"] + " found.")
        else:
            raise Exception("Error getting vehicles: " + str(vehicles.status_code))
    else:
        if isinstance(settings.volvoData["vin"], list):
            # If setting is a list, copy
            vins = settings.volvoData["vin"]
        else:
            # If setting is a string, append to list
            vins.append(settings.volvoData["vin"])

    if len(vins) == 0:
        raise Exception("No vehicle found, exiting application!")
    else:
        print("Vin: " + str(vins) + " found!")


def disable_climate():
    mqtt.assumed_climate_state = "OFF"
    mqtt.update_car_data()


def api_call(url, method, vin, sensor_id=None):
    global token_expires_at
    if datetime.now() >= token_expires_at:
        refresh_auth()

    global recharge_response, recharge_last_update
    if url == RECHARGE_STATUS_URL:
        # Minimize API calls for recharge API
        if not bool(recharge_response):
            # No API Data cached, get fresh data from API
            print("Starting " + method + " call against " + url)
            response = session.get(url.format(vin), timeout=15)
            recharge_response = response
            recharge_last_update = datetime.now()
        else:
            if (datetime.now() - recharge_last_update).total_seconds() >= settings["updateInterval"]:
                # Old Data in Cache, updateing
                print("Starting " + method + " call against " + url)
                response = session.get(url.format(vin), timeout=15)
                recharge_response = response
                recharge_last_update = datetime.now()
            else:
                # Data is up do date, returning cached data
                response = recharge_response
    elif method == "GET":
        print("Starting " + method + " call against " + url)
        response = session.get(url.format(vin), timeout=15)
    elif method == "POST":
        print("Starting " + method + " call against " + url)
        response = session.post(url.format(vin), timeout=20)
    else:
        print("Unkown method posted: " + method + ". Returning nothing")
        return ""

    if response.status_code == 200:
        data = response.json()
    else:
        if url == CLIMATE_START_URL and response.status_code == 503:
            print("Car in use, cannot start pre climatization")
            mqtt.assumed_climate_state = "OFF"
            mqtt.update_car_data()
        else:
            print("API Call failed. Status Code: " + str(response.status_code) + ". Error: " + response.text)
        return ""

    if url == VEHICLE_DETAILS_URL:
        return data["data"]
    elif sensor_id == "battery_charge_level":
        if "batteryChargeLevel" in data["data"]:
            return data["data"]["batteryChargeLevel"]["value"]
        else:
            return ""
    elif sensor_id == "electric_range":
        if "electricRange" in data["data"]:
            return data["data"]["electricRange"]["value"]
        else:
            return ""
    elif sensor_id == "charging_system_status":
        if "charging_system_status" in data["data"]:
            return charging_system_states[data["data"]["chargingSystemStatus"]["value"]]
        else:
            return ""
    elif sensor_id == "estimated_charging_time":
        if "chargingSystemStatus" in data["data"]:
            charging_system_state = charging_system_states[data["data"]["chargingSystemStatus"]["value"]]
            if charging_system_state == "Charging":
                return data["data"]["estimatedChargingTime"]["value"]
            else:
                return 0
        else:
            return ""
    elif sensor_id == "estimated_charging_finish_time":
        if "chargingSystemStatus" in data["data"]:
            charging_system_state = charging_system_states[data["data"]["chargingSystemStatus"]["value"]]
            if charging_system_state == "Charging":
                charging_time = int(data["data"]["estimatedChargingTime"]["value"])
                charging_finished = datetime.now() + timedelta(minutes=charging_time)
                return format_datetime(charging_finished, format="medium", locale=settings["babelLocale"])
            else:
                return None
        else:
            return None
    elif sensor_id == "lock_status":
        return data["data"]["carLocked"]["value"]
    else:
        return ""