from rest_framework import serializers

from sitemanagement.models import FAQ, News

class FAQMainSerializer(serializers.ModelSerializer):
    class Meta:
        model = FAQ
        fields = "__all__"

class NewsMainSerializer(serializers.ModelSerializer):
    class Meta:
        model = News
        fields= "__all__"
        
class TelegramSerializer(serializers.Serializer):
    """Отправляем сообщение в телеграм канал компании с полями:
    - Источник
    - Имя пользователя
    - Номер телефона
    - Сообщение"""
    
    SOURCE_CHOICES = [
        ("main", "Main"),
        ("admin", "Admin"),
        ("userAccount", "Account"),
        ("contact-us", "Contact Us")
    ]
    
    source = serializers.ChoiceField(choices=SOURCE_CHOICES)
    name = serializers.CharField(max_length=255)
    phone_number = serializers.CharField(max_length=20)
    message = serializers.CharField(max_length=1000)

    def create(self, validated_data):
        return type('TelegramMessage', (), validated_data)