WebKitTools/Scripts/webkitpy/statusserver.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 # WebKit's Python module for interacting with the Commit Queue status page.
  30
  31 from webkitpy.webkit_logging import log
  32 from webkitpy.webkit_mechanize import Browser
  33
  34 # WebKit includes a built copy of BeautifulSoup in Scripts/webkitpy
  35 # so this import should always succeed.
  36 from .BeautifulSoup import BeautifulSoup
  37
  38 import urllib2
  39
  40
  41 class StatusServer:
  42     default_host = "webkit-commit-queue.appspot.com"
  43
  44     def __init__(self, host=default_host):
  45         self.set_host(host)
  46         self.browser = Browser()
  47
  48     def set_host(self, host):
  49         self.host = host
  50         self.url = "http://%s" % self.host
  51
  52     def results_url_for_status(self, status_id):
  53         return "%s/results/%s" % (self.url, status_id)
  54
  55     def update_status(self, queue_name, status, patch=None, results_file=None):
  56         # During unit testing, host is None
  57         if not self.host:
  58             return
  59
  60         log(status)
  61         update_status_url = "%s/update-status" % self.url
  62         self.browser.open(update_status_url)
  63         self.browser.select_form(name="update_status")
  64         self.browser['queue_name'] = queue_name
  65         if patch:
  66             if patch.get('bug_id'):
  67                 self.browser['bug_id'] = str(patch['bug_id'])
  68             if patch.get('id'):
  69                 self.browser['patch_id'] = str(patch['id'])
  70         self.browser['status'] = status
  71         if results_file:
  72             self.browser.add_file(results_file, "text/plain", "results.txt", 'results_file')
  73         response = self.browser.submit()
  74         return response.read() # This is the id of the newly created status object.
  75
  76     def patch_status(self, queue_name, patch_id):
  77         update_status_url = "%s/patch-status/%s/%s" % (self.url, queue_name, patch_id)
  78         try:
  79             return urllib2.urlopen(update_status_url).read()
  80         except urllib2.HTTPError, e:
  81             if e.code == 404:
  82                 return None
  83             raise e