stgit/commands/clean.py

   1 from stgit.argparse import opt
   2 from stgit.commands.common import DirectoryHasRepository
   3 from stgit.lib import transaction
   4
   5 __copyright__ = """
   6 Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com>
   7
   8 This program is free software; you can redistribute it and/or modify
   9 it under the terms of the GNU General Public License version 2 as
  10 published by the Free Software Foundation.
  11
  12 This program is distributed in the hope that it will be useful,
  13 but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 GNU General Public License for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with this program; if not, see http://www.gnu.org/licenses/.
  19 """
  20
  21 help = 'Delete the empty patches in the series'
  22 kind = 'stack'
  23 usage = ['']
  24 description = """
  25 Delete the empty patches in the whole series or only those applied or
  26 unapplied. A patch is considered empty if the two commit objects
  27 representing its boundaries refer to the same tree object."""
  28
  29 args = []
  30 options = [
  31     opt(
  32         '-a',
  33         '--applied',
  34         action='store_true',
  35         short='Delete the empty applied patches',
  36     ),
  37     opt(
  38         '-u',
  39         '--unapplied',
  40         action='store_true',
  41         short='Delete the empty unapplied patches',
  42     ),
  43 ]
  44
  45 directory = DirectoryHasRepository()
  46
  47
  48 def _clean(stack, clean_applied, clean_unapplied):
  49     trans = transaction.StackTransaction(stack, 'clean', allow_conflicts=True)
  50
  51     def del_patch(pn):
  52         if pn in stack.patchorder.applied:
  53             if pn == stack.patchorder.applied[-1]:
  54                 # We're about to clean away the topmost patch. Don't
  55                 # do that if we have conflicts, since that means the
  56                 # patch is only empty because the conflicts have made
  57                 # us dump its contents into the index and worktree.
  58                 if stack.repository.default_index.conflicts():
  59                     return False
  60             return clean_applied and trans.patches[pn].data.is_nochange()
  61         elif pn in stack.patchorder.unapplied:
  62             return clean_unapplied and trans.patches[pn].data.is_nochange()
  63
  64     for pn in trans.delete_patches(del_patch):
  65         trans.push_patch(pn)
  66     trans.run()
  67
  68
  69 def func(parser, options, args):
  70     """Delete the empty patches in the series"""
  71     if len(args) != 0:
  72         parser.error('incorrect number of arguments')
  73
  74     if not (options.applied or options.unapplied):
  75         options.applied = options.unapplied = True
  76
  77     _clean(directory.repository.current_stack, options.applied, options.unapplied)