Update upload.py @ fae51921ad9d

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."""
 
 import logging
@@ skipping 604 matching lines @@
   SetProperty(settings.GetTreeStatusUrl(error_ok=True), 'Tree status URL',
               'tree-status-url', False)
   SetProperty(settings.GetViewVCUrl(), 'ViewVC URL', 'viewvc-url', True)
 
   # TODO: configure a default branch to diff against, rather than this
   # svn-based hackery.
 
 
 class ChangeDescription(object):
   """Contains a parsed form of the change description."""
-  def __init__(self, subject, log_desc, reviewers):
-    self.subject = subject
+  def __init__(self, log_desc, reviewers):
     self.log_desc = log_desc
     self.reviewers = reviewers
     self.description = self.log_desc
 
-  def Update(self):
-    initial_text = """# Enter a description of the change.
+  def Prompt(self):
+    content = """# Enter a description of the change.
 # This will displayed on the codereview site.
 # The first line will also be used as the subject of the review.
 """
-    initial_text += self.description
+    content += self.description
     if ('\nR=' not in self.description and
         '\nTBR=' not in self.description and
         self.reviewers):
-      initial_text += '\nR=' + self.reviewers
+      content += '\nR=' + self.reviewers
     if '\nBUG=' not in self.description:
-      initial_text += '\nBUG='
+      content += '\nBUG='
     if '\nTEST=' not in self.description:
-      initial_text += '\nTEST='
-    initial_text = initial_text.rstrip('\n') + '\n'
-    content = gclient_utils.RunEditor(initial_text, True)
+      content += '\nTEST='
+    content = content.rstrip('\n') + '\n'
+    content = gclient_utils.RunEditor(content, True)
     if not content:
       DieWithError('Running editor failed')
     content = re.compile(r'^#.*$', re.MULTILINE).sub('', content).strip()
-    if not content:
+    if not content.strip():
       DieWithError('No CL description, aborting')
-    self.ParseDescription(content)
+    self.description = content
 
-  def ParseDescription(self, description):
+  def ParseDescription(self):
     """Updates the list of reviewers and subject from the description."""
-    if not description:
-      self.description = description
-      return
-
-    self.description = description.strip('\n') + '\n'
-    self.subject = description.split('\n', 1)[0]
+    self.description = self.description.strip('\n') + '\n'
     # Retrieves all reviewer lines
     regexp = re.compile(r'^\s*(TBR|R)=(.+)$', re.MULTILINE)
-    self.reviewers = ','.join(
+    reviewers = ','.join(
         i.group(2).strip() for i in regexp.finditer(self.description))
+    if reviewers:
+      self.reviewers = reviewers
 
   def IsEmpty(self):
     return not self.description
 
 
 def FindCodereviewSettingsFile(filename='codereview.settings'):
   """Finds the given file starting in the cwd and going up.
 
   Only looks up to the top of the repository unless an
   'inherit-review-settings-ok' file exists in the root of the repository.
@@ skipping 181 matching lines @@
 
 def GerritUpload(options, args, cl):
   """upload the current branch to gerrit."""
   # We assume the remote called "origin" is the one we want.
   # It is probably not worthwhile to support different workflows.
   remote = 'origin'
   branch = 'master'
   if options.target_branch:
     branch = options.target_branch
 
-  log_desc = CreateDescriptionFromLog(args)
+  log_desc = options.message or CreateDescriptionFromLog(args)
   if options.reviewers:
     log_desc += '\nR=' + options.reviewers
-  change_desc = ChangeDescription(options.message, log_desc,
-                                  options.reviewers)
-  change_desc.ParseDescription(log_desc)
+  change_desc = ChangeDescription(log_desc, options.reviewers)
+  change_desc.ParseDescription()
   if change_desc.IsEmpty():
     print "Description is empty; aborting."
     return 1
 
   receive_options = []
   cc = cl.GetCCList().split(',')
   if options.cc:
     cc += options.cc.split(',')
   cc = filter(None, cc)
   if cc:
@@ skipping 12 matching lines @@
   # TODO(ukai): parse Change-Id: and set issue number?
   return 0
 
 
 def RietveldUpload(options, args, cl):
   """upload the patch to rietveld."""
   upload_args = ['--assume_yes']  # Don't ask about untracked files.
   upload_args.extend(['--server', cl.GetRietveldServer()])
   if options.emulate_svn_auto_props:
     upload_args.append('--emulate_svn_auto_props')
-  if options.from_logs and not options.message:
-    print 'Must set message for subject line if using desc_from_logs'
-    return 1
 
   change_desc = None
 
   if cl.GetIssue():
     if options.message:
       upload_args.extend(['--message', options.message])
     upload_args.extend(['--issue', cl.GetIssue()])
     print ("This branch is associated with issue %s. "
            "Adding patch to that issue." % cl.GetIssue())
   else:
-    log_desc = CreateDescriptionFromLog(args)
-    change_desc = ChangeDescription(options.message, log_desc,
-                                    options.reviewers)
-    if not options.from_logs:
-      change_desc.Update()
+    message = options.message or CreateDescriptionFromLog(args)
+    change_desc = ChangeDescription(message, options.reviewers)
+    if not options.force:
+      change_desc.Prompt()
+    change_desc.ParseDescription()
 
     if change_desc.IsEmpty():
       print "Description is empty; aborting."
       return 1
 
-    upload_args.extend(['--message', change_desc.subject])
-    upload_args.extend(['--description', change_desc.description])
+    upload_args.extend(['--message', change_desc.description])
     if change_desc.reviewers:
       upload_args.extend(['--reviewers', change_desc.reviewers])
     if options.send_mail:
       if not change_desc.reviewers:
         DieWithError("Must specify reviewers to send email.")
       upload_args.append('--send_mail')
     cc = ','.join(filter(None, (cl.GetCCList(), options.cc)))
     if cc:
       upload_args.extend(['--cc', cc])
 
@@ skipping 533 matching lines @@
              'and retry or visit go/isgaeup.\n%s') % (e.code, str(e)))
 
   # Not a known command. Default to help.
   GenUsage(parser, 'help')
   return CMDhelp(parser, argv)
 
 
 if __name__ == '__main__':
   fix_encoding.fix_encoding()
   sys.exit(main(sys.argv[1:]))