#!/usr/bin/python2.5

#

# Copyright 2009 the Melange authors.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

#   http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

"""Views for Cron.

"""

__authors__ = [

    '"Sverre Rabbelier" <sverre@rabbelier.nl>',

  ]

import random

from django import http

from soc.logic import dicts

from soc.logic.models.priority_group import logic as priority_group_logic

from soc.logic.models.job import logic as job_logic

from soc.views.helper import access

from soc.views.models import base

import soc.cron.job

class View(base.View):

  """View methods for the Cron model.

  """

  def __init__(self, params=None):

    """Defines the fields and methods required for the base View class

    to provide the user with list, public, create, edit and delete views.

    Params:

      params: a dict with params for this View

    """

    rights = access.Checker(params)

    new_params = {}

    new_params['rights'] = rights

    new_params['logic'] = priority_group_logic

    new_params['name'] = "Cron"

    new_params['django_patterns_defaults'] = [

        (r'^%(url_name)s/(?P<access_type>poke)$',

          'soc.views.models.%(module_name)s.poke', 'Poke %(name_short)s'),

        ]

    params = dicts.merge(params, new_params)

    super(View, self).__init__(params=params)

  def poke(self, request, access_type, page_name):

    """View called by the cron system that handles jobs.

    Args:

      request: the standard Django HTTP request object

      access_type : the name of the access type which should be checked

      page_name: the page name displayed in templates as page and header title

    """

    order = ['-priority']

    query = priority_group_logic.getQueryForFields(order=order)

    groups = priority_group_logic.getAll(query)

    handler = soc.cron.job.handler

    groups_touched = 0

    jobs_completed = 0

    for group in groups:

      groups_touched += 1

      filter = {

          'priority_group': group,

          'status': 'waiting',

          }

      queryGen = lambda: job_logic.getQueryForFields(filter=filter)

      jobs = job_logic.entityIterator(queryGen, batch_size=10)

      retry_jobs = []

      for job in handler.iterate(jobs, retry_jobs):

        if random.randint(0, 3) > 0:

          retry_jobs.append(job)

          continue

        job_key = job.key().id()

        status = handler.handle(job_key)

        if status is handler.OUT_OF_TIME:

          break

        if status is handler.ERRORED:

          retry_jobs.append(job)

        jobs_completed += 1

    response = 'Completed %d jobs in %d priority groups.' % (

        jobs_completed, groups_touched)

    return http.HttpResponse(response)

view = View()

poke = view.poke