reflexAgent2.py

from randomClient import RandomClient
from featureExtractor import *
import random, argparse, sys
from PyQt5.QtCore import QCoreApplication

class reflexAgent(RandomClient):
    def __init__(self, client_id = None, serverPort = 0, 
            serverAddress = 'localhost', name = 'Noname', client_type = 'PLAYER'):
        super().__init__(client_id, serverPort, serverAddress, name, client_type)

        
    def chooseTile(self, actions):
        if len(actions) == 1: # covers the case that the only tile is 'Nothing'
            return actions[0]
        outcomes = placementOutcome(self.state, actions)
        for k,v in outcomes:
            if v == 'found':
                return k
        return random.choice(actions)

    def chooseNewCompany(self, actions): return random.choice(actions)
    def chooseSurvivor(self, actions): return random.choice(actions)
    def chooseLiquidate(self, actions): return random.choice(actions)

    def chooseStock(self, actions): 
        actions.remove('Done')
        if len(actions) == 0:
            return 'Done'
        if len(actions) == 1:
            return actions[0]

        choices = list()
        for company in actions:
            bonuses = getBonus(self.state, company) 
            if bonuses[0]['Player'] == bonuses[1]['Player'] and bonuses[0]['Player'] != self.name:
                choices.append(company)
            elif self.name != bonuses[0]['Player'] and self.name != bonuses[1]['Player']:
                choices.append(company)

        if len(choices):
            return random.choice(choices)
        else:
            return random.choice(actions)

    def chooseEndGame(self, actions): return "Yes"


if __name__ == '__main__':
    app = QCoreApplication(sys.argv)
    parser = argparse.ArgumentParser()
    parser.add_argument('-p', '--serverPort', type = int)
    parser.add_argument('-n', '--playerName', type = str)
    args = parser.parse_args()
    a = reflexAgent(name = args.playerName, serverPort = args.serverPort)
    app.exec_()