diff -r 000000000000 -r 4f2f89ce4247 WebKitTools/QueueStatusServer/model/attachment.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/WebKitTools/QueueStatusServer/model/attachment.py Fri Sep 17 09:02:29 2010 +0300 @@ -0,0 +1,141 @@ +# Copyright (C) 2009 Google Inc. All rights reserved. +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions are +# met: +# +# * Redistributions of source code must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# * Redistributions in binary form must reproduce the above +# copyright notice, this list of conditions and the following disclaimer +# in the documentation and/or other materials provided with the +# distribution. +# * Neither the name of Google Inc. nor the names of its +# contributors may be used to endorse or promote products derived from +# this software without specific prior written permission. +# +# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT +# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, +# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +import re + +from google.appengine.api import memcache + +from model.queues import queues, name_with_underscores +from model.queuestatus import QueueStatus +from model.workitems import WorkItems + + +class Attachment(object): + @classmethod + def dirty(cls, attachment_id): + memcache.delete(str(attachment_id), namespace="attachment-summary") + + @classmethod + def recent(cls, limit=1): + statuses = QueueStatus.all().order("-date") + # Notice that we use both a set and a list here to keep the -date ordering. + ids = [] + visited_ids = set() + for status in statuses: + attachment_id = status.active_patch_id + if not attachment_id: + continue + if attachment_id in visited_ids: + continue + visited_ids.add(attachment_id) + ids.append(attachment_id) + if len(visited_ids) >= limit: + break + return map(cls, ids) + + def __init__(self, attachment_id): + self.id = attachment_id + self._summary = None + self._cached_queue_positions = None + + def summary(self): + if self._summary: + return self._summary + self._summary = memcache.get(str(self.id), namespace="attachment-summary") + if self._summary: + return self._summary + self._summary = self._fetch_summary() + memcache.set(str(self.id), self._summary, namespace="attachment-summary") + return self._summary + + def state_from_queue_status(self, status): + table = { + "Pass" : "pass", + "Fail" : "fail", + } + state = table.get(status.message) + if state: + return state + if status.message.startswith("Error:"): + return "error" + if status: + return "pending" + return None + + def position_in_queue(self, queue_name): + return self._queue_positions().get(queue_name) + + def status_for_queue(self, queue_name): + underscore_queue_name = name_with_underscores(queue_name) + # summary() is a horrible API and should be killed. + queue_summary = self.summary().get(underscore_queue_name) + if not queue_summary: + return None + return queue_summary.get("status") + + def bug_id(self): + return self.summary().get("bug_id") + + def _queue_positions(self): + if self._cached_queue_positions: + return self._cached_queue_positions + # FIXME: Should we be mem-caching this? + self._cached_queue_positions = self._calculate_queue_positions() + return self._cached_queue_positions + + def _calculate_queue_positions(self): + queue_positions = {} + for work_items in WorkItems.all().fetch(limit=len(queues)): + queue_name = str(work_items.queue_name) + try: + position = work_items.item_ids.index(self.id) + # Display 1-based indecies to the user. + queue_positions[queue_name] = position + 1 + except ValueError, e: + queue_positions[queue_name] = None + return queue_positions + + # FIXME: This is controller/view code and does not belong in a model. + def _fetch_summary(self): + summary = { "attachment_id" : self.id } + + first_status = QueueStatus.all().filter('active_patch_id =', self.id).get() + if not first_status: + # We don't have any record of this attachment. + return summary + summary["bug_id"] = first_status.active_bug_id + + for queue in queues: + summary[queue] = None + status = QueueStatus.all().filter('queue_name =', queue).filter('active_patch_id =', self.id).order('-date').get() + if status: + summary[name_with_underscores(queue)] = { + "state": self.state_from_queue_status(status), + "status": status, + } + return summary