Add support for specifying my_activity.py list format.

my_activity.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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 stats about your activity.
 
 Example:
   - my_activity.py  for stats for the current week (last week on mondays).
   - my_activity.py -Q  for stats for last quarter.
@@ skipping 452 matching lines @@
     issues = filter(self.filter_issue, issues)
     issues = sorted(issues, key=lambda i: i['modified'], reverse=True)
     return issues
 
   def process_google_code_issue(self, project, issue):
     ret = {}
     ret['created'] = datetime_from_google_code(issue['published']['$t'])
     ret['modified'] = datetime_from_google_code(issue['updated']['$t'])
 
     ret['owner'] = ''
-    if 'issues:owner' in issue:
-      ret['owner'] = issue['issues:owner'][0]['issues:username'][0]['$t']
+    if 'issues$owner' in issue:
+      ret['owner'] = issue['issues$owner']['issues$username']['$t']
     ret['author'] = issue['author'][0]['name']['$t']
 
     if 'shorturl' in project:
       issue_id = issue['id']['$t']
       issue_id = issue_id[issue_id.rfind('/') + 1:]
       ret['url'] = 'http://%s/%d' % (project['shorturl'], int(issue_id))
     else:
       issue_url = issue['link'][1]
       if issue_url['rel'] != 'alternate':
         raise RuntimeError
@@ skipping 216 matching lines @@
     except ImportError:
       print "Failed to import WebKit committer list, skipping WebKit checks."
       self.webkit_repo = None
       return
 
     if not webkit_account(self.user):
       email = self.user + "@chromium.org"
       print "No %s in committers.py, skipping WebKit checks." % email
       self.webkit_repo = None
 
-  @staticmethod
-  def print_change(change):
-    print '%s %s' % (
-        change['review_url'],
-        change['header'],
-        )
+  def print_change(self, change):
+    self.print_generic(self.options.output_format,
+                       self.options.output_format_changes,
+                       change['header'],
+                       change['review_url'],
+                       change['author'])
+
+  def print_issue(self, issue):
+    optional_values = {
+        'owner': issue['owner'],
+    }
+    self.print_generic(self.options.output_format,
+                       self.options.output_format_issues,
+                       issue['header'],
+                       issue['url'],
+                       issue['author'],
+                       optional_values)
+
+  def print_review(self, review):
+    self.print_generic(self.options.output_format,
+                       self.options.output_format_reviews,
+                       review['header'],
+                       review['review_url'],
+                       review['author'])
 
   @staticmethod
-  def print_issue(issue):
-    print '%s %s' % (
-        issue['url'],
-        issue['header'],
-        )
+  def print_generic(default_fmt, specific_fmt,
+                    title, url, author,
+                    optional_values=None):
+    output_format = specific_fmt if specific_fmt is not None else default_fmt
+    output_format = unicode(output_format)
+    required_values = {
+        'title': title,
+        'url': url,
+        'author': author,
+    }
+    # Merge required and optional values.
+    if optional_values is not None:
+      values = dict(required_values.items() + optional_values.items())
+    else:
+      values = required_values
+    print output_format.format(**values)
+
 
   def filter_issue(self, issue, should_filter_by_user=True):
     def maybe_filter_username(email):
       return not should_filter_by_user or username(email) == self.user
     if (maybe_filter_username(issue['author']) and
         self.filter_modified(issue['created'])):
       return True
     if (maybe_filter_username(issue['owner']) and
         (self.filter_modified(issue['created']) or
          self.filter_modified(issue['modified']))):
@@ skipping 50 matching lines @@
       reviews = filter(lambda r: not username(r['owner']) == self.user, reviews)
       self.reviews += reviews
 
     for instance in git_instances:
       self.reviews += self.git_search(instance, reviewer=self.user)
 
   def print_reviews(self):
     if self.reviews:
       print '\nReviews:'
       for review in self.reviews:
