Пример #1
0
 def get(self, request, *args, **kwargs):
     return self.render_to_response(context = {
         'models': {
             model_name: models.get_finance_class(model_name)
             for model_name in finance_categories
         }
     })
Пример #2
0
 def get(self, request, *args, **kwargs):
     model_class = models.get_finance_class(self.kwargs['category'])
     finance_object = get_object_or_404(model_class, pk = int(self.kwargs['pk']))
     context = self._get_context()
     context.update({
         'finance_object': finance_object,
     })
     return self.render_to_response(context = context)
Пример #3
0
 def _get_context(self):
     category = self.kwargs['category']
     model = models.get_finance_class(category)
     return {
         'category': category,
         'model': model,
         'need_log': model.need_log
     }
Пример #4
0
def create_viewset(model_name):

    model_class = models.get_finance_class(model_name)
    log_model_class = models.get_log_class(model_name)

    class FinanceAPIViewSet(viewsets.ReadOnlyModelViewSet, ListCacheMixin):

        model = model_class
        serializer_class = serializers.get_serializer_class(model_class)

    class FinanceLogAPIViewSet(viewsets.ReadOnlyModelViewSet, ListCacheMixin):

        model = log_model_class
        serializer_class = serializers.get_log_serializer_class(log_model_class)

        def get_queryset(self):
            pk = int(self.kwargs['finance_pk'])
            return log_model_class.objects.filter(finance_object = pk)

    return (FinanceAPIViewSet, FinanceLogAPIViewSet)