Revert of Gerrit git cl: implement git cl patch.

git_cl.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.
 
 # Copyright (C) 2008 Evan Martin <martine@danga.com>
 
 """A git-command for integrating reviews on Rietveld and Gerrit."""
 
 from distutils.version import LooseVersion
@@ skipping 816 matching lines @@
   """Returns current branch or None.
 
   For refs/heads/* branches, returns just last part. For others, full ref.
   """
   branchref = GetCurrentBranchRef()
   if branchref:
     return ShortBranchName(branchref)
   return None
 
 
-class _ParsedIssueNumberArgument(object):
-  def __init__(self, issue=None, patchset=None, hostname=None):
-    self.issue = issue
-    self.patchset = patchset
-    self.hostname = hostname
-
-  @property
-  def valid(self):
-    return self.issue is not None
-
-
-class _RietveldParsedIssueNumberArgument(_ParsedIssueNumberArgument):
-  def __init__(self, *args, **kwargs):
-    self.patch_url = kwargs.pop('patch_url', None)
-    super(_RietveldParsedIssueNumberArgument, self).__init__(*args, **kwargs)
-
-
-def ParseIssueNumberArgument(arg):
-  """Parses the issue argument and returns _ParsedIssueNumberArgument."""
-  fail_result = _ParsedIssueNumberArgument()
-
-  if arg.isdigit():
-    return _ParsedIssueNumberArgument(issue=int(arg))
-  if not arg.startswith('http'):
-    return fail_result
-  url = gclient_utils.UpgradeToHttps(arg)
-  try:
-    parsed_url = urlparse.urlparse(url)
-  except ValueError:
-    return fail_result
-  for cls in _CODEREVIEW_IMPLEMENTATIONS.itervalues():
-    tmp = cls.ParseIssueURL(parsed_url)
-    if tmp is not None:
-      return tmp
-  return fail_result
-
-
 class Changelist(object):
   """Changelist works with one changelist in local branch.
 
   Supports two codereview backends: Rietveld or Gerrit, selected at object
   creation.
 
   Not safe for concurrent multi-{thread,process} use.
   """
 
   def __init__(self, branchref=None, issue=None, codereview=None, **kwargs):
@@ skipping 403 matching lines @@
     try:
       return presubmit_support.DoPresubmitChecks(change, committing,
           verbose=verbose, output_stream=sys.stdout, input_stream=sys.stdin,
           default_presubmit=None, may_prompt=may_prompt,
           rietveld_obj=self._codereview_impl.GetRieveldObjForPresubmit())
     except presubmit_support.PresubmitFailure, e:
       DieWithError(
           ('%s\nMaybe your depot_tools is out of date?\n'
            'If all fails, contact maruel@') % e)
 
-  def CMDPatchIssue(self, issue_arg, reject, nocommit, directory):
-    """Fetches and applies the issue patch from codereview to local branch."""
-    if issue_arg.isdigit():
-      parsed_issue_arg = _RietveldParsedIssueNumberArgument(int(issue_arg))
-    else:
-      # Assume url.
-      parsed_issue_arg = self._codereview_impl.ParseIssueURL(
-          urlparse.urlparse(issue_arg))
-    if not parsed_issue_arg or not parsed_issue_arg.valid:
-      DieWithError('Failed to parse issue argument "%s". '
-                   'Must be an issue number or a valid URL.' % issue_arg)
-    return self._codereview_impl.CMDPatchWithParsedIssue(
-        parsed_issue_arg, reject, nocommit, directory)
-
   # Forward methods to codereview specific implementation.
 
   def CloseIssue(self):
     return self._codereview_impl.CloseIssue()
 
   def GetStatus(self):
     return self._codereview_impl.GetStatus()
 
   def GetCodereviewServer(self):
     return self._codereview_impl.GetCodereviewServer()
@@ skipping 69 matching lines @@
     """Returns a list of reviewers approving the change.
 
     Note: not necessarily committers.
     """
     raise NotImplementedError()
 
   def GetMostRecentPatchset(self):
     """Returns the most recent patchset number from the codereview site."""
     raise NotImplementedError()
 
