import json


class PayrollmastersmappingResponse:
    id, template_id, segment_id, paycomponent_id, percentage, amount = (None,)*6

    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_template_id(self, template_id):
        self.template_id = template_id

    def set_segment_id(self, segment_id):
        self.segment_id = segment_id

    def set_paycompoent_id(self, paycomponent_id):
        self.paycomponent_id = paycomponent_id

    def set_percentage(self, percentage):
        self.percentage = str(percentage)

    def set_amount(self, amount):
        self.amount = str(amount)

    def set_paycomponent_data(self, paycomponent_id, arr):
        self.paycomponent = None
        for i in arr:
            if i.id == paycomponent_id:
                self.paycomponent =i
                break

    def set_segment_data(self, segment_id, arr):
        self.segment = None
        for i in arr:
            if i.id == segment_id:
                self.segment = i
                break

    def set_template_data(self, template_id, arr):
        self.template = None
        for i in arr:
            if i.id == template_id:
                self.template = i
                break