api.py

import requests
import json
import random
import logging


model_name = "ac_prediction_model"


def readFromFile(path, key):
    f = open(path, 'r')
    data = json.load(f)
    if key == "sensor_details":
        data = data["sensor_details"]
        return data
    elif key == "controller_details":
        data = data["controller_details"]
        return data

def get_public_ip():
    resp = requests.get("http://api.ipify.org/").content.decode()
    # return "172.17.0.1"
    # return "OPEN API"
    return resp


def getSensorInstances(path="app.json"):
    sensor_details = readFromFile(
        path, "sensor_details")

    pub_ip = get_public_ip()
    url = f"http://{pub_ip}:5000/"+'getSensorInstances'
    # logging.warning(url)
    response = requests.post(url=url, json={
        "sensor_type": sensor_type[0],
        "sensor_location": sensor_location
    }).content
    data = json.loads(response.decode())
    sensor_instances = data["sensor_instances"]
    return sensor_instances, no_of_instances


def getControlInstances(path="app.json"):
    sensor_type, sensor_location = readFromFile(path, "controller_details")

    pub_ip = get_public_ip()

    url = f"http://{pub_ip}:6000/"+'getControlInstances'
    # logging.warning(url)
    response = requests.post(url=url, json={
        "sensor_type": sensor_type,
        "sensor_location": sensor_location
    }).content
    data = json.loads(response.decode())
    control_instances = data['control_instances']
    return control_instances


def getSensorData():
    all_instances, no_of_instances = getSensorInstances()
    sensor_instances = random.sample(all_instances, no_of_instances)

    pub_ip = get_public_ip()
    url = f"http://{pub_ip}:5000/"+'getSensorData'
    # logging.warning(url)
    response = requests.post(url=url, json={
        "topic_name": sensor_instances[0]
    }).content
    data = json.loads(response.decode())
    data = data['sensor_data']
    return data[-1]


def controllerAction(data):
    all_instances = getControlInstances()
    instance = all_instances[0]
    # url = control_url+'performAction'
    pub_ip = get_public_ip()
    url = f"http://{pub_ip}:6000/"+'performAction'
    response = requests.post(url=url, json={
        "sensor_type": instance["sensor_type"],
        "sensor_ip": instance["sensor_ip"],
        "sensor_port": instance["sensor_port"],
        "data": int(data)
    }).content
    return response.decode()


def predict(data):
    # MAKE API call to the model
    # url = model_url+'predict'
    pub_ip = get_public_ip()
    url = f"http://{pub_ip}:5003/"+'predict'
    # logging.warning("Data: ", data.tolist())
    # data = data.tolist()
    # logging.warning(type(data))
    response = requests.post(url=url, json={
        "data": data.tolist(),
        "model_name": model_name
    }).content
    prediction = json.loads(response.decode())
    prediction = prediction["predicted_value"]
    return prediction


# getSensorData("light", "himalaya-block")
# readFromFile()