def get_context_data(self, country_code, **kwargs):
context = super(LibraryView, self).get_context_data(**kwargs)
country = Country.for_code(country_code)
context['country'] = country
context['country_code'] = country_code
context['countries'] = Country.objects.select_related(
'country').prefetch_related('localities', 'publication_set',
'country').all()
context['countries_json'] = json.dumps(
{c.code: c.as_json()
for c in context['countries']})
serializer = DocumentSerializer(context={'request': self.request},
many=True)
docs = DocumentViewSet.queryset.filter(work__country=country)
context['documents_json'] = json.dumps(
serializer.to_representation(docs))
serializer = WorkSerializer(context={'request': self.request},
many=True)
works = WorkViewSet.queryset.filter(country=country)
context['works_json'] = json.dumps(serializer.to_representation(works))
return context
def get_context_data(self, **kwargs):
context = super(WorkAmendmentsView, self).get_context_data(**kwargs)
work = self.object
docs = DocumentViewSet.queryset.filter(work=work).all()
serializer = DocumentSerializer(context={'request': self.request},
many=True)
context['documents_json'] = json.dumps(
serializer.to_representation(docs))
serializer = WorkAmendmentSerializer(context={'request': self.request},
many=True)
amendments = work.amendments.prefetch_related('created_by_user',
'updated_by_user',
'amending_work')
context['amendments_json'] = json.dumps(
serializer.to_representation(amendments))
return context