-  def CMDPatchWithParsedIssue(self, parsed_issue_arg, reject, nocommit,
-                              directory):
-    """Fetches and applies the issue.
 
-    Arguments:
-      parsed_issue_arg: instance of _ParsedIssueNumberArgument.
-      reject: if True, reject the failed patch instead of switching to 3-way
-        merge. Rietveld only.
-      nocommit: do not commit the patch, thus leave the tree dirty. Rietveld
-        only.
-      directory: switch to directory before applying the patch. Rietveld only.
-    """
-    raise NotImplementedError()
-
-  @staticmethod
-  def ParseIssueURL(parsed_url):
-    """Parses url and returns instance of _ParsedIssueNumberArgument or None if
-    failed."""
-    raise NotImplementedError()
-
-
 class _RietveldChangelistImpl(_ChangelistCodereviewBase):
   def __init__(self, changelist, auth_config=None, rietveld_server=None):
     super(_RietveldChangelistImpl, self).__init__(changelist)
     assert settings, 'must be initialized in _ChangelistCodereviewBase'
     settings.GetDefaultServerUrl()
 
     self._rietveld_server = rietveld_server
     self._auth_config = auth_config
     self._props = None
     self._rpc_server = None
@@ skipping 151 matching lines @@
   def GetCodereviewServerSetting(self):
     """Returns the git setting that stores this change's rietveld server."""
     branch = self.GetBranch()
     if branch:
       return 'branch.%s.rietveldserver' % branch
     return None
 
   def GetRieveldObjForPresubmit(self):
     return self.RpcServer()
 
-  def CMDPatchWithParsedIssue(self, parsed_issue_arg, reject, nocommit,
-                              directory):
-    # TODO(maruel): Use apply_issue.py
-
-    # PatchIssue should never be called with a dirty tree.  It is up to the
-    # caller to check this, but just in case we assert here since the
-    # consequences of the caller not checking this could be dire.
-    assert(not git_common.is_dirty_git_tree('apply'))
-    assert(parsed_issue_arg.valid)
-    self._changelist.issue = parsed_issue_arg.issue
-    if parsed_issue_arg.hostname:
-      self._rietveld_server = 'https://%s' % parsed_issue_arg.hostname
-
-    if parsed_issue_arg.patch_url:
-      assert parsed_issue_arg.patchset
-      patchset = parsed_issue_arg.patchset
-      patch_data = urllib2.urlopen(parsed_issue_arg.patch_url).read()
-    else:
-      patchset = parsed_issue_arg.patchset or self.GetMostRecentPatchset()
-      patch_data = self.GetPatchSetDiff(self.GetIssue(), patchset)
-
-    # Switch up to the top-level directory, if necessary, in preparation for
-    # applying the patch.
-    top = settings.GetRelativeRoot()
-    if top:
-      os.chdir(top)
-
-    # Git patches have a/ at the beginning of source paths.  We strip that out
-    # with a sed script rather than the -p flag to patch so we can feed either
-    # Git or svn-style patches into the same apply command.
-    # re.sub() should be used but flags=re.MULTILINE is only in python 2.7.
-    try:
-      patch_data = subprocess2.check_output(
-          ['sed', '-e', 's|^--- a/|--- |; s|^+++ b/|+++ |'], stdin=patch_data)
-    except subprocess2.CalledProcessError:
-      DieWithError('Git patch mungling failed.')
-    logging.info(patch_data)
-
-    # We use "git apply" to apply the patch instead of "patch" so that we can
-    # pick up file adds.
-    # The --index flag means: also insert into the index (so we catch adds).
-    cmd = ['git', 'apply', '--index', '-p0']
-    if directory:
-      cmd.extend(('--directory', directory))
-    if reject:
-      cmd.append('--reject')
-    elif IsGitVersionAtLeast('1.7.12'):
-      cmd.append('--3way')
-    try:
-      subprocess2.check_call(cmd, env=GetNoGitPagerEnv(),
-                             stdin=patch_data, stdout=subprocess2.VOID)
-    except subprocess2.CalledProcessError:
-      print 'Failed to apply the patch'
-      return 1
-
-    # If we had an issue, commit the current state and register the issue.
-    if not nocommit:
-      RunGit(['commit', '-m', (self.GetDescription() + '\n\n' +
-                               'patch from issue %(i)s at patchset '
-                               '%(p)s (http://crrev.com/%(i)s#ps%(p)s)'
-                               % {'i': self.GetIssue(), 'p': patchset})])
-      self.SetIssue(self.GetIssue())
-      self.SetPatchset(patchset)
-      print "Committed patch locally."
-    else:
-      print "Patch applied to index."
-    return 0
-
-  @staticmethod
-  def ParseIssueURL(parsed_url):
-    if not parsed_url.scheme or not parsed_url.scheme.startswith('http'):
-      return None
-    # Typical url: https://domain/<issue_number>[/[other]]
-    match = re.match('/(\d+)(/.*)?$', parsed_url.path)
-    if match:
-      return _RietveldParsedIssueNumberArgument(
-          issue=int(match.group(1)),
-          hostname=parsed_url.netloc)
-    # Rietveld patch: https://domain/download/issue<number>_<patchset>.diff
-    match = re.match(r'/download/issue(\d+)_(\d+).diff$', parsed_url.path)
-    if match:
-      return _RietveldParsedIssueNumberArgument(
-          issue=int(match.group(1)),
-          patchset=int(match.group(2)),
-          hostname=parsed_url.netloc,
-          patch_url=gclient_utils.UpgradeToHttps(parsed_url.geturl()))
-    return None
-
 
 class _GerritChangelistImpl(_ChangelistCodereviewBase):
   def __init__(self, changelist, auth_config=None):
     # auth_config is Rietveld thing, kept here to preserve interface only.
     super(_GerritChangelistImpl, self).__init__(changelist)
     self._change_id = None
     self._gerrit_server = None  # e.g. https://chromium-review.googlesource.com
     self._gerrit_host = None   # e.g. chromium-review.googlesource.com
 
   def _GetGerritHost(self):
