server_side.py

import socket
import random
import datetime

LISTEN_PORT = 420
LISTEN_HOST = "localhost"

STOP_MESSAGE = "QUIT"
SERVER_NAME_MESSAGE = "Some cool name"
WELCOME_MESSAGE = ("Select from the menu:\n" +
                   "TIME - Get current local time\n" +
                   "NAME - Get server name\n" +
                   "RAND - Get random number between 1 to 10\n" +
                   "QUIT - Close the socket")


def main():
    client_msg = ""
    listen_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    listen_sock.bind((LISTEN_HOST, LISTEN_PORT))
    listen_sock.listen(1)

    print("Listening " + LISTEN_HOST + ':' + str(LISTEN_PORT))

    client_sock, client_address = listen_sock.accept()  # CONNECTED

    client_sock.sendall(WELCOME_MESSAGE.encode())

    while client_msg != STOP_MESSAGE:
        client_msg = client_sock.recv(4).decode()  # RECEIVED MESSAGE
        print("RECEIVED: " + client_msg)

        client_sock.sendall(str(do(client_msg)).encode())  # SENT MESSAGE

    client_sock.close()
    listen_sock.close()


def do(selected):
    if selected == "TIME":
        return datetime.datetime.now()
    elif selected == "NAME":
        return SERVER_NAME_MESSAGE
    elif selected == "RAND":
        return random.randint(1, 11)
    elif selected == "QUIT":
        return "Socket Closed."
    else:
        return WELCOME_MESSAGE


if __name__ == '__main__':
    main()