com2

web/chart_views.py

@@ -0,0 +1,148 @@
+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')