from rest_framework import routers, viewsets, serializers

from mail.models import Email, EmailAttachment


class AttachmentSerializer(serializers.ModelSerializer):
    class Meta:
        model = EmailAttachment
        fields = ['hash', 'mime_type', 'name']


class EmailSerializer(serializers.ModelSerializer):
    class Meta:
        model = Email
        fields = '__all__'


class EmailViewSet(viewsets.ModelViewSet):
    serializer_class = EmailSerializer
    queryset = Email.objects.all()


router = routers.SimpleRouter()
router.register(r'mails', EmailViewSet, basename='mails')

urlpatterns = router.urls