chiark - git - mdw - stgit/blob - stgit/commands/undo.py

   1 # -*- coding: utf-8 -*-
   2
   3 __copyright__ = """
   4 Copyright (C) 2008, Karl Hasselström <kha@treskal.com>
   5
   6 This program is free software; you can redistribute it and/or modify
   7 it under the terms of the GNU General Public License version 2 as
   8 published by the Free Software Foundation.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program; if not, write to the Free Software
  17 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  18 """
  19
  20 from stgit.argparse import opt
  21 from stgit.commands import common
  22 from stgit.lib import git, log, transaction
  23 from stgit.out import out
  24
  25 help = 'Undo the last operation'
  26 kind = 'stack'
  27 usage = ['']
  28 description = """
  29 Reset the patch stack to the previous state. Consecutive invocations
  30 of "stg undo" will take you ever further into the past."""
  31
  32 options = [
  33     opt('-n', '--number', type = 'int', metavar = 'N', default = 1,
  34         short = 'Undo the last N commands'),
  35     opt('--hard', action = 'store_true',
  36         short = 'Discard changes in your index/worktree')]
  37
  38 directory = common.DirectoryHasRepositoryLib()
  39
  40 def func(parser, options, args):
  41     stack = directory.repository.current_stack
  42     if options.number < 1:
  43         raise common.CmdException('Bad number of commands to undo')
  44     state = log.undo_state(stack, options.number)
  45     trans = transaction.StackTransaction(stack, 'undo %d' % options.number,
  46                                          discard_changes = options.hard)
  47     try:
  48         log.reset_stack(trans, stack.repository.default_iw, state, [])
  49     except transaction.TransactionHalted:
  50         pass
  51     return trans.run(stack.repository.default_iw)