-        self.print_change(review)
+        self.print_review(review)
 
   def get_issues(self):
     for project in google_code_projects:
       self.issues += self.google_code_issue_search(project)
 
     for instance in bugzilla_instances:
       self.issues += self.bugzilla_issues(instance, self.user)
 
+  def print_issues(self):
+    if self.issues:
+      print '\nIssues:'
+      for issue in self.issues:
+        self.print_issue(issue)
+
   def process_activities(self):
     # If a webkit bug was a review, don't list it as an issue.
     ids = {}
     for review in self.reviews + self.changes:
       if 'webkit_bug_id' in review:
         ids[review['webkit_bug_id']] = True
 
     def duplicate_issue(issue):
       if 'webkit_bug_id' not in issue:
         return False
       return issue['webkit_bug_id'] in ids
 
     self.issues = filter(lambda issue: not duplicate_issue(issue), self.issues)
 
-  def print_issues(self):
-    if self.issues:
-      print '\nIssues:'
-      for c in self.issues:
-        self.print_issue(c)
-
   def print_activity(self):
     self.print_changes()
     self.print_reviews()
     self.print_issues()
 
 
 def main():
   # Silence upload.py.
   rietveld.upload.verbosity = 0
 
@@ skipping 22 matching lines @@
       '-Y', '--this_year', action='store_true',
       help='Use this year\'s dates')
   parser.add_option(
       '-w', '--week_of', metavar='<date>',
       help='Show issues for week of the date')
   parser.add_option(
       '-a', '--auth',
       action='store_true',
       help='Ask to authenticate for instances with no auth cookie')
 
-  group = optparse.OptionGroup(parser, 'Activity Types',
+  activity_types_group = optparse.OptionGroup(parser, 'Activity Types',
                                'By default, all activity will be looked up and '
                                'printed. If any of these are specified, only '
                                'those specified will be searched.')
-  group.add_option(
+  activity_types_group.add_option(
       '-c', '--changes',
       action='store_true',
       help='Show changes.')
-  group.add_option(
+  activity_types_group.add_option(
       '-i', '--issues',
       action='store_true',
       help='Show issues.')
-  group.add_option(
+  activity_types_group.add_option(
       '-r', '--reviews',
       action='store_true',
       help='Show reviews.')
-  parser.add_option_group(group)
+  parser.add_option_group(activity_types_group)
+
+  output_format_group = optparse.OptionGroup(parser, 'Output Format',
+                              'By default, all activity will be printed in the '
+                              'following format: {url} {title}. This can be '
+                              'changed for either all activity types or '
+                              'individually for each activity type. The format '
+                              'is defined as documented for '
+                              'string.format(...). The variables available for '
+                              'all activity types are url, title and author. '
+                              'Format options for specific activity types will '
+                              'override the generic format.')
+  output_format_group.add_option(
+      '-f', '--output-format', metavar='<format>',
+      default=u'{url} {title}',
+      help='Specifies the format to use when printing all your activity.')
+  output_format_group.add_option(
+      '--output-format-changes', metavar='<format>',
+      default=None,
+      help='Specifies the format to use when printing changes.')
+  output_format_group.add_option(
+      '--output-format-issues', metavar='<format>',
+      default=None,
+      help='Specifies the format to use when printing issues. Has support '
+           'for the additional variable owner.')
+  output_format_group.add_option(
+      '--output-format-reviews', metavar='<format>',
+      default=None,
+      help='Specifies the format to use when printing reviews.')
+  parser.add_option_group(output_format_group)
 
   # Remove description formatting
   parser.format_description = (
       lambda _: parser.description)  # pylint: disable=E1101
 
   options, args = parser.parse_args()
   options.local_user = os.environ.get('USER')
   if args:
     parser.error('Args unsupported')
   if not options.user:
@@ skipping 51 matching lines @@
   print '\n\n\n'
 
   my_activity.print_changes()
   my_activity.print_reviews()
   my_activity.print_issues()
   return 0
 
 
 if __name__ == '__main__':
   sys.exit(main())