from rest_framework import serializers
from partner.models import Partner


class PartnerSerializer(serializers.ModelSerializer):
    backlink_url = serializers.CharField(source="url")
    partnership_icon = serializers.CharField()
    partnership_name = serializers.CharField()

    class Meta:
        model = Partner
        fields = (
            "backlink_url",
            "partnership_icon",
            "partnership_name"
        )

    def validate(self, data):
        data["image"] = partnership_to_image_url.get(data["partnership_name"]).get(data["partnership_icon"])
        data.pop("partnership_name")
        data.pop("partnership_icon")
        return data

    def create(self, validated_data):
        return Partner.objects.create(**validated_data)


partnership_to_image_url = {
    # partnership_name
    "tables-auberges": {
        # partnership_icon
        "Table Gastronomique":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/tables-auberges/Table Gastronomique-c1908231742bc0d0909c03cdac723c1daa3ee620134e8a6fbfae78efdaab8263.png",
        "Bistrot Gourmand":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/tables-auberges/Bistrot Gourmand-fdf0527d8ed05d9c7d26456e531301325582f04e1f8e5d4c283ee4eac367dba1.png",
        "Auberges de Village":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/tables-auberges/Auberges de Village-39aab1bbc678a6a5f7e0a2c9b8565bb6d21b01294371533580cdff0d9aa80da0.png",
        "Table de Prestige":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/tables-auberges/Table de Prestige-6bf4ab0fcc56d89ffc97cc71553f0b1d21b92f4b5ef7656b832ab03e5c6ff31c.png",
        "Table de Terroir":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/tables-auberges/Table de Terroir-e72a35a60c4a4575c162398d7355fc77787f368ff240ff8725dc24a37a497be7.png"
    },
    "riedel": {
        "riedel":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/riedel/riedel-1fe0fa8298ae76d0a2de6e00316d2dd4357cace659cc0398629ae4c117bd114e.png"
    },
    "lavazza": {
        "lavazza":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/lavazza/lavazza-9a013b9f954dd63ba871019a1fcd55ea5cb84b38ab2dbff47ee9464f671195bb.png"
    },
    "kaviari": {
        "Kaviari Paris":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/kaviari/Kaviari Paris-e6c4fe81a1d5257854f9b8e28cb244b01a9129ed01056cb98e809f70ec7b5fcf.png"
    },
    "mutti": {
        "mutti":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/mutti/mutti-86f8514b03e096b8e6843e8885756d311486c461a5b8c4f2f10f75995b146814.png"
    },
    "nespresso": {
        "nespresso":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/nespresso/nespresso-c401e553731c68a364754fe512f25889398d42f98bf8edbe1233695b546d1c5d.png"
    },
    "le-manoir": {
        "le-manoir":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/le-manoir/le-manoir-d6a616bebca2948c147ed7c58d9883682ee2d4a3779dea98e9d1a94f17e14771.png"
    },
    "discovery-cheque": {
        "discovery-cheque":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/discovery-cheque/discovery-cheque-665a3c60d3d6302897863b90245f405979760a27bea06c98aa4130b2c61276fb.png"
    },
    "mineralwater-selters": {
        "Mineral Water Selters":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/mineralwater-selters/Mineral Water Selters-2bac82327955c4357b51feff50bd6a2466705f85cb8e59875cb0192a161fd1d4.png"
    },
    "mineral-water-acquamorelli": {
        "Mineral Water Acqua Morelli":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/mineral-water-acquamorelli/Mineral Water Acqua Morelli-4032ba6b4da7f79dcf005809423fc663cfb49199d6ce95f3cfe99a7705359ac4.png"
    },
    "mastercard-maestro": {
        "MastercardMaestro":
            "https://1dc3f33f6d.optimicdn.com/assets/establishment-backlinks/mastercard-maestro/MastercardMaestro-718680bc48d98c4a7417a14a62e4ba2d3af980361256739d980e7b453552b5d8.png"
    }
}