mdw@git.distorted.org.uk Git - stgit/blob - stgit/commands/resolved.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, basedir
  25 from stgit.config import config, file_extensions
  26 from stgit.gitmergeonefile import interactive_merge
  27
  28
  29 help = 'mark a file conflict as solved'
  30 usage = """%prog [options] [<files...>]
  31
  32 Mark a merge conflict as resolved. The conflicts can be seen with the
  33 'status' command, the corresponding files being prefixed with a
  34 'C'. This command also removes any <file>.{ancestor,current,patched}
  35 files."""
  36
  37 options = [make_option('-a', '--all',
  38                        help = 'mark all conflicts as solved',
  39                        action = 'store_true'),
  40            make_option('-r', '--reset', metavar = '(ancestor|current|patched)',
  41                        help = 'reset the file(s) to the given state'),
  42            make_option('-i', '--interactive',
  43                        help = 'run the interactive merging tool',
  44                        action = 'store_true')]
  45
  46 def func(parser, options, args):
  47     """Mark the conflict as resolved
  48     """
  49     if options.reset \
  50            and options.reset not in file_extensions():
  51         raise CmdException, 'Unknown reset state: %s' % options.reset
  52
  53     if options.all and not options.interactive:
  54         resolved_all(options.reset)
  55         return
  56
  57     conflicts = git.get_conflicts()
  58
  59     if len(args) != 0:
  60         files = args
  61     elif options.all:
  62         files = conflicts
  63     else:
  64         parser.error('incorrect number of arguments')
  65
  66     if not conflicts:
  67         raise CmdException, 'No more conflicts'
  68
  69     # check for arguments validity
  70     if not options.all:
  71         for filename in files:
  72             if not filename in conflicts:
  73                 raise CmdException, 'No conflicts for "%s"' % filename
  74
  75     # resolved
  76     try:
  77         for filename in files:
  78             if options.interactive:
  79                 interactive_merge(filename)
  80             resolved(filename, options.reset)
  81             del conflicts[conflicts.index(filename)]
  82     finally:
  83         # save or remove the conflicts file. Needs a finally clause to
  84         # ensure that already solved conflicts are marked
  85         if conflicts == []:
  86             os.remove(os.path.join(basedir.get(), 'conflicts'))
  87         else:
  88             f = file(os.path.join(basedir.get(), 'conflicts'), 'w+')
  89             f.writelines([line + '\n' for line in conflicts])
  90             f.close()