chiark / gitweb /
Clean-up the number of imports in main.py
[stgit] / stgit / commands / pop.py
1
2 __copyright__ = """
3 Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com>
4
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License version 2 as
7 published by the Free Software Foundation.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 """
18
19 import sys, os
20 from optparse import OptionParser, make_option
21
22 from stgit.commands.common import *
23 from stgit.utils import *
24 from stgit import stack, git
25
26
27 help = 'pop one or more patches from the stack'
28 usage = """%prog [options] [<patch>]
29
30 Pop the topmost patch or a range of patches starting with the topmost
31 one from the stack. The command fails if there are local changes or
32 conflicts. If a patch name is given as argument, the command will pop
33 all the patches up to the given one."""
34
35 options = [make_option('-a', '--all',
36                        help = 'pop all the applied patches',
37                        action = 'store_true'),
38            make_option('-n', '--number', type = 'int',
39                        help = 'pop the specified number of patches'),
40            make_option('-k', '--keep',
41                        help = 'keep the local changes',
42                        action = 'store_true')]
43
44
45 def func(parser, options, args):
46     """Pop the topmost patch from the stack
47     """
48     if len(args) > 1:
49         parser.error('incorrect number of arguments')
50
51     check_conflicts()
52     check_head_top_equal()
53
54     if not options.keep:
55         check_local_changes()
56
57     applied = crt_series.get_applied()
58     if not applied:
59         raise CmdException, 'No patches applied'
60     # the popping is done in reverse order
61     applied.reverse()
62
63     if options.all:
64         patches = applied
65     elif options.number:
66         patches = applied[:options.number]
67     elif len(args) == 1:
68         upto_patch = args[0]
69         if upto_patch not in applied:
70             if upto_patch in crt_series.get_unapplied():
71                 raise CmdException, 'Patch "%s" is not currently applied' \
72                       % upto_patch
73             else:
74                 raise CmdException, 'Patch "%s" does not exist' % upto_patch
75         patches = applied[:applied.index(upto_patch)]
76     else:
77         patches = [applied[0]]
78
79     if patches == []:
80         raise CmdException, 'No patches to pop'
81
82     pop_patches(patches, options.keep)
83
84     print_crt_patch()