main.py

import sys
import socket
import json
from agents import SuperAgent
from agents import Strategist
from queue import Queue


def main():
    teamSize = get_teamSize()

    # In case teamSize is returned and not None, start up the agents
    if teamSize:
        a_list = []

        # The input queue is used to send requests from the agents to
        # the strategist. The output queue for the other way around.
        input_queue = Queue(maxsize=teamSize)
        output_queue = Queue(maxsize=teamSize)

        # The strategist helps update and merge the graphs.
        # Comment the next 2 lines if you don't want to use a strategist.
        strategist = Strategist(f"Strategist", [input_queue, output_queue])
        strategist.start()

        for i in range(1, teamSize + 1):
            a_list.append(SuperAgent(f"agentA{i}", "1"))
            a_list[-1].start()


def get_teamSize():
    """
    Returns the configuration of the server as
    returned by the SIM-CONFIG message request
    args:
        None
    returns:
        teamSize
    """
    # Create socket object.
    host, port = "localhost", 12300
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # Connect to server.
    try:
        sock.connect((host, port))

    # In case of error throw error message
    except ConnectionRefusedError:
        print("Could not connect to port")
        return

    # Create status-request
    request = {
        "type": "status-request",
        "content": {}
    }

    # Send status-request and listen for response
    sock.sendall((json.dumps(request) + "\0").encode())

    msg = sock.recv(65536).decode().split("\0")[0]
    msg = json.loads(msg)
    # print(msg)

    # If message is received, parse it and return it otherwise return None
    if len(msg) > 1:
        currentSimulation = msg["content"]["currentSimulation"]
        teamSizes = msg["content"]["teamSizes"]

        # If currentSimulation index is -1, simulation has not started yet
        if currentSimulation == -1:
            return teamSizes[0]
        else:
            return teamSizes[currentSimulation]
    else:
        return None

    sock.close()


if __name__ == "__main__":
    main()