@@ skipping 158 matching lines @@
           may_prompt=not force,
           verbose=verbose,
           change=self.GetChange(self.GetCommonAncestorWithUpstream(), None))
       if not hook_results.should_continue():
         return 1
 
     self.SubmitIssue(wait_for_merge=True)
     print('Issue %s has been submitted.' % self.GetIssueURL())
     return 0
 
-  def CMDPatchWithParsedIssue(self, parsed_issue_arg, reject, nocommit,
-                              directory):
-    assert not reject
-    assert not nocommit
-    assert not directory
-    assert parsed_issue_arg.valid
-
-    self._changelist.issue = parsed_issue_arg.issue
-
-    if parsed_issue_arg.hostname:
-      self._gerrit_host = parsed_issue_arg.hostname
-      self._gerrit_server = 'https://%s' % self._gerrit_host
-
-    detail = self._GetChangeDetail(['ALL_REVISIONS'])
-
-    if not parsed_issue_arg.patchset:
-      # Use current revision by default.
-      revision_info = detail['revisions'][detail['current_revision']]
-      patchset = int(revision_info['_number'])
-    else:
-      patchset = parsed_issue_arg.patchset
-      for revision_info in detail['revisions'].itervalues():
-        if int(revision_info['_number']) == parsed_issue_arg.patchset:
-          break
-      else:
-        DieWithError('Couldn\'t find patchset %i in issue %i' %
-                     (parsed_issue_arg.patchset, self.GetIssue()))
-
-    fetch_info = revision_info['fetch']['http']
-    RunGit(['fetch', fetch_info['url'], fetch_info['ref']])
-    RunGit(['cherry-pick', 'FETCH_HEAD'])
-    self.SetIssue(self.GetIssue())
-    self.SetPatchset(patchset)
-    print('Committed patch for issue %i pathset %i locally' %
-          (self.GetIssue(), self.GetPatchset()))
-    return 0
-
-  @staticmethod
-  def ParseIssueURL(parsed_url):
-    if not parsed_url.scheme or not parsed_url.scheme.startswith('http'):
-      return None
-    # Gerrit's new UI is https://domain/c/<issue_number>[/[patchset]]
-    # But current GWT UI is https://domain/#/c/<issue_number>[/[patchset]]
-    # Short urls like https://domain/<issue_number> can be used, but don't allow
-    # specifying the patchset (you'd 404), but we allow that here.
-    if parsed_url.path == '/':
-      part = parsed_url.fragment
-    else:
-      part = parsed_url.path
-    match = re.match('(/c)?/(\d+)(/(\d+)?/?)?$', part)
-    if match:
-      return _ParsedIssueNumberArgument(
-          issue=int(match.group(2)),
-          patchset=int(match.group(4)) if match.group(4) else None,
-          hostname=parsed_url.netloc)
-    return None
-
 
 _CODEREVIEW_IMPLEMENTATIONS = {
   'rietveld': _RietveldChangelistImpl,
   'gerrit': _GerritChangelistImpl,
 }
 
 
 class ChangeDescription(object):
   """Contains a parsed form of the change description."""
   R_LINE = r'^[ \t]*(TBR|R)[ \t]*=[ \t]*(.*?)[ \t]*$'
