frontend/croschart/views.py

# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Entry point for django urls code to invoke views."""

import datetime

from django.http import HttpResponse
from django.shortcuts import render_to_response

import autotest_lib.frontend.croschart.validators as validators
import autotest_lib.frontend.croschart.chartviews as chartviews
import autotest_lib.frontend.croschart.labtest.models as labtest_models
import autotest_lib.frontend.croschart.perfchart.models as perfchart_models
import autotest_lib.frontend.croschart.releasereport.models as release_models
import autotest_lib.frontend.croschart.testreport.models as testreport_models
import autotest_lib.frontend.croschart.reportviews as reportviews

from autotest_lib.frontend.croschart.charterrors import ChartInputError


VLISTS = {
    'chart': {
        'from_build': [validators.CrosChartValidator,
                       validators.BuildRangeValidator],
        'from_date': [validators.CrosChartValidator,
                      validators.DateRangeValidator],
        'interval': [validators.CrosChartValidator,
                     validators.IntervalRangeValidator]},
    'chartplatforms': {
        'from_build': [validators.CrosPlatformsValidator,
                       validators.BuildRangeValidator],
        'from_date': [validators.CrosPlatformsValidator,
                      validators.DateRangeValidator],
        'interval': [validators.CrosPlatformsValidator,
                     validators.IntervalRangeValidator]},
    'chartreport': {
        'from_build': [validators.CrosReportValidator,
                       validators.BuildRangeValidator],
        'from_date': [validators.CrosReportValidator,
                      validators.DateRangeValidator],
        'interval': [validators.CrosReportValidator,
                     validators.IntervalRangeValidator]},
    'releasereport': {
        'from_build': [validators.CrosReportValidator,
                       validators.BuildRangeValidator]},
    'testreport': {
        'from_date': [validators.DateRangeValidator],
        'interval': [validators.IntervalRangeValidator]}}


def ValidateParameters(request, vlist):
  """Returns a list of appropriate validators."""
  # Catches when no interval supplied.
  for range_key in vlist.keys() + [None]:
    if request.GET.get(range_key, None):
      break
  if not range_key:
    raise ChartInputError('One interval-type parameter must be supplied.')
  validators.Validate(request, vlist[range_key])
  if range_key == 'interval':
    salt = datetime.date.isoformat(datetime.date.today())
  else:
    salt = None
  return salt


def PlotChart(request):
  """Plot the requested chart from /chart?..."""
  try:
    salt = ValidateParameters(request, VLISTS['chart'])
    return chartviews.PlotChart(
        request, 'perfchart/plot_chart.html',
        perfchart_models.GetRangedKeyByBuildLinechartData, salt)
  except ChartInputError as e:
    tpl_hostname = request.get_host()
    return render_to_response('plot_syntax.html', locals())


def PlotChartDiff(request):
  """Plot the requested chart from /chartdiff?... and a split diff view."""
  try:
    salt = ValidateParameters(request, VLISTS['chart'])
    return chartviews.PlotChart(
        request, 'perfchart/plot_chartdiff.html',
        perfchart_models.GetRangedKeyByBuildLinechartData, salt)
  except ChartInputError as e:
    tpl_hostname = request.get_host()
    return render_to_response('plot_syntax.html', locals())


def PlotChartPlatforms(request):
  """Plot the requested report from /chartplatforms?..."""
  try:
    ValidateParameters(request, VLISTS['chartplatforms'])
    return reportviews.PlotReport(request, 'chartplatforms/plot_chartplatforms.html')
  except ChartInputError as e:
    tpl_hostname = request.get_host()
    return render_to_response('plot_syntax.html', locals())


def PlotChartReport(request):
  """Plot the requested report from /report?..."""
  try:
    ValidateParameters(request, VLISTS['chartreport'])
    return reportviews.PlotReport(request, 'chartreport/plot_chartreport.html')
  except ChartInputError as e:
    tpl_hostname = request.get_host()
    return render_to_response('plot_syntax.html', locals())


def PlotReleaseReport(request):
  """Plot the requested report from /releasereport?..."""
  try:
    salt = ValidateParameters(request, VLISTS['releasereport'])
    return chartviews.PlotChart(
        request, 'releasereport/plot_releasereport.html',
        release_models.GetReleaseReportData, salt)
  except ChartInputError as e:
    tpl_hostname = request.get_host()
    return render_to_response('plot_syntax.html', locals())


def PlotTestReport(request):
  """Plot the requested report from /testreport?..."""
  try:
    salt = ValidateParameters(request, VLISTS['testreport'])
    return chartviews.PlotChart(
        request, 'testreport/plot_testreport.html',
        testreport_models.GetRangedTestReportData, salt)
  except ChartInputError as e:
    tpl_hostname = request.get_host()
    return render_to_response('plot_syntax.html', locals())


def PlotLabTestReport(request):
  """Plot the requested report from /labtestreport?..."""
  try:
    salt = ValidateParameters(request, VLISTS['testreport'])
    return chartviews.PlotChart(
        request, 'labtest/plot_labtestreport.html',
        labtest_models.GetRangedLabTestReportData, salt)
  except ChartInputError as e:
    tpl_hostname = request.get_host()
    return render_to_response('plot_syntax.html', locals())