72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
|
from chartjs.views.lines import BaseLineChartView
|
||
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
||
|
|
from django.utils.decorators import method_decorator
|
||
|
|
from django.views.decorators.csrf import csrf_exempt
|
||
|
|
|
||
|
|
|
||
|
|
class CSRFExemptMixin(object):
|
||
|
|
@method_decorator(csrf_exempt)
|
||
|
|
def dispatch(self, *args, **kwargs):
|
||
|
|
return super(CSRFExemptMixin, self).dispatch(*args, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
class VolumeChartJSONView(CSRFExemptMixin, LoginRequiredMixin, BaseLineChartView):
|
||
|
|
def post(self, request, *args, **kwargs):
|
||
|
|
print("POST")
|
||
|
|
context = self.get_context_data(**kwargs)
|
||
|
|
return self.render_to_response(context)
|
||
|
|
|
||
|
|
def get_context_data(self, **kwargs):
|
||
|
|
data = super(VolumeChartJSONView, self).get_context_data(**kwargs)
|
||
|
|
# data["colors"] = islice(next_color(), 0, 50)
|
||
|
|
print("KWARGS", kwargs)
|
||
|
|
return data
|
||
|
|
|
||
|
|
def get_labels(self):
|
||
|
|
"""Return 7 labels for the x-axis."""
|
||
|
|
return ["January", "February", "March", "April", "May", "June", "July"]
|
||
|
|
|
||
|
|
def get_providers(self):
|
||
|
|
"""Return names of datasets."""
|
||
|
|
return ["Central", "Eastside", "Westside"]
|
||
|
|
|
||
|
|
def get_data(self):
|
||
|
|
"""Return 3 datasets to plot."""
|
||
|
|
|
||
|
|
return [
|
||
|
|
[75, 44, 92, 11, 44, 95, 35],
|
||
|
|
[41, 92, 18, 3, 73, 87, 92],
|
||
|
|
[87, 21, 94, 3, 90, 13, 65],
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class SentimentChartJSONView(CSRFExemptMixin, LoginRequiredMixin, BaseLineChartView):
|
||
|
|
def post(self, request, *args, **kwargs):
|
||
|
|
print("POST")
|
||
|
|
print(request.POST)
|
||
|
|
context = self.get_context_data(**kwargs)
|
||
|
|
return self.render_to_response(context)
|
||
|
|
|
||
|
|
def get_context_data(self, **kwargs):
|
||
|
|
data = super(SentimentChartJSONView, self).get_context_data(**kwargs)
|
||
|
|
# data["colors"] = islice(next_color(), 0, 50)
|
||
|
|
print("KWARGS", kwargs)
|
||
|
|
return data
|
||
|
|
|
||
|
|
def get_labels(self):
|
||
|
|
"""Return 7 labels for the x-axis."""
|
||
|
|
return ["January", "February", "March", "April", "May", "June", "July"]
|
||
|
|
|
||
|
|
def get_providers(self):
|
||
|
|
"""Return names of datasets."""
|
||
|
|
return ["Central", "Eastside", "Westside"]
|
||
|
|
|
||
|
|
def get_data(self):
|
||
|
|
"""Return 3 datasets to plot."""
|
||
|
|
|
||
|
|
return [
|
||
|
|
[75, 44, 92, 11, 44, 95, 35],
|
||
|
|
[41, 92, 18, 3, 73, 87, 92],
|
||
|
|
[87, 21, 94, 3, 90, 13, 65],
|
||
|
|
]
|