mybuddy/reports/graphs/temperature_change.py

# -*- coding: utf-8 -*-
from django.utils.translation import gettext as _

import plotly.offline as plotly
import plotly.graph_objs as go

from reports import utils


def temperature_change(objects):
    """
    Create a graph showing temperature over time.
    :param objects: a QuerySet of Temperature instances.
    :returns: a tuple of the graph's html and javascript.
    """
    objects = objects.order_by("-time")

    trace = go.Scatter(
        name=_("Temperature"),
        x=list(objects.values_list("time", flat=True)),
        y=list(objects.values_list("temperature", flat=True)),
    )

    layout_args = utils.default_graph_layout_options()
    layout_args["barmode"] = "stack"
    layout_args["title"] = _("<b>Temperature</b>")
    layout_args["xaxis"]["title"] = _("Time")
    layout_args["xaxis"]["type"] = "date"
    layout_args["xaxis"]["autorange"] = True
    layout_args["xaxis"]["autorangeoptions"] = utils.autorangeoptions(trace.x)
    layout_args["xaxis"]["rangeselector"] = utils.rangeselector_time()
    layout_args["yaxis"]["title"] = _("Temperature")

    fig = go.Figure({"data": [trace], "layout": go.Layout(**layout_args)})
    output = plotly.plot(fig, output_type="div", include_plotlyjs=False)
    return utils.split_graph_output(output)