import requests
import time
import ast
import json

# Base url to perform request to Ubidots API
BASE_URL = "https://industrial.api.ubidots.com"

REQUESTS_FUNCTIONS = {"get": requests.get, "post": requests.post}

# If you changed the data configuration name to other 
# than the one used in the article, set this variable to that name 
DATA_CONFIGURATION_NAME: str = 'ModbusServerDataHTTP'

def main(args):
    # Print args for debug
    print(args)
    # Get the payload containing the data of our interest
    data: dict = args[DATA_CONFIGURATION_NAME]
    # Build the device label using the server name and its ip address
    device_label: str = data['server_name'].replace('_', '-') + '-' + data['ip'].replace('.', '-')
    # Get the variable's name 
    variable_label: str = data['name']
    # Divide the value by 10 according to the device's datasheet. Adjust this accordingly
    value: int = int(data['data']) / 10
    # Get the Ubidots token
    token: str = args.get('token', None)
    
    if token is None:
        print("[ERROR] Please configure you Ubidots token for this Ubifunction or define it within it.")
        return {"status": "error"}
        
    if not data:
        print("No data")
        return {"error" : "No data present in the payload"}
        
    payload: dict = {variable_label : value}
    req = update_device(device_label, payload, token)
 
    print("[INFO] Request result:")
    print(req.text)

    return {"status": "Ok", "result": req.json()}
    
    
    
def update_device(device, payload, token):
    """
    updates a variable with a single dot
    """

    url = "{}/api/v1.6/devices/{}".format(BASE_URL, device)
    headers = {"X-Auth-Token": token, "Content-Type": "application/json"}

    req = create_request(url, headers, payload, attempts=5, request_type="post")
    
    return req

def create_request(url, headers, data, attempts, request_type):
    """
    Function to create a request to the server
    """

    request_func = REQUESTS_FUNCTIONS.get(request_type)

    kwargs = {"url": url, "headers": headers}

    if request_type == "post":
        kwargs["json"] = data

    try:
        req = request_func(**kwargs)
        print("[INFO] Request result: {}".format(req.text))
        status_code = req.status_code
        time.sleep(1)

        while status_code >= 400 and attempts < 5:
            req = request_func(**kwargs)
            print("[INFO] Request result: {}".format(req.text))
            status_code = req.status_code
            attempts += 1
            time.sleep(1)

        return req
    except Exception as e:
        print("[ERROR] There was an error with the request, details:")
        print(e)
        return None