forked from github/plane
abaa65b4b7
* dev: initialize plane analytics * dev: plane analytics endpoint * dev: update endpoint to give data with segments as well * dev: analytics with count and effort paramters * feat: analytics endpoints * feat: saved analytics * dev: remove print logs * dev: rename x_axis to dimension in response * dev: remove color queries * dev: update query for None values * feat: analytics export * dev: update code structure send color when state or label and fix none count * dev: uncomment try catch block * dev: fix segment keyerror * dev: default analytics endpoint * dev: fix segmented results * dev: default analytics endpoint and colors for segment * dev: total issues and open issues by state * dev: segment colors * dev: fix total issue annotate * dev: effort segmentation * dev: total estimates and open estimates * fix: effort when not segmented * dev: send avatar for default analytics
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
# Python imports
|
|
from itertools import groupby
|
|
|
|
# Django import
|
|
from django.db import models
|
|
from django.db.models import Count, DateField, F, Sum, Value, Case, When, Q
|
|
from django.db.models.functions import Cast, Concat, Coalesce
|
|
|
|
|
|
def build_graph_plot(queryset, x_axis, y_axis, segment=None):
|
|
if x_axis in ["created_at", "completed_at"]:
|
|
queryset = queryset.annotate(dimension=Cast(x_axis, DateField()))
|
|
x_axis = "dimension"
|
|
else:
|
|
queryset = queryset.annotate(dimension=F(x_axis))
|
|
x_axis = "dimension"
|
|
|
|
if x_axis in ["created_at", "start_date", "target_date", "completed_at"]:
|
|
queryset = queryset.exclude(x_axis__is_null=True)
|
|
|
|
queryset = queryset.values(x_axis)
|
|
|
|
# Group queryset by x_axis field
|
|
|
|
if y_axis == "issue_count":
|
|
queryset = (
|
|
queryset.annotate(
|
|
is_null=Case(
|
|
When(dimension__isnull=True, then=Value("None")),
|
|
default=Value("not_null"),
|
|
output_field=models.CharField(max_length=8),
|
|
),
|
|
dimension_ex=Coalesce("dimension", Value("null")),
|
|
)
|
|
.values("dimension")
|
|
)
|
|
if segment:
|
|
queryset = queryset.annotate(segment=F(segment)).values("dimension", "segment")
|
|
else:
|
|
queryset = queryset.values("dimension")
|
|
|
|
queryset = queryset.annotate(count=Count("*")).order_by("dimension")
|
|
|
|
|
|
if y_axis == "effort":
|
|
queryset = queryset.annotate(effort=Sum("estimate_point")).order_by(x_axis)
|
|
if segment:
|
|
queryset = queryset.annotate(segment=F(segment)).values("dimension", "segment", "effort")
|
|
else:
|
|
queryset = queryset.values("dimension", "effort")
|
|
|
|
result_values = list(queryset)
|
|
grouped_data = {}
|
|
for date, items in groupby(result_values, key=lambda x: x[str("dimension")]):
|
|
grouped_data[str(date)] = list(items)
|
|
|
|
return grouped_data
|