Fixed employee establishments method

apps/establishment/serializers/back.py

@@ -526,3 +526,23 @@ class EstablishmentAdminListSerializer(UserShortSerializer):
             'username',
             'email'
         ]
+
+
+class EstablishmentEmployeeListSerializer(serializers.ModelSerializer):
+    """Establishments from employee serializer"""
+
+    restaurant = EstablishmentListCreateSerializer(read_only=True, source='establishment')
+    position = PositionBackSerializer(read_only=True)
+    state = serializers.CharField(read_only=True, source='status')
+    start = serializers.DateTimeField(read_only=True, source='from_date')
+    end = serializers.DateTimeField(read_only=True, source='to_date')
+
+    class Meta:
+        model = models.EstablishmentEmployee
+        fields = [
+            'restaurant',
+            'position',
+            'state',
+            'start',
+            'end',
+        ]

apps/establishment/views/back.py

@@ -1,6 +1,4 @@
 """Establishment app views."""
-from django.core.exceptions import ObjectDoesNotExist
-from django.http import Http404
 from django.shortcuts import get_object_or_404
 from django_filters.rest_framework import DjangoFilterBackend
 from rest_framework import generics, permissions
@@ -38,13 +36,13 @@ class EmployeeEstablishmentsListView(generics.ListAPIView):
     """Establishment by employee list view."""
 
     permission_classes = [IsWineryReviewer | IsCountryAdmin | IsEstablishmentManager]
-    queryset = models.Establishment.objects.all()
+    queryset = models.EstablishmentEmployee.objects.all()
-    serializer_class = serializers.EstablishmentListCreateSerializer
+    serializer_class = serializers.EstablishmentEmployeeListSerializer
 
     def get_queryset(self):
         pk = self.kwargs.get('pk')
         employee = get_object_or_404(models.Employee, pk=pk)
-        return employee.establishments.with_extended_related()
+        return employee.establishmentemployee_set.all()
 
 
 class EmployeePositionsListView(generics.ListAPIView):