@@ skipping 1877 matching lines @@
 def CMDland(parser, args):
   """Commits the current changelist via git."""
   if settings.GetIsGitSvn() or git_footers.get_footer_svn_id():
     print('This appears to be an SVN repository.')
     print('Are you sure you didn\'t mean \'git cl dcommit\'?')
     print('(Ignore if this is the first commit after migrating from svn->git)')
     ask_for_data('[Press enter to push or ctrl-C to quit]')
   return SendUpstream(parser, args, 'land')
 
 
+def ParseIssueNum(arg):
+  """Parses the issue number from args if present otherwise returns None."""
+  if re.match(r'\d+', arg):
+    return arg
+  if arg.startswith('http'):
+    return re.sub(r'.*/(\d+)/?', r'\1', arg)
+  return None
+
+
 @subcommand.usage('<patch url or issue id or issue url>')
 def CMDpatch(parser, args):
   """Patches in a code review."""
   parser.add_option('-b', dest='newbranch',
                     help='create a new branch off trunk for the patch')
   parser.add_option('-f', '--force', action='store_true',
                     help='with -b, clobber any existing branch')
   parser.add_option('-d', '--directory', action='store', metavar='DIR',
                     help='Change to the directory DIR immediately, '
-                         'before doing anything else. Rietveld only.')
+                         'before doing anything else.')
   parser.add_option('--reject', action='store_true',
                     help='failed patches spew .rej files rather than '
-                        'attempting a 3-way merge. Rietveld only.')
+                        'attempting a 3-way merge')
   parser.add_option('-n', '--no-commit', action='store_true', dest='nocommit',
-                    help='don\'t commit after patch applies. Rietveld only.')
+                    help="don't commit after patch applies")
 
-
-  group = optparse.OptionGroup(
-      parser,
-      'Options for continuing work on the current issue uploaded from a '
-      'different clone (e.g. different machine). Must be used independently '
-      'from the other options. No issue number should be specified, and the '
-      'branch must have an issue number associated with it')
-  group.add_option('--reapply', action='store_true', dest='reapply',
-                   help='Reset the branch and reapply the issue.\n'
-                        'CAUTION: This will undo any local changes in this '
-                        'branch')
+  group = optparse.OptionGroup(parser,
+                    """Options for continuing work on the current issue uploaded
+from a different clone (e.g. different machine). Must be used independently from
+the other options. No issue number should be specified, and the branch must have
+an issue number associated with it""")
+  group.add_option('--reapply', action='store_true',
+                    dest='reapply',
+                    help="""Reset the branch and reapply the issue.
+CAUTION: This will undo any local changes in this branch""")
 
   group.add_option('--pull', action='store_true', dest='pull',
-                    help='Performs a pull before reapplying.')
+                    help="Performs a pull before reapplying.")
   parser.add_option_group(group)
 
   auth.add_auth_options(parser)
   (options, args) = parser.parse_args(args)
   auth_config = auth.extract_auth_config_from_options(options)
 
-  cl = Changelist(auth_config=auth_config)
-
   issue_arg = None
   if options.reapply :
     if len(args) > 0:
-      parser.error('--reapply implies no additional arguments.')
+      parser.error("--reapply implies no additional arguments.")
 
+    cl = Changelist()
     issue_arg = cl.GetIssue()
     upstream = cl.GetUpstreamBranch()
     if upstream == None:
-      parser.error('No upstream branch specified. Cannot reset branch')
+      parser.error("No upstream branch specified. Cannot reset branch")
 
     RunGit(['reset', '--hard', upstream])
     if options.pull:
       RunGit(['pull'])
   else:
     if len(args) != 1:
-      parser.error('Must specify issue number or url')
-    issue_arg = args[0]
+      parser.error("Must specify issue number")
 
-  if not issue_arg:
+    issue_arg = ParseIssueNum(args[0])
+
+  # The patch URL works because ParseIssueNum won't do any substitution
+  # as the re.sub pattern fails to match and just returns it.
+  if issue_arg == None:
     parser.print_help()
     return 1
 
