mirror of https://github.com/snachodog/mybuddy.git
204 lines
7.2 KiB
Python
204 lines
7.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
from collections import OrderedDict
|
|
|
|
from django.utils import timezone, formats
|
|
from django.utils.translation import gettext as _
|
|
|
|
import plotly.offline as plotly
|
|
import plotly.graph_objs as go
|
|
|
|
from core.utils import duration_string
|
|
|
|
from reports import utils
|
|
|
|
|
|
def sleep_pattern(instances):
|
|
"""
|
|
Create a graph showing blocked out periods of sleep during each day.
|
|
:param instances: a QuerySet of Sleep instances.
|
|
:returns: a tuple of the the graph's html and javascript.
|
|
"""
|
|
times = {}
|
|
labels = {}
|
|
last_end_time = None
|
|
adjustment = None
|
|
for instance in instances:
|
|
start_time = timezone.localtime(instance.start)
|
|
end_time = timezone.localtime(instance.end)
|
|
start_date = start_time.date().isoformat()
|
|
end_date = end_time.date().isoformat()
|
|
duration = instance.duration
|
|
|
|
# Ensure that lists are initialized for the start and end date (as they
|
|
# may be different dates).
|
|
if start_date not in times:
|
|
times[start_date] = []
|
|
if start_date not in labels:
|
|
labels[start_date] = []
|
|
if end_date not in times:
|
|
times[end_date] = []
|
|
if end_date not in labels:
|
|
labels[end_date] = []
|
|
|
|
# Check if the previous entry crossed midnight (see below).
|
|
if adjustment:
|
|
_add_adjustment(adjustment, times, labels)
|
|
last_end_time = timezone.localtime(adjustment['end_time'])
|
|
adjustment = None
|
|
|
|
# If the dates do not match, set up an adjustment for the next day.
|
|
if end_time.date() != start_time.date():
|
|
adj_start_time = end_time.replace(hour=0, minute=0, second=0)
|
|
adjustment = {
|
|
'column': end_time.date().isoformat(),
|
|
'start_time': adj_start_time,
|
|
'end_time': end_time,
|
|
'duration': end_time - adj_start_time
|
|
}
|
|
|
|
# Adjust end_time for the current entry.
|
|
end_time = end_time.replace(
|
|
year=start_time.year, month=start_time.month,
|
|
day=start_time.day, hour=23, minute=59, second=0)
|
|
duration = end_time - start_time
|
|
|
|
if not last_end_time:
|
|
last_end_time = start_time.replace(hour=0, minute=0, second=0)
|
|
|
|
# Awake time.
|
|
times[start_date].append((start_time - last_end_time).seconds/60)
|
|
labels[start_date].append(None)
|
|
|
|
# Asleep time.
|
|
times[start_date].append(duration.seconds/60)
|
|
labels[start_date].append(_format_label(duration, start_time,
|
|
end_time))
|
|
|
|
# Update the previous entry duration if an offset change occurred.
|
|
# This can happen when an entry crosses a daylight savings time change.
|
|
if start_time.utcoffset() != end_time.utcoffset():
|
|
diff = start_time.utcoffset() - end_time.utcoffset()
|
|
duration -= timezone.timedelta(seconds=diff.seconds)
|
|
yesterday = (end_time - timezone.timedelta(days=1))
|
|
yesterday = yesterday.date().isoformat()
|
|
times[yesterday][len(times[yesterday]) - 1] = duration.seconds/60
|
|
labels[yesterday][len(times[yesterday]) - 1] = _format_label(
|
|
duration, start_time, end_time)
|
|
|
|
last_end_time = end_time
|
|
|
|
# Handle any left over adjustment (if the last entry crossed midnight).
|
|
if adjustment:
|
|
_add_adjustment(adjustment, times, labels)
|
|
|
|
# Create dates for x-axis using a 12:00:00 time to ensure correct
|
|
# positioning of bars (covering entire day).
|
|
dates = []
|
|
for time in list(times.keys()):
|
|
dates.append('{} 12:00:00'.format(time))
|
|
|
|
traces = []
|
|
color = 'rgba(255, 255, 255, 0)'
|
|
|
|
# Set iterator and determine maximum iteration for dates.
|
|
i = 0
|
|
max_i = 0
|
|
for date_times in times.values():
|
|
max_i = max(len(date_times), max_i)
|
|
while i < max_i:
|
|
y = {}
|
|
text = {}
|
|
for date in times.keys():
|
|
try:
|
|
y[date] = times[date][i]
|
|
text[date] = labels[date][i]
|
|
except IndexError:
|
|
y[date] = None
|
|
text[date] = None
|
|
i += 1
|
|
traces.append(go.Bar(
|
|
x=dates,
|
|
y=list(y.values()),
|
|
hovertext=list(text.values()),
|
|
# `hoverinfo` is deprecated but if we use the new `hovertemplate`
|
|
# the "filler" areas for awake time get a hover that says "null"
|
|
# and there is no way to prevent this currently with Plotly.
|
|
hoverinfo='text',
|
|
marker={'color': color},
|
|
showlegend=False,
|
|
))
|
|
if color == 'rgba(255, 255, 255, 0)':
|
|
color = 'rgb(35, 110, 150)'
|
|
else:
|
|
color = 'rgba(255, 255, 255, 0)'
|
|
|
|
layout_args = utils.default_graph_layout_options()
|
|
layout_args['margin']['b'] = 100
|
|
|
|
layout_args['barmode'] = 'stack'
|
|
layout_args['bargap'] = 0
|
|
layout_args['hovermode'] = 'closest'
|
|
layout_args['title'] = _('<b>Sleep Pattern</b>')
|
|
layout_args['height'] = 800
|
|
|
|
layout_args['xaxis']['title'] = _('Date')
|
|
layout_args['xaxis']['tickangle'] = -65
|
|
layout_args['xaxis']['tickformat'] = '%b %e\n%Y'
|
|
layout_args['xaxis']['ticklabelmode'] = 'period'
|
|
layout_args['xaxis']['rangeselector'] = utils.rangeselector_date()
|
|
|
|
start = timezone.localtime().strptime('12:00 AM', '%I:%M %p')
|
|
ticks = OrderedDict()
|
|
ticks[0] = start.strftime('%I:%M %p')
|
|
for i in range(0, 60*24, 30):
|
|
ticks[i] = formats.time_format(start + timezone.timedelta(minutes=i),
|
|
'TIME_FORMAT')
|
|
|
|
layout_args['yaxis']['title'] = _('Time of day')
|
|
layout_args['yaxis']['range'] = [24*60, 0]
|
|
layout_args['yaxis']['tickmode'] = 'array'
|
|
layout_args['yaxis']['tickvals'] = list(ticks.keys())
|
|
layout_args['yaxis']['ticktext'] = list(ticks.values())
|
|
layout_args['yaxis']['tickfont'] = {'size': 10}
|
|
|
|
fig = go.Figure({
|
|
'data': traces,
|
|
'layout': go.Layout(**layout_args)
|
|
})
|
|
output = plotly.plot(fig, output_type='div', include_plotlyjs=False)
|
|
return utils.split_graph_output(output)
|
|
|
|
|
|
def _add_adjustment(adjustment, times, labels):
|
|
"""
|
|
Adds "adjustment" data for entries that cross midnight.
|
|
:param adjustment: Column, start time, end time, and duration of entry.
|
|
:param times: Graph times data.
|
|
:param labels: Graph labels data.
|
|
"""
|
|
# Fake (0) entry to keep the color switching logic working.
|
|
times[adjustment['column']].append(0)
|
|
labels[adjustment['column']].append(0)
|
|
|
|
# Real adjustment entry.
|
|
times[adjustment['column']].append(
|
|
adjustment['duration'].seconds / 60)
|
|
labels[adjustment['column']].append(_format_label(
|
|
adjustment['duration'],
|
|
adjustment['start_time'],
|
|
adjustment['end_time']))
|
|
|
|
|
|
def _format_label(duration, start_time, end_time):
|
|
"""
|
|
Formats a time block label.
|
|
:param duration: Duration.
|
|
:param start_time: Start time.
|
|
:param end_time: End time.
|
|
:return: Formatted string with duration, start, and end time.
|
|
"""
|
|
return 'Asleep {} ({} to {})'.format(
|
|
duration_string(duration),
|
|
formats.time_format(start_time, 'TIME_FORMAT'),
|
|
formats.time_format(end_time, 'TIME_FORMAT'))
|