# -*- 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'] = _('Sleep Pattern') 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'))