-  if cl.IsGerrit():
-    if options.reject:
-      parser.error('--reject is not supported with Gerrit codereview.')
-    if options.nocommit:
-      parser.error('--nocommit is not supported with Gerrit codereview.')
-    if options.directory:
-      parser.error('--directory is not supported with Gerrit codereview.')
-
   # We don't want uncommitted changes mixed up with the patch.
   if git_common.is_dirty_git_tree('patch'):
     return 1
 
+  # TODO(maruel): Use apply_issue.py
+  # TODO(ukai): use gerrit-cherry-pick for gerrit repository?
+
   if options.newbranch:
     if options.reapply:
       parser.error("--reapply excludes any option other than --pull")
     if options.force:
       RunGit(['branch', '-D', options.newbranch],
           stderr=subprocess2.PIPE, error_ok=True)
     RunGit(['checkout', '-b', options.newbranch,
             Changelist().GetUpstreamBranch()])
 
-  return cl.CMDPatchIssue(issue_arg, options.reject, options.nocommit,
-                          options.directory)
+  return PatchIssue(issue_arg, options.reject, options.nocommit,
+                    options.directory, auth_config)
+
+
+def PatchIssue(issue_arg, reject, nocommit, directory, auth_config):
+  # PatchIssue should never be called with a dirty tree.  It is up to the
+  # caller to check this, but just in case we assert here since the
+  # consequences of the caller not checking this could be dire.
+  assert(not git_common.is_dirty_git_tree('apply'))
+
+  # TODO(tandrii): implement for Gerrit.
+  if type(issue_arg) is int or issue_arg.isdigit():
+    # Input is an issue id.  Figure out the URL.
+    issue = int(issue_arg)
+    cl = Changelist(issue=issue, codereview='rietveld', auth_config=auth_config)
+    patchset = cl.GetMostRecentPatchset()
+    patch_data = cl._codereview_impl.GetPatchSetDiff(issue, patchset)
+  else:
+    # Assume it's a URL to the patch. Default to https.
+    issue_url = gclient_utils.UpgradeToHttps(issue_arg)
+    match = re.match(r'(.*?)/download/issue(\d+)_(\d+).diff', issue_url)
+    if not match:
+      DieWithError('Must pass an issue ID or full URL for '
+          '\'Download raw patch set\'')
+    issue = int(match.group(2))
+    cl = Changelist(issue=issue, codereview='rietveld',
+                    rietveld_server=match.group(1), auth_config=auth_config)
+    patchset = int(match.group(3))
+    patch_data = urllib2.urlopen(issue_arg).read()
+
+  # Switch up to the top-level directory, if necessary, in preparation for
+  # applying the patch.
+  top = settings.GetRelativeRoot()
+  if top:
+    os.chdir(top)
+
+  # Git patches have a/ at the beginning of source paths.  We strip that out
+  # with a sed script rather than the -p flag to patch so we can feed either
+  # Git or svn-style patches into the same apply command.
+  # re.sub() should be used but flags=re.MULTILINE is only in python 2.7.
+  try:
+    patch_data = subprocess2.check_output(
+        ['sed', '-e', 's|^--- a/|--- |; s|^+++ b/|+++ |'], stdin=patch_data)
+  except subprocess2.CalledProcessError:
+    DieWithError('Git patch mungling failed.')
+  logging.info(patch_data)
+
+  # We use "git apply" to apply the patch instead of "patch" so that we can
+  # pick up file adds.
+  # The --index flag means: also insert into the index (so we catch adds).
+  cmd = ['git', 'apply', '--index', '-p0']
+  if directory:
+    cmd.extend(('--directory', directory))
+  if reject:
+    cmd.append('--reject')
+  elif IsGitVersionAtLeast('1.7.12'):
+    cmd.append('--3way')
+  try:
+    subprocess2.check_call(cmd, env=GetNoGitPagerEnv(),
+                           stdin=patch_data, stdout=subprocess2.VOID)
+  except subprocess2.CalledProcessError:
+    print 'Failed to apply the patch'
+    return 1
+
+  # If we had an issue, commit the current state and register the issue.
+  if not nocommit:
+    RunGit(['commit', '-m', (cl.GetDescription() + '\n\n' +
+                             'patch from issue %(i)s at patchset '
+                             '%(p)s (http://crrev.com/%(i)s#ps%(p)s)'
+                             % {'i': issue, 'p': patchset})])
+    cl = Changelist(codereview='rietveld', auth_config=auth_config,
+                    rietveld_server=cl.GetCodereviewServer())
+    cl.SetIssue(issue)
+    cl.SetPatchset(patchset)
+    print "Committed patch locally."
+  else:
+    print "Patch applied to index."
+  return 0
 
 
 def CMDrebase(parser, args):
   """Rebases current branch on top of svn repo."""
   # Provide a wrapper for git svn rebase to help avoid accidental
   # git svn dcommit.
   # It's the only command that doesn't use parser at all since we just defer
   # execution to git-svn.
 
   return RunGitWithCode(['svn', 'rebase'] + args)[1]
