from .alliance_audience import AllianceAudience


class KlaviyoAudience(AllianceAudience):
    """API doc describing prebuild Klaviyo properties
    - Add or Update User https://apidocs.klaviyo.com/reference/track-identify#identify
    """

    source_attribute_ids = [
        1,  # userEmail
        49,  # enrUserFirstName
        4,  # userLastName
        # 6,  # userPhone
        55,  # enrLocality
        17,  # userState
        12,  # userCountry
        18,  # userZip
    ]

    def __init__(self, collection_id, schema: str, alliance_schema: str = None):
        super().__init__(
            collection_id=collection_id, schema=schema, alliance_schema=alliance_schema
        )

    def get_audience_data(self):
        data = self._generate_available_data()
        data = data.rename(
            columns={
                "1": "email",
                "49": "first_name",
                "4": "last_name",
                # '6': 'phone_number',
                "55": "city",
                "17": "region",
                "12": "country",
                "18": "zip",
            }
        )

        return data