person.py

"""
Person implementation
"""


class Person:
    '''
    Class that represents a single person

    Possible states:
        - active_ua
        - passive_ua
        - surzhyk
        - passive_ru
        - active_ru

    Possible ages:
        - youth (0)
        - adult (1)
        - senior (2)
    '''
    ACTIVE_UA = 4
    PASSIVE_UA = 3
    SURZHYK = 2
    PASSIVE_RU = 1
    ACTIVE_RU = 0
    TRANSITION_LIST = ["a_ru", "p_ru", "surz",
                       "p_ua", "a_ua"]
    YOUTH = 0
    ADULT = 1
    SENIOR = 2

    def __init__(self, age_group, coordinates, state):
        self.age_group = age_group
        self.coordinates = coordinates
        self.state = state
        self.transition_prob = 0
        self.age = 0
        self.infamily = False

    def get_coords(self):
        return self.coordinates

    def change_state(self, prob):
        '''
        Updates transition_prob according to the neighbours.
        '''
        if prob < abs(self.transition_prob):
            self.state += int(self.transition_prob / abs(self.transition_prob))
            self.state = self.ACTIVE_UA if self.state > self.ACTIVE_UA else self.state
            self.state = self.ACTIVE_RU if self.state < self.ACTIVE_RU else self.state
            self.transition_prob = 0

    def __repr__(self) -> str:
        return self.TRANSITION_LIST[self.state]