149 lines
4.5 KiB
Python
149 lines
4.5 KiB
Python
import io
|
|
from collections.__init__ import defaultdict
|
|
from datetime import timedelta
|
|
|
|
import numpy as np
|
|
from django.http import HttpResponse
|
|
from django.views.generic import DetailView
|
|
|
|
from web.models import User, ActivityLog, Activity
|
|
|
|
|
|
class UserChartsPie(DetailView):
|
|
model = User
|
|
pk_url_kwarg = 'user_id'
|
|
|
|
def render_to_response(self, context, **response_kwargs):
|
|
import matplotlib
|
|
matplotlib.use('Agg')
|
|
import matplotlib.pyplot as plt
|
|
fig = plt.figure(figsize=(6, 6), dpi=100)
|
|
ax = fig.add_subplot(111)
|
|
|
|
total_time = defaultdict(float)
|
|
|
|
logs = ActivityLog.objects.filter(user=self.object).all()
|
|
for log in logs:
|
|
total_time[str(log.activity)] += (log.end_time - log.start_time).total_seconds()
|
|
|
|
sectors = sorted(list(total_time.items()), key=lambda x: x[1])
|
|
|
|
ax.pie(x=[i[1] for i in sectors], labels=[i[0] for i in sectors])
|
|
|
|
buf = io.BytesIO()
|
|
fig.savefig(buf, format='png')
|
|
b = buf.getvalue()
|
|
return HttpResponse(b, content_type='image/png')
|
|
|
|
|
|
class UserChartsActivityAll(DetailView):
|
|
model = User
|
|
pk_url_kwarg = 'user_id'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context['activity'] = Activity.objects.filter(id=self.kwargs['activity_id']).first()
|
|
return context
|
|
|
|
def bar_chart(self, plt, ax, logs, day_l, day_r, day_count, days):
|
|
day_seconds = np.zeros(day_count)
|
|
|
|
for log in logs:
|
|
for i, day in enumerate(days):
|
|
l = max(log.start_time, day)
|
|
r = min(log.end_time, day + timedelta(days=1))
|
|
|
|
if r > l:
|
|
day_seconds[i] += (r - l).total_seconds()
|
|
|
|
day_hours = day_seconds / timedelta(hours=1).total_seconds()
|
|
|
|
ax.bar(days, day_hours)
|
|
plt.xticks(days, [f'{i:%m-%d}' for i in days], rotation=60)
|
|
plt.ylabel('кол-во часов')
|
|
|
|
def tracker_chart(self, plt, ax, logs, day_l, day_r, day_count, days):
|
|
ys = []
|
|
widths = []
|
|
lefts = []
|
|
|
|
for log in logs:
|
|
for i, day in enumerate(days):
|
|
l = max(log.start_time, day)
|
|
r = min(log.end_time, day + timedelta(days=1))
|
|
|
|
if r > l:
|
|
ys.append(day)
|
|
widths.append((r - l) / timedelta(hours=1))
|
|
lefts.append((l - day) / timedelta(hours=1))
|
|
|
|
ax.barh(y=ys, width=widths, left=lefts, zorder=2)
|
|
plt.xlim(0, 24)
|
|
plt.xticks(range(24), [f'{i:02d}:00' for i in range(24)])
|
|
plt.yticks(days, [f'{i:%m-%d}' for i in days])
|
|
plt.grid(True, zorder=1)
|
|
|
|
def render_to_response(self, context, **response_kwargs):
|
|
logs = list(ActivityLog.objects.filter(user=self.object, activity=context['activity']).order_by('start_time').all())
|
|
|
|
day_r = logs[-1].end_time.replace(hour=0, minute=0, second=0, microsecond=0)
|
|
day_l = max(
|
|
logs[0].start_time.replace(hour=0, minute=0, second=0, microsecond=0),
|
|
day_r - timedelta(days=14)
|
|
)
|
|
|
|
day_count = (day_r - day_l) // timedelta(days=1) + 1
|
|
|
|
days = [day_l + timedelta(days=1) * i for i in range(day_count)]
|
|
|
|
import matplotlib
|
|
matplotlib.use('Agg')
|
|
import matplotlib.pyplot as plt
|
|
fig = plt.figure(figsize=(16, 12), dpi=100)
|
|
|
|
ax = fig.add_subplot(211)
|
|
self.bar_chart(plt, ax, logs, day_l, day_r, day_count, days)
|
|
|
|
ax = fig.add_subplot(212)
|
|
self.tracker_chart(plt, ax, logs, day_l, day_r, day_count, days)
|
|
|
|
buf = io.BytesIO()
|
|
fig.savefig(buf, format='png')
|
|
b = buf.getvalue()
|
|
return HttpResponse(b, content_type='image/png')
|
|
|
|
|
|
class UserActivityChartsTracker(DetailView):
|
|
model = User
|
|
pk_url_kwarg = 'user_id'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context['activity'] = Activity.objects.filter(id=self.kwargs['activity_id']).first()
|
|
return context
|
|
|
|
def render_to_response(self, context, **response_kwargs):
|
|
print('start_tracker')
|
|
import matplotlib
|
|
matplotlib.use('Agg')
|
|
import matplotlib.pyplot as plt
|
|
fig = plt.figure(figsize=(16, 6), dpi=100)
|
|
ax = fig.add_subplot(111)
|
|
|
|
logs = list(ActivityLog.objects.filter(user=self.object, activity=context['activity']).order_by('start_time').all())
|
|
|
|
day_r = logs[-1].end_time.replace(hour=0, minute=0, second=0, microsecond=0)
|
|
day_l = max(
|
|
logs[0].start_time.replace(hour=0, minute=0, second=0, microsecond=0),
|
|
day_r - timedelta(days=14)
|
|
)
|
|
|
|
day_count = (day_r - day_l) // timedelta(days=1) + 1
|
|
|
|
days = [day_l + timedelta(days=1) * i for i in range(day_count)]
|
|
|
|
buf = io.BytesIO()
|
|
fig.savefig(buf, format='png')
|
|
b = buf.getvalue()
|
|
return HttpResponse(b, content_type='image/png')
|