Объединить набор запросов из двух баз данных в Django Rest - PullRequest
Новая
       23

Объединить набор запросов из двух баз данных в Django Rest

0 голосов
/ 21 мая 2019

У меня в API есть два маршрута, которые выглядят так:

http://mywebsite/websites/website_1/annonces/

http://mywebsite/websites/website_2/annonces/

Мне нужно сделать ListAPIView, объединяющий эти 2 маршрута, но каждый маршрут вызывает свою собственную базу данных.

Две базы данных созданы с использованием одной и той же модели django. Я сделал две базы данных, потому что они лучше подходят моей архитектуре.

Проблема в том, что в моих базах данных нет столбца, указывающего на веб-сайт, с которого получены записи. Записи различаются только по именам своей базы данных.

Я хочу получить все записи по одному маршруту, но также могу сказать, из какой они базы данных, в ответе json. 

 class Annonce(models.Model):
     annonce_type = models.CharField(max_length=200, blank=True, null=True)
     annonce_id = models.CharField(unique=True, max_length=200, blank=True, null=True)
     url = models.TextField(blank=True, null=True)
     region = models.TextField(blank=True, null=True)

 class AnnoncesList(generics.ListAPIView):
     authentication_classes = ()
     permission_classes = ()
     serializer_class = AnnonceListSerializer
     pagination_class = LargeResultsSetPagination
     filter_backends = (DjangoFilterBackend,)
     filterset_fields = ('advert_type', 'asset_type', 'sales_type', 'price', 'area', 'department', 'department_id', 'city', 'postal_code')

     def get_queryset(self):
         queryset = Annonce.objects.using(self.kwargs["website_name"]).all()
         return queryset

1 Ответ

1 голос
/ 22 мая 2019

Создайте набор запросов для каждой базы данных, затем используйте annotate(), чтобы добавить столбец website_name для каждой записи в наборе запросов.Объедините наборы запросов в список (установите флажок this ) (попадет во все элементы базы данных), убедитесь, что наборы запросов уже правильно отфильтрованы. 

from itertools import chain

from rest_framework.generics import ListAPIView
from rest_framework.response import Response

from django.db.models import Value, CharField


class AnnonceMergedList(ListAPIView):
    serializer_class = AnnonceMergedListSerializer
    queryset = Annonce.objects.all()

    def list(self, request, **kwargs):
        # Make the querysets for each database
        q1 = self.get_queryset().using('website_1').annotate(website_name=Value('website_1', CharField()))
        q2 = self.get_queryset().using('website_2').annotate(website_name=Value('website_2', CharField()))

        # Filtering the querysets
        q1 = self.filter_queryset(q1)
        q2 = self.filter_queryset(q2)

        # Merge (hit items on database)
        data = list(chain(q1, q2))

        serializer = self.get_serializer(data, many=True)
        return Response(serializer.data)

Сериализатор для этого представления долженполучить website_name для отображения веб-сайта записи с 

class AnnonceMergedListSerializer(serializers.ModelSerializer):
    website_name = serializers.CharField(read_only=True) # Field from annotation

    class Meta:
        model = Annonce
        fields = ('website_name', ...)
...