1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
##############################################################################
# Copyright (c) 2020 Parker Berberian, Sawyer Bergeron, Sean Smith and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
import os
from booking.models import Booking
from resource_inventory.models import ResourceQuery, ResourceProfile
from datetime import datetime, timedelta
from collections import Counter
import pytz
class StatisticsManager(object):
@staticmethod
def getContinuousBookingTimeSeries(span=28):
"""
Calculate Booking usage data points.
Gathers all active bookings that fall in interval [(now - span), (now + 1 week)].
x data points are every 12 hours
y values are the integer number of bookings/users active at time
"""
anuket_colors = [
'#6BDAD5', # Turquoise
'#E36386', # Pale Violet Red
'#F5B335', # Sandy Brown
'#007473', # Teal
'#BCE194', # Gainsboro
'#00CE7C', # Sea Green
]
lfedge_colors = [
'#0049B0',
'#B481A5',
'#6CAFE4',
'#D33668',
'#28245A'
]
x = []
y = []
users = []
projects = []
profiles = {str(profile): [] for profile in ResourceProfile.objects.all()}
now = datetime.now(pytz.utc)
delta = timedelta(days=span)
start = now - delta
end = now + timedelta(weeks=1)
bookings = Booking.objects.filter(
start__lte=end,
end__gte=start
).prefetch_related("collaborators")
# get data
while start <= end:
active_users = 0
books = bookings.filter(
start__lte=start,
end__gte=start
).prefetch_related("collaborators")
for booking in books:
active_users += booking.collaborators.all().count() + 1
x.append(str(start.month) + '-' + str(start.day))
y.append(books.count())
step_profiles = Counter([
str(config.profile)
for book in books
for config in book.resource.template.getConfigs()
])
for profile in ResourceProfile.objects.all():
profiles[str(profile)].append(step_profiles[str(profile)])
users.append(active_users)
start += timedelta(hours=12)
in_use = len(ResourceQuery.filter(working=True, booked=True))
not_in_use = len(ResourceQuery.filter(working=True, booked=False))
maintenance = len(ResourceQuery.filter(working=False))
projects = [x.project for x in bookings]
proj_count = sorted(Counter(projects).items(), key=lambda x: x[1])
project_keys = [proj[0] for proj in proj_count[-5:]]
project_keys = ['None' if x is None else x for x in project_keys]
project_counts = [proj[1] for proj in proj_count[-5:]]
resources = {key: [x, value] for key, value in profiles.items()}
return {
"resources": resources,
"booking": [x, y],
"user": [x, users],
"utils": [in_use, not_in_use, maintenance],
"projects": [project_keys, project_counts],
"colors": anuket_colors if os.environ.get('TEMPLATE_OVERRIDE_DIR') == 'laas' else lfedge_colors
}
|
