import json


class PaycomponentsegmentmappingResponse:
    paycomponent_id, segment_id = (None,) * 2

    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_paycomponent_id(self, paycomponent_id):
        self.paycomponent = paycomponent_id

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

    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