from peewee import *

from src.model.BaseModel import BaseModel
from src.model.User import User
from src.model.enums.income_tax.IncomeTaxBreakdown import IncomeTaxBreakdown
from src.model.enums.income_tax.IncomeTaxContribution import IncomeTaxContribution
from src.model.enums.income_tax.IncomeTaxDeduction import IncomeTaxDeduction
from src.model.enums.income_tax.IncomeTaxEventType import IncomeTaxEventType


class IncomeTaxEvent(BaseModel):
    """
    IncomeTaxEvent class
    """

    user = ForeignKeyField(User, backref="income_tax_events")
    event_id = IntegerField()
    event_type = CharField()
    starting_amount = BigIntegerField()
    amount = BigIntegerField()
    breakdown_list = TextField()
    deduction_list = TextField()
    contribution_list = TextField()

    class Meta:
        db_table = "income_tax_event"

    def get_tax_deduction_list(self) -> list[IncomeTaxDeduction]:
        """
        Get the tax deduction list
        :return: The tax deduction list
        """
        return IncomeTaxDeduction.from_string(str(self.deduction_list))

    def get_tax_breakdown_list(self) -> list[IncomeTaxBreakdown]:
        """
        Get the tax breakdown list
        :return: The tax breakdown list
        """
        return IncomeTaxBreakdown.from_string(str(self.breakdown_list))

    def get_tax_contribution_list(self) -> list[IncomeTaxContribution]:
        """
        Get the tax contribution list
        :return: The tax contribution list
        """
        return IncomeTaxContribution.from_string(str(self.contribution_list))

    def get_event_type(self) -> IncomeTaxEventType:
        """
        Get the event type
        :return: The event type
        """
        return IncomeTaxEventType(self.event_type)

    def get_event_type_description(self) -> str:
        """
        Get the event type description
        :return: The event type description
        """
        return self.get_event_type().get_description()


IncomeTaxEvent.create_table()