http_server.py

"""I receive multichain username, password and ip from slave nodes and get their Savoir instances
to control them """

# pylint: disable=invalid-name, global-statement

import json
from http.server import BaseHTTPRequestHandler, HTTPServer, HTTPStatus
from typing import Type

from .abstract import AbstractSlave, AbstractSetup
from .project_logger import set_up_logging
from .meta_scenario import SLAVES_SYNC

LOG = set_up_logging(__name__)
_SLAVE_NODES = []
SETUP = AbstractSetup()
SLAVE_CLASS = AbstractSlave


class SimpleHTTPRequestHandler(BaseHTTPRequestHandler):
    """I handle requests form the slaves who send their user data"""

    def do_POST(self):
        LOG.info(self)
        content_length = int(self.headers['Content-Length'])
        body = self.rfile.read(content_length)
        self.send_response(HTTPStatus.OK)
        self.end_headers()
        config = json.loads(body.decode('utf-8'))
        self.add_slave_nodes(config)

    def add_slave_nodes(self, config):
        global _SLAVE_NODES
        _SLAVE_NODES = [slave for slave in _SLAVE_NODES if slave.is_alive()] + \
                       [SLAVE_CLASS(config, SETUP)]
        LOG.info("Added connection to slave %d", len(_SLAVE_NODES))
        LOG.info(_SLAVE_NODES)
        SLAVES_SYNC.put(_SLAVE_NODES)


def start_slave_server(setup: AbstractSetup, slave_class: Type[AbstractSlave]):
    LOG.info("Masternode is ready for connections")
    global SLAVE_CLASS, SETUP
    SLAVE_CLASS = slave_class
    SETUP = setup
    httpd = HTTPServer(('', 60000), SimpleHTTPRequestHandler)
    httpd.serve_forever()