@@ skipping 381 matching lines @@
 
 def CMDdiff(parser, args):
   """Shows differences between local tree and last upload."""
   auth.add_auth_options(parser)
   options, args = parser.parse_args(args)
   auth_config = auth.extract_auth_config_from_options(options)
   if args:
     parser.error('Unrecognized args: %s' % ' '.join(args))
 
   # Uncommitted (staged and unstaged) changes will be destroyed by
-  # "git reset --hard" if there are merging conflicts in CMDPatchIssue().
+  # "git reset --hard" if there are merging conflicts in PatchIssue().
   # Staged changes would be committed along with the patch from last
   # upload, hence counted toward the "last upload" side in the final
   # diff output, and this is not what we want.
   if git_common.is_dirty_git_tree('diff'):
     return 1
 
   cl = Changelist(auth_config=auth_config)
   issue = cl.GetIssue()
   branch = cl.GetBranch()
   if not issue:
     DieWithError('No issue found for current branch (%s)' % branch)
   TMP_BRANCH = 'git-cl-diff'
   base_branch = cl.GetCommonAncestorWithUpstream()
 
   # Create a new branch based on the merge-base
   RunGit(['checkout', '-q', '-b', TMP_BRANCH, base_branch])
   try:
-    rtn = cl.CMDPatchIssue(issue, reject=False, nocommit=False, directory=None)
+    # Patch in the latest changes from rietveld.
+    rtn = PatchIssue(issue, False, False, None, auth_config)
     if rtn != 0:
       RunGit(['reset', '--hard'])
       return rtn
 
     # Switch back to starting branch and diff against the temporary
     # branch containing the latest rietveld patch.
     subprocess2.check_call(['git', 'diff', TMP_BRANCH, branch, '--'])
   finally:
     RunGit(['checkout', '-q', branch])
     RunGit(['branch', '-D', TMP_BRANCH])
@@ skipping 187 matching lines @@
       stdout = RunCommand(cmd + [gn_diff_file], cwd=top_dir)
       if opts.diff:
         sys.stdout.write(stdout)
 
   return return_value
 
 
 @subcommand.usage('<codereview url or issue id>')
 def CMDcheckout(parser, args):
   """Checks out a branch associated with a given Rietveld issue."""
-  # TODO(tandrii): consider adding this for Gerrit?
   _, args = parser.parse_args(args)
 
   if len(args) != 1:
     parser.print_help()
     return 1
 
-  issue_arg = ParseIssueNumberArgument(args[0])
-  if issue_arg.valid:
+  target_issue = ParseIssueNum(args[0])
+  if target_issue == None:
     parser.print_help()
     return 1
-  target_issue = issue_arg.issue
 
   key_and_issues = [x.split() for x in RunGit(
       ['config', '--local', '--get-regexp', r'branch\..*\.rietveldissue'])
       .splitlines()]
   branches = []
   for key, issue in key_and_issues:
     if issue == target_issue:
       branches.append(re.sub(r'branch\.(.*)\.rietveldissue', r'\1', key))
 
   if len(branches) == 0:
@@ skipping 70 matching lines @@
 if __name__ == '__main__':
   # These affect sys.stdout so do it outside of main() to simplify mocks in
   # unit testing.
   fix_encoding.fix_encoding()
   colorama.init(wrap="TERM" not in os.environ)
   try:
     sys.exit(main(sys.argv[1:]))
   except KeyboardInterrupt:
     sys.stderr.write('interrupted\n')
     sys.exit(1)