ForestAdmin/django-forest

import uuid
from datetime import datetime
import logging

from django.http import JsonResponse

logger = logging.getLogger(__name__)


class StatsMixin:
    def serialize(self, value):
        if isinstance(value, datetime):
            return value.strftime('%d/%m/%Y')
        return value

    def fill_data(self, data, key, value):
        serialized_key = self.serialize(key)
        if serialized_key in data:
            data[serialized_key] += value
        else:
            data[serialized_key] = value

    def handle_values(self, params, request, queryset=None):
        values = None
        _type = params['type']
        if _type in ('Value', 'Objective'):
            values = self.get_value(params, queryset, request)
        elif _type == 'Pie':
            values = self.get_pie(params, queryset)
        elif _type == 'Line':
            values = self.get_line(params, queryset)
        elif _type == 'Leaderboard':
            values = self.get_leaderboard(params, queryset)

        return values

    def handle_chart(self, params, request, queryset=None):
        res = {
            'data': {
                'attributes': {},
                'type': 'stats',
                'id': uuid.uuid4()
            }}

        if 'type' in params:
            res['data']['attributes'] = {
                'value': self.handle_values(params, request, queryset)
            }
        return res

    def chart(self, params, request, queryset=None):
        try:
            res = self.handle_chart(params, request, queryset)
        except Exception as e:
            logger.exception(e)
            return self.error_response(e)
        else:
            return JsonResponse(res, safe=False)