energy_transfer_bank.py

from flask import Flask, escape, request
import threading, json, time
from spannungsteiler.util import broker_util
import socket, logging



mutex= threading.Lock()
app = Flask(__name__)
log = logging.getLogger('werkzeug')
log.disabled = True

energy_requested = {}
energy_provided = {}


def get_local_ip():
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    s.connect(("9.9.9.9", 80))
    ip = s.getsockname()[0]
    s.close()
    return ip

def subscribe_to_topics():

    topics = [
        "spannungsteiler_demand_publish",
        "spannungsteiler_offer_publish"
    ]

    for i in topics:
        broker_util.send("subscribe", {
            "sender": "energy_bank",
            "address": "http://{}:5000/energy_bank".format(get_local_ip()),
            "interestedIn": i
        })

def start_server():
    app = Flask(__name__)

    @app.route('/energy_bank', methods=["POST"])
    def endpoint():


        payload = request.json["event"]
        if "demand" in payload["payload"]:
            amount = payload["payload"]["demand"]
        else:
            amount = payload["payload"]["offer"]

        sender_id = payload["id"]

        mutex.acquire()
        if amount < 0:
            energy_requested[sender_id] = amount
        elif amount > 0:
            energy_provided[sender_id] = amount
        else:
            mutex.release()
            return ""

        mutex.release()
        return json.dumps({'success':True}), 200, {'ContentType':'application/json'}

    t = threading.Thread(target=app.run, kwargs={"host": "0.0.0.0"})
    t.daemon = False
    t.start()


if __name__ == "__main__":


    subscribe_to_topics()
    start_server()

    while True:
        mutex.acquire()


        provided_sum = 0
        requested_sum = 0

        provided_sum = sum(energy_provided.values())

        requested_sum = sum(energy_requested.values())


        for (amount, key) in enumerate(energy_provided):
            answere = 0

            if amount <= abs(requested_sum):
                answere -= amount

            else:
                answere = requested_sum

            broker_util.send_transaction_execution("energy_bank", answere, key)
            requested_sum -= answere


        for (key, amount) in enumerate(energy_requested):
            answere = 0

            if abs(amount) <= provided_sum:
                answere -= amount

            else:
                answere = provided_sum

            broker_util.send_transaction_execution("energy_bank", answere, key)
            provided_sum -= answere      
        
        mutex.release()
        time.sleep(0.6)