client.py

import requests
import time
from threading import Thread
import argparse
import random

class PyClient:
    def __init__(self, server_url='localhost', port=8000, verbose=False):
        self.server_url = f'http://127.0.0.1' if server_url == 'localhost' else server_url
        self.verbose = verbose
        self.port = port

    def get_url(self):
        if (random.random() > 0.5):
            return self.server_url + ':' + str(self.port)
        else:
            return self.server_url + ':' + str(self.port)

    def push(self, key, value):
        message = key+','+value
        response = requests.post(self.get_url() + '/push', data=message)
        if response.text == "The Server Overloaded":
            print("The Server Overloaded")
            return
        if self.verbose:
            print('Received from server: ' + response.text)
        return response.text

    def pull(self):
        response = requests.get(self.get_url() + '/pull')
        print(response.text)
        if response.text == "The Server Overloaded":
            print("The Server Overloaded")
            return "The Server Overloaded", "The Server Overloaded"
        if self.verbose:
            print('Received from server: ' + response.text)
        if response.text != "no message":
            return response.text.split(',')
        return response.text

    def subscribe_runner(self, url, f):
        while True:
            response = requests.get(self.get_url() + '/pull')
            data = response.text
            if data == "The Server Overloaded":
                print("The Server Overloaded")
                #kill the thread
                break
            if data == "no message":
                time.sleep(1)
            else:
                data = data.split(',')
                #print(data)
                f(data[0], data[1])
                time.sleep(0.01)

    def subscribe(self, f):
        thread = Thread(target=self.subscribe_runner, args=(self.server_url, f))
        thread.start()