Add --count support.

my_reviews.py

 #!/usr/bin/env python
 # 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.
 
 """Get rietveld stats.
 
 Example:
   - my_reviews.py -o me@chromium.org -Q  for stats for last quarter.
 """
@@ skipping 27 matching lines @@
     # By default, hide commit-bot and the domain.
     reviewers = set(r.split('@', 1)[0] for r in issue['reviewers'])
     reviewers -= set(('commit-bot',))
     # Strip time.
     timestamp = issue['created'][:10]
 
     # More information is available, print issue.keys() to see them.
     print '%d: %s %s' % (issue['issue'], timestamp, ', '.join(reviewers))
 
 
+def print_count(owner, reviewer, created_after, created_before, instance_url):
+  remote = rietveld.Rietveld(instance_url, None, None)
+  print len(list(remote.search(
+      owner=owner,
+      reviewer=reviewer,
+      created_after=created_after,
+      created_before=created_before,
+      keys_only=False)))
+
+
 def get_previous_quarter(today):
   """There are four quarters, 01-03, 04-06, 07-09, 10-12.
 
   If today is in the last month of a quarter, assume it's the current quarter
   that is requested.
   """
   year = today.year
   month = today.month - (today.month % 3)
   if not month:
     month = 12
     year -= 1
   previous_month = month - 2
   return (
       '%d-%02d-01' % (year, previous_month),
       '%d-%02d-01' % (year, month))
 
 
 def main():
+  # Silence upload.py.
+  rietveld.upload.verbosity = 0
   parser = optparse.OptionParser(description=sys.modules[__name__].__doc__)
+  parser.add_option(
+      '--count', action='store_true',
+      help='Just count instead of printing individual issues')
   parser.add_option('-o', '--owner')
   parser.add_option('-r', '--reviewer')
   parser.add_option('-c', '--created_after')
   parser.add_option('-C', '--created_before')
-  parser.add_option('-Q', '--last_quarter', action='store_true')
+  parser.add_option(
+      '-Q', '--last_quarter', action='store_true',
+      help='Use last quarter dates as filter')

[Dirk Pranke, 2011/09/22 01:13:15]

Nit: "last quarter's"

   parser.add_option('-i', '--instance_url', default='codereview.chromium.org')
   # Remove description formatting
   parser.format_description = lambda x: parser.description
   options, args = parser.parse_args()
   if args:
     parser.error('Args unsupported')
   if not options.owner and not options.reviewer:
     options.owner = os.environ['EMAIL_ADDRESS']
     if '@' not in options.owner:
       parser.error('Please specify at least -o or -r')
-    print 'Defaulting to owner=%s' % options.owner
+    print >> sys.stderr, 'Defaulting to owner=%s' % options.owner
   if options.last_quarter:
     today = datetime.date.today()
     options.created_after, options.created_before = get_previous_quarter(today)
-    print 'Using range %s to %s' % (
+    print >> sys.stderr, 'Using range %s to %s' % (
         options.created_after, options.created_before)
-  print_reviews(
-      options.owner, options.reviewer,
-      options.created_after, options.created_before,
-      options.instance_url)
+  if options.count:
+    print_count(
+        options.owner, options.reviewer,
+        options.created_after, options.created_before,
+        options.instance_url)
+  else:
+    print_reviews(
+        options.owner, options.reviewer,
+        options.created_after, options.created_before,
+        options.instance_url)
   return 0
 
 
 if __name__ == '__main__':
   sys.exit(main())