2009-12-02 Adam Barth <abarth@webkit.org>
[WebKit-https.git] / WebKitTools / QueueStatusServer / queue_status.py
1 # Copyright (C) 2009 Google Inc. All rights reserved.
2 #
3 # Redistribution and use in source and binary forms, with or without
4 # modification, are permitted provided that the following conditions are
5 # met:
6
7 #     * Redistributions of source code must retain the above copyright
8 # notice, this list of conditions and the following disclaimer.
9 #     * Redistributions in binary form must reproduce the above
10 # copyright notice, this list of conditions and the following disclaimer
11 # in the documentation and/or other materials provided with the
12 # distribution.
13 #     * Neither the name of Google Inc. nor the names of its
14 # contributors may be used to endorse or promote products derived from
15 # this software without specific prior written permission.
16
17 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28
29 import cgi
30 import os
31
32 # Request a modern Django
33 from google.appengine.dist import use_library
34 use_library('django', '1.1')
35
36 from google.appengine.ext.webapp import template
37 from google.appengine.api import users
38 from google.appengine.ext import webapp
39 from google.appengine.ext.webapp.util import run_wsgi_app
40 from google.appengine.ext import db
41
42 webapp.template.register_template_library('filters.webkit_extras')
43
44
45 class QueueStatus(db.Model):
46     author = db.UserProperty()
47     queue_name = db.StringProperty()
48     active_bug_id = db.IntegerProperty()
49     active_patch_id = db.IntegerProperty()
50     message = db.StringProperty(multiline=True)
51     date = db.DateTimeProperty(auto_now_add=True)
52
53
54 class MainPage(webapp.RequestHandler):
55     def get(self):
56         statuses_query = QueueStatus.all().filter('queue_name =', 'commit-queue').order('-date')
57         statuses = statuses_query.fetch(6)
58         if not statuses:
59             return self.response.out.write("No status to report.")
60         template_values = {
61             'last_status' : statuses[0],
62             'recent_statuses' : statuses[1:],
63         }
64         self.response.out.write(template.render('index.html', template_values))
65
66
67 class PatchStatus(webapp.RequestHandler):
68     def get(self, queue_name, attachment_id):
69         statuses = QueueStatus.all().filter('queue_name =', queue_name).filter('active_patch_id =', int(attachment_id)).order('-date').fetch(1)
70         if not statuses:
71             self.error(404)
72             return
73         self.response.out.write(statuses[0].message)
74
75
76 class StatusSummary(object):
77     def _status_to_code(self, status):
78         code_names = {
79             "Pass": "pass",
80             "Pending": "pending",
81             "Fail": "fail",
82             "Error": "error",
83         }
84         return code_names.get(status, "none")
85
86     def _queue_name_to_code(self, queue_name):
87         code_names = {
88             "style-queue": "style",
89         }
90         return code_names[queue_name]
91
92     queues = [
93         "style-queue",
94     ]
95
96     def __init__(self):
97         self._summary = {}
98
99     def summarize(self, attachment_id):
100         if self._summary.get(attachment_id):
101             return self._summary.get(attachment_id)
102
103         attachment_summary = {}
104         for queue in self.queues:
105             statuses = QueueStatus.all().filter('queue_name =', queue).filter('active_patch_id =', attachment_id).order('-date').fetch(1)
106             status_code = self._status_to_code(statuses[0].message if statuses else None)
107             queue_code = self._queue_name_to_code(queue)
108             attachment_summary[queue_code] = status_code
109
110         self._summary[attachment_id] = attachment_summary
111         return attachment_summary
112
113
114 class StatusBubble(webapp.RequestHandler):
115     def get(self, attachment_id):
116         status_summary = StatusSummary()
117         template_values = {
118             "queue_status" : status_summary.summarize(int(attachment_id)),
119         }
120         self.response.out.write(template.render('status_bubble.html', template_values))
121
122
123 class UpdateStatus(webapp.RequestHandler):
124     def get(self):
125         self.response.out.write(template.render('update_status.html', None))
126
127     def _int_from_request(self, name):
128         string_value = self.request.get(name)
129         try:
130             int_value = int(string_value)
131             return int_value
132         except ValueError, TypeError:
133             pass
134         return None
135
136     def post(self):
137         queue_status = QueueStatus()
138
139         if users.get_current_user():
140             queue_status.author = users.get_current_user()
141
142         queue_name = self.request.get('queue_name')
143         queue_status.queue_name = queue_name
144         queue_status.active_bug_id = self._int_from_request('bug_id')
145         queue_status.active_patch_id = self._int_from_request('patch_id')
146         queue_status.message = self.request.get('status')
147         queue_status.put()
148         self.redirect('/')
149
150
151 routes = [
152     ('/', MainPage),
153     ('/update-status', UpdateStatus),
154     (r'/patch-status/(.*)/(.*)', PatchStatus),
155     (r'/status-bubble/(.*)', StatusBubble),
156 ]
157
158 application = webapp.WSGIApplication(routes, debug=True)
159
160 def main():
161     run_wsgi_app(application)
162
163 if __name__ == "__main__":
164     main()