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         for queue in self.queues:
104             statuses = QueueStatus.all().filter('queue_name =', queue).filter('active_patch_id =', attachment_id).order('-date').fetch(1)
105             status_code = self._status_to_code(statuses[0].message if statuses else None)
106             queue_code = self._queue_name_to_code(queue)
107             attachment_summary[queue_code] = status_code
108
109         self._summary[attachment_id] = attachment_summary
110         return attachment_summary
111
112
113 class StatusBubble(webapp.RequestHandler):
114     def get(self, attachment_id):
115         status_summary = StatusSummary()
116         template_values = {
117             "queue_status" : status_summary.summarize(int(attachment_id)),
118         }
119         self.response.out.write(template.render('status_bubble.html', template_values))
120
121
122 class UpdateStatus(webapp.RequestHandler):
123     def get(self):
124         self.response.out.write(template.render('update_status.html', None))
125
126     def _int_from_request(self, name):
127         string_value = self.request.get(name)
128         try:
129             int_value = int(string_value)
130             return int_value
131         except ValueError, TypeError:
132             pass
133         return None
134
135     def post(self):
136         queue_status = QueueStatus()
137
138         if users.get_current_user():
139             queue_status.author = users.get_current_user()
140
141         queue_name = self.request.get('queue_name')
142         queue_status.queue_name = queue_name
143         queue_status.active_bug_id = self._int_from_request('bug_id')
144         queue_status.active_patch_id = self._int_from_request('patch_id')
145         queue_status.message = self.request.get('status')
146         queue_status.put()
147         self.redirect('/')
148
149
150 routes = [
151     ('/', MainPage),
152     ('/update-status', UpdateStatus),
153     (r'/patch-status/(.*)/(.*)', PatchStatus),
154     (r'/status-bubble/(.*)', StatusBubble),
155 ]
156
157 application = webapp.WSGIApplication(routes, debug=True)
158
159 def main():
160     run_wsgi_app(application)
161
162 if __name__ == "__main__":
163     main()