from rest_framework.serializers import ModelSerializer
from apps.pos_device import models as pos_models


class ProviderCompanySerializer(ModelSerializer):
    class Meta:
        model = pos_models.ProviderCompany
        fields = '__all__'


class DeviceSerializer(ModelSerializer):
    class Meta:
        model = pos_models.Device
        fields = '__all__'

    def to_representation(self, instance):
        """ custom output of serializer """
        representation = super().to_representation(instance)

        representation['organization'] = {'name': instance.organization.name}

        return representation


class DeviceVersionSerializer(ModelSerializer):
    class Meta:
        model = pos_models.DeviceVersion
        fields = '__all__'


class SessionSerializer(ModelSerializer):
    class Meta:
        model = pos_models.Sessions
        fields = '__all__'


class DeviceAssignmentSerializer(ModelSerializer):
    class Meta:
        model = pos_models.DeviceAssignment
        fields = '__all__'


class StakeHoldersSerializer(ModelSerializer):
    class Meta:
        model = pos_models.StakeHolders
        fields = '__all__'