client.py

import argparse
import socket

from colorama import Fore, Style


def create_client(server_host, server_port):
    # Create socket
    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # Connect to server
    client.connect((server_host, server_port))

    return client


def send_client(client, message):
    print(Fore.BLUE + f'Sending a message: {message}')
    client.send(message.encode())


def receive_client(client):
    message = client.recv(1024).decode()

    if not message:
        client.close()
        return None

    print(Fore.GREEN + f'Received a message: {message}')
    print(Style.RESET_ALL)
    return message


def client_user_input():
    proc_count = int(input(Fore.MAGENTA + 'Enter process level count (for each level the number of processes is 2^level): '))

    input_list = list(map(int, input(Fore.MAGENTA + 'Enter the list: ').strip().split()))

    result = '('
    result += str(proc_count)
    result += ', '
    result += '['
    result += ','.join([str(i) for i in input_list])
    result += '])'

    return result


def main():
    argParser = argparse.ArgumentParser()
    argParser.add_argument("-sh", "--server_host", type=str, default='localhost', help="Host name")
    argParser.add_argument("-sp", "--server_port", type=int, default='8008', help="Port number")
    args = argParser.parse_args()

    client = create_client(args.server_host, args.server_port)
    send_client(client, client_user_input())
    receive_client(client)


if __name__ == "__main__":
    main()