from .alliance_audience import AllianceAudience


class CsvAudience(AllianceAudience):
    """
    Generating general data set for csv export
    """

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

    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",
                "55": "city",
                "17": "region",
                "12": "country",
                "18": "zip",
                "48": "gender",
            }
        )

        return data