from common import userstatus
class Protectees():
    def update(self, nick, user, host):
        if nick in self.protectees: # On record.
            if userstatus.atleast_halfop(self, user, self.homechannel) or nick == self.connection.get_nickname():   # Update. Is atleast halfop or bot itself.
                self.protectees[nick].update({'ident': nick + "!" + user + "@" + host})
            else:   # Delete.
                del self.protectees[nick]
        else:   # Append.
            if userstatus.atleast_halfop(self, user, self.homechannel) or nick == self.connection.get_nickname():   # Update. Is atleast halfop or bot itself.
                self.protectees[nick] = {'ident': nick + "!" + user + "@" + host}

class Aggressiveness():
    def retalliation_reason(self, connection, protectee, behaviour):
        if protectee == connection.get_nickname():  # Bot itself.
             return "Aggression channel function = " + behaviour + ": Self defense."
        else:
            return "Aggression channel function = " + behaviour + ": " + protectee + " is atlast halfop in " + self.homechannel + "."