red_agent.py

"""
Red Agent
@Authors | @Student ID
+-------------------+
Reiden Rufin | 22986337
Nathan Eden | 22960674    
"""

import random
import prettytable as pt
class red_agent:
    messages = {
        0: "Vote for red team!",
        1: "Please vote for us",
        2: "If you dont vote for us you are a bad person",
        3: "Blue team a democratic left wing upper right alt full circle libtard",
        4: "If blue team wins the future will be dark",
        5: "Blue team is going to lead us to the ground",
        6: "Why vote for blue team? they will destroy our future",
        7: "Blue voters will be punished severely and publicly and will be beaten",
        8: "if do not vote for us we will find you and burn your house down",
        9: "Blue voters and their families will be publically tortured and killed",
    }
    
    followers = None
    potency = None
    uncertainty_lower_limit = 0.0
    uncertainty_upper_limit = 0.0
    unique_id = 0
    def __init__(self, user_playing, lower_limit, upper_limit):
        self.followers = 0
        self.potency = 0
        self.user_playing = user_playing 
        self.uncertainty_lower_limit = lower_limit
        self.uncertainty_upper_limit = upper_limit
        
    def get_message_potency_follower_loss(self, message):
        #uses the message chosen to return a potency, follower loss and uncertainty change
        potency = 0
        follower_loss = 0
        uncertainty_change = 0
        if message == self.messages[0] or message == self.messages[1]:
            potency = 0.2
            follower_loss = 0.15625
            uncertainty_change = 0.03125
        elif message == self.messages[2] or message == self.messages[3]:
            potency = 0.4
            follower_loss = 0.03125
            uncertainty_change = 0.0625
        elif message == self.messages[4] or message == self.messages[5]:
            potency = 0.6
            follower_loss = 0.00625
            uncertainty_change = 0.125
        elif message == self.messages[6] or message == self.messages[7]:
            potency = 0.8
            follower_loss = 0.0125
            uncertainty_change = 0.25
        elif message == self.messages[8] or message == self.messages[9]:
            potency = 1.0
            follower_loss = 0.025
            uncertainty_change = 0.5
        return [potency, follower_loss, uncertainty_change]

    def will_vote_status_change(self, potency):
        return random.randint(0, 100) <= potency * 100
    
    def red_move(self, green_agent, message):
        follower_loss_count = 0
        uncertainty_change = 0.0
        if(green_agent.communicate):
            potency, follower_loss, uncertainty_change = self.get_message_potency_follower_loss(message)
            follower_loss_count += follower_loss
            if (green_agent.vote_status == False):
                uncertainty_change *= -1
            if(self.will_vote_status_change(potency)):
                green_agent.vote_status = False
        return uncertainty_change, follower_loss_count

    def send_message(self):
        message_to_send = ""
        if(self.user_playing):
            message_output = []
            for messages in self.messages:
                message_output.append(self.messages[messages])
            table = pt.PrettyTable()
            table.field_names = ["Message Number", '\x1b[1;33;45m' + "Message" + '\x1b[0m']
            for i in range(len(message_output)):
                table.add_row([i, message_output[i]])
            print(table)
            message = input("Please enter a message(0 - 9): ")
            try:
                message_to_send = message_output[int(message)]
            except:
                print("Invalid Move. Please Send Another Message.")
                self.send_message()
                
        else:
            message_to_send = random.choice(list(self.messages.values()))
        
        print("Red Sending message: ", message_to_send)
        return message_to_send
        
    def evaluate(self, green_team):
        score = 0
        for green_agent in green_team:
            if(green_agent.communicate):
                score += 0.25
                if(green_agent.vote_status == False):
                    score += 0.5
                else:
                    score -= 0.5
        return score