+
−#!/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