Add my_reviews.py script for perf time.

rietveld.py

 # Copyright (c) 2011 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 """Defines class Rietveld to easily access a rietveld instance.
 
 Security implications:
 
 The following hypothesis are made:
 - Rietveld enforces:
   - Nobody else than issue owner can upload a patch set
@@ skipping 224 matching lines @@
         ('message_only', 'True'),
         ('send_mail', 'True'),
         ('no_redirect', 'True')])
 
   def set_flag(self, issue, patchset, flag, value):
     return self.post('/%s/edit_flags' % issue, [
         ('last_patchset', str(patchset)),
         ('xsrf_token', self.xsrf_token()),
         (flag, value)])
 
+  def search(
+      self,
+      owner=None, reviewer=None,
+      base=None,
+      closed=None, private=None, commit=None,
+      created_before=None, created_after=None,
+      modified_before=None, modified_after=None,
+      per_request=None, keys_only=False,
+      with_messages=False):
+    """Yields search results."""
+    # These are expected to be strings.
+    string_keys = {
+        'owner': owner,
+        'reviewer': reviewer,
+        'base': base,
+        'created_before': created_before,
+        'created_after': created_after,
+        'modified_before': modified_before,
+        'modified_after': modified_after,
+    }
+    # These are either None, False or True.
+    three_state_keys = {
+      'closed': closed,
+      'private': private,
+      'commit': commit,
+    }
+
+    url = '/search?format=json'
+    # Sort the keys mainly to ease testing.
+    for key in sorted(string_keys):
+      value = string_keys[key]
+      if value:
+        url += '&%s=%s' % (key, urllib2.quote(value))
+    for key in sorted(three_state_keys):
+      value = three_state_keys[key]
+      if value is not None:
+        url += '&%s=%d' % (key, int(value) + 1)
+
+    if keys_only:
+      url += '&keys_only=True'
+    if with_messages:
+      url += '&with_messages=True'
+    if per_request:
+      url += '&limit=%d' % per_request
+
+    cursor = ''
+    while True:
+      output = self.get(url + cursor)
+      if output.startswith('<'):
+        # It's an error message. Return as no result.
+        break
+      data = json.loads(output) or {}
+      if not data.get('results'):
+        break
+      for i in data['results']:
+        yield i
+      cursor = '&cursor=%s' % data['cursor']
+
   def get(self, request_path, **kwargs):
     kwargs.setdefault('payload', None)
     return self._send(request_path, **kwargs)
 
   def post(self, request_path, data, **kwargs):
     ctype, body = upload.EncodeMultipartFormData(data, [])
     return self._send(request_path, payload=body, content_type=ctype, **kwargs)
 
   def _send(self, request_path, **kwargs):
     """Sends a POST/GET to Rietveld.  Returns the response body."""
@@ skipping 14 matching lines @@
         if retry >= (maxtries - 1):
           raise
         if not 'Name or service not known' in e.reason:
           # Usually internal GAE flakiness.
           raise
       # If reaching this line, loop again. Uses a small backoff.
       time.sleep(1+maxtries*2)
 
   # DEPRECATED.
   Send = get