import json

from payrollservice.util.payrollutil import data_roundup


class EmployeeadvancepaymentResponse:
    id, advance_id, paid_date, balance_amount, paid_amount = (None,)*5

    def get(self):
        return json.dumps(self, default=lambda o: o.__dict__,
                          sort_keys=True, indent=4)

    def set_id(self, id):
        self.id = id

    def set_advance_id(self, advance_id):
        self.advance_id = advance_id

    def set_paid_date(self, paid_date):
        self.paid_date = str(paid_date.date())

    def set_balance_amount(self, balance_amount):
        self.balance_amount = data_roundup(balance_amount)

    def set_paid_amount(self, paid_amount):
        self.paid_amount = data_roundup(paid_amount)

    def set_employee_id(self, employee_id):
        self.employee_id = employee_id