highchart.py

# coding=utf-8
"""
Munges data into formats appropriate for Highcharts usage

http://api.highcharts.com/highcharts
"""
from datetime import datetime
from dateutil.parser import parse
from itertools import chain


def timestamp(a_datetime):
    return int((a_datetime - datetime(1970,1,1)).total_seconds()) * 1000


def line_chart(entries, name=''):
    rows = []
    
    for entry in entries:
        date = entry.get('dateString') or entry.get('display_time') or entry['date']
        amount = entry.get('sgv', entry.get('amount', entry.get('glucose')))
    
        rows.append({
            'x': timestamp(parse(date)),
            'y': amount,
            'name': name
        })
    
    return rows


def glucose_target_range_chart(targets, *args):
    rows = []

    start_timestamp = None
    end_timestamp = None
    
    for entry in chain(*args):
        date = entry['x']
        if start_timestamp is None or date < start_timestamp:
            start_timestamp = date
        
        if end_timestamp is None or date > end_timestamp:
            end_timestamp = date

    if start_timestamp is not None and end_timestamp is not None:
        start_time = datetime.fromtimestamp(start_timestamp / 1000).time()
        end_time = datetime.fromtimestamp(end_timestamp / 1000).time()
    
        start_target = targets.at(start_time)
    
        # TODO: Parse multiple targets
    
        end_target = targets.at(end_time)
        rows.append({
            'x': start_timestamp,
            'low': end_target['low'],
            'high': end_target['high']
        })
        rows.append({
            'x': end_timestamp,
            'low': end_target['low'],
            'high': end_target['high']
        })

    return rows


def input_history_area_chart(normalized_history):
    basal = []
    bolus = []
    square = []
    carbs = []

    for entry in normalized_history:
        if entry['unit'] == 'U/hour':
            values = [
                {
                    'x': timestamp(parse(entry['start_at'])),
                    'y': entry['amount'],
                    'name': entry['description']
                },
                {
                    'x': timestamp(parse(entry['end_at'])),
                    'y': entry['amount'],
                    'name': entry['description']
                },
                {
                    'x': timestamp(parse(entry['end_at'])),
                    'y': None
                }
            ]

            if entry['type'] == 'TempBasal':
                basal += values
            else:
                square += values
        elif entry['unit'] == 'U':
            bolus += [
                {
                    'x': timestamp(parse(entry['start_at'])),
                    'y': entry['amount'],
                    'name': entry['description']
                },
                {
                    'x': timestamp(parse(entry['end_at'])),
                    'y': None
                },
            ]
        elif entry['unit'] == 'g':
            carbs += [
                {
                    'x': timestamp(parse(entry['start_at'])),
                    'y': entry['amount'],
                    'name': entry['description']
                },
                {
                    'x': timestamp(parse(entry['end_at'])),
                    'y': None
                },
            ]

    return basal, bolus, square, carbs