1#!/usr/local/bin/python 2# $DragonFly: src/tools/tools/splitpatch/splitpatch.py,v 1.1 2005/01/10 22:20:27 joerg Exp $ 3 4"""Split a patch file into one patch for each file.""" 5 6__copyright__ = """ 7Copyright (c) 2004 Joerg Sonnenberger. All rights reserved. 8 9Redistribution and use in source and binary forms, with or without 10modification, are permitted provided that the following conditions are 11met: 12 13 * Redistributions of source code must retain the above copyright 14 notice, this list of conditions and the following disclaimer. 15 16 * Redistributions in binary form must reproduce the above copyright 17 notice, this list of conditions and the following disclaimer in the 18 documentation and/or other materials provided with the distribution. 19 20 * Neither the name of the authors nor the names of there 21 contributors may be used to endorse or promote products derived from 22 this software without specific prior written permission. 23 24THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS 25IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 26TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A 27PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR 28CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 29EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 30PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 31PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 32LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 33NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 34SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 35""" 36 37 38import os.path 39 40def directory_save(filename, patch, suffix = None, root = None, forceful = False): 41 """ 42 Saves patch into filename.suffix relative to root. 43 """ 44 if suffix is None: 45 suffix = ".patch" 46 if root is None: 47 root = "" 48 output_name = os.path.join(root, "%s%s" % (filename.replace('/',','), suffix)) 49 50 if os.path.exists(output_name) and not forceful: 51 raise IOError, 'file exists' 52 f = open(output_name,"w") 53 f.write(patch) 54 f.close() 55 56def splitpatch(source, output = directory_save, quiet = False): 57 """ 58 Split the patch in source into independent pieces 59 and call output on the result with the guessed filename 60 and the patch itself. 61 62 source has to provide an iterator like file objects. 63 """ 64 diff_line = { " ": True, "+": True, "-": True, "@": True, "!": True, '*': True } 65 buf = [] 66 filename = None 67 for line in source: 68 if not filename: 69 if line.startswith('***'): 70 # context diff 71 filename = line.split()[1] 72 elif line.startswith('+++'): 73 # unified diff 74 filename = line.split()[1] 75 76 if filename and not quiet: 77 print "Found patch for %s" % filename 78 79 buf.append(line) 80 elif diff_line.get(line[0]): 81 # valid line for a patch 82 buf.append(line) 83 else: 84 # invalid line for a patch, write out current buffer 85 output(filename, "".join(buf)) 86 87 filename = None 88 buf = [] 89 90 if filename: 91 output(filename, "".join(buf)) 92 93def main(): 94 from optparse import OptionParser 95 import sys 96 parser = OptionParser("usage: %prog [-q] [-f] [-s suffix] [input]") 97 parser.add_option("-q", "--quiet", action="store_true", dest="quiet", help="do not print names of the subpatches") 98 parser.add_option("-f", "--force", action="store_true", dest="force", help="overwrite existing patches") 99 parser.add_option("-s", "--suffix", type="string", dest="suffix", help="use SUFFIX instead of .patch for the created patches") 100 parser.add_option("-d", "--directory", type="string", dest="directory", help="create patches in DIRECTORY") 101 (options, args) = parser.parse_args() 102 if len(args) > 1: 103 parser.error("incorrect number of arguments") 104 if args: 105 source = open(args[0]) 106 else: 107 source = sys.stdin 108 splitpatch(source, lambda filename, patch: directory_save(filename, patch, forceful = options.force, 109 suffix = options.suffix, root = options.directory), quiet = options.quiet) 110 111if __name__ == '__main__': 112 main() 113