mdw@git.distorted.org.uk Git - stgit/blob - stgit/commands/pick.py

   1 __copyright__ = """
   2 Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com>
   3
   4 This program is free software; you can redistribute it and/or modify
   5 it under the terms of the GNU General Public License version 2 as
   6 published by the Free Software Foundation.
   7
   8 This program is distributed in the hope that it will be useful,
   9 but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 GNU General Public License for more details.
  12
  13 You should have received a copy of the GNU General Public License
  14 along with this program; if not, write to the Free Software
  15 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  16 """
  17
  18 import sys, os
  19 from optparse import OptionParser, make_option
  20
  21 from stgit.commands.common import *
  22 from stgit.utils import *
  23 from stgit.out import *
  24 from stgit import stack, git
  25 from stgit.stack import Series
  26
  27
  28 help = 'import a patch from a different branch or a commit object'
  29 usage = """%prog [options] [<patch@branch>|<commit>]
  30
  31 Import a patch from a different branch or a commit object into the
  32 current series. By default, the name of the imported patch is used as
  33 the name of the current patch. It can be overridden with the '--name'
  34 option. A commit object can be reverted with the '--reverse'
  35 option. The log and author information are those of the commit object."""
  36
  37 options = [make_option('-n', '--name',
  38                        help = 'use NAME as the patch name'),
  39            make_option('-r', '--reverse',
  40                        help = 'reverse the commit object before importing',
  41                        action = 'store_true'),
  42            make_option('-p', '--parent', metavar = 'COMMITID',
  43                        help = 'use COMMITID as parent'),
  44            make_option('-x', '--expose',
  45                        help = 'append the imported commit id to the patch log',
  46                        action = 'store_true'),
  47            make_option('--fold',
  48                        help = 'fold the commit object into the current patch',
  49                        action = 'store_true'),
  50            make_option('--update',
  51                        help = 'like fold but only update the current patch files',
  52                        action = 'store_true'),
  53            make_option('--unapplied',
  54                        help = 'keep the patch unapplied',
  55                        action = 'store_true')]
  56
  57
  58 def func(parser, options, args):
  59     """Import a commit object as a new patch
  60     """
  61     if len(args) != 1:
  62         parser.error('incorrect number of arguments')
  63
  64     if not options.unapplied:
  65         check_local_changes()
  66         check_conflicts()
  67         check_head_top_equal()
  68
  69     commit_str = args[0]
  70     commit_id = git_id(commit_str)
  71     commit = git.Commit(commit_id)
  72
  73     if options.fold or options.update:
  74         if not crt_series.get_current():
  75             raise CmdException, 'No patches applied'
  76     else:
  77         patch_branch = commit_str.split('@')
  78         if options.name:
  79             patchname = options.name
  80         elif len(patch_branch) == 2:
  81             patchname = patch_branch[0]
  82         else:
  83             patchname = None
  84
  85     if options.parent:
  86         parent = git_id(options.parent)
  87     else:
  88         parent = commit.get_parent()
  89
  90     if not options.reverse:
  91         bottom = parent
  92         top = commit_id
  93     else:
  94         bottom = commit_id
  95         top = parent
  96
  97     if options.fold:
  98         out.start('Folding commit %s' % commit_id)
  99
 100         # try a direct git-apply first
 101         if not git.apply_diff(bottom, top):
 102             git.merge(bottom, git.get_head(), top, recursive = True)
 103
 104         out.done()
 105     elif options.update:
 106         rev1 = git_id('//bottom')
 107         rev2 = git_id('//top')
 108         files = git.barefiles(rev1, rev2).split('\n')
 109
 110         out.start('Updating with commit %s' % commit_id)
 111
 112         if not git.apply_diff(bottom, top, files = files):
 113             raise CmdException, 'Patch updating failed'
 114
 115         out.done()
 116     else:
 117         message = commit.get_log()
 118         if options.expose:
 119             message += '(imported from commit %s)\n' % commit.get_id_hash()
 120         author_name, author_email, author_date = \
 121                      name_email_date(commit.get_author())
 122
 123         out.start('Importing commit %s' % commit_id)
 124
 125         newpatch = crt_series.new_patch(patchname, message = message, can_edit = False,
 126                                         unapplied = True, bottom = bottom, top = top,
 127                                         author_name = author_name,
 128                                         author_email = author_email,
 129                                         author_date = author_date)
 130         # in case the patch name was automatically generated
 131         patchname = newpatch.get_name()
 132
 133         # find a patchlog to fork from
 134         (refpatchname, refbranchname, refpatchid) = parse_rev(commit_str)
 135         if refpatchname and not refpatchid and \
 136                (not refpatchid or refpatchid == 'top'):
 137             # FIXME: should also support picking //top.old
 138             if refbranchname:
 139                 # assume the refseries is OK, since we already resolved
 140                 # commit_str to a git_id
 141                 refseries = Series(refbranchname)
 142             else:
 143                 refseries = crt_series
 144             patch = refseries.get_patch(refpatchname)
 145             if patch.get_log():
 146                 out.info("Log was %s" % newpatch.get_log())
 147                 out.info("Setting log to %s\n" %  patch.get_log())
 148                 newpatch.set_log(patch.get_log())
 149                 out.info("Log is now %s" % newpatch.get_log())
 150             else:
 151                 out.info("No log for %s\n" % patchname)
 152
 153         if not options.unapplied:
 154             modified = crt_series.push_patch(patchname)
 155         else:
 156             modified = False
 157
 158         if crt_series.empty_patch(patchname):
 159             out.done('empty patch')
 160         elif modified:
 161             out.done('modified')
 162         else:
 163             out.done()
 164
 165     print_crt_patch()