14
15
# along with this program; if not, write to the Free Software
15
16
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
from __future__ import absolute_import
18
"""Diff and patch functionality"""
21
22
from subprocess import Popen, PIPE
23
from .errors import NoDiff3
26
from .errors import NoDiff3, BzrError
24
27
from .textfile import check_text_path
26
"""Diff and patch functionality"""
28
__docformat__ = "restructuredtext"
29
class PatchFailed(BzrError):
31
_fmt = """Patch application failed"""
34
class PatchInvokeError(BzrError):
36
_fmt = """Error invoking patch: %(errstr)s%(stderr)s"""
37
internal_error = False
39
def __init__(self, e, stderr=''):
41
self.errstr = os.strerror(e.errno)
42
self.stderr = '\n' + stderr
31
45
_do_close_fds = True
97
111
if status not in (0, 1):
98
112
raise Exception(stderr)
99
f = open(out_file, 'wb')
113
with open(out_file, 'wb') as f:
118
def patch_tree(tree, patches, strip=0, reverse=False, dry_run=False,
119
quiet=False, out=None):
120
"""Apply a patch to a tree.
123
tree: A MutableTree object
124
patches: list of patches as bytes
125
strip: Strip X segments of paths
126
reverse: Apply reversal of patch
129
return run_patch(tree.basedir, patches, strip, reverse, dry_run,
133
def run_patch(directory, patches, strip=0, reverse=False, dry_run=False,
134
quiet=False, _patch_cmd='patch', target_file=None, out=None):
135
args = [_patch_cmd, '-d', directory, '-s', '-p%d' % strip, '-f']
137
args.append('--quiet')
139
if sys.platform == "win32":
140
args.append('--binary')
145
if sys.platform.startswith('freebsd'):
146
args.append('--check')
148
args.append('--dry-run')
152
if target_file is not None:
153
args.append(target_file)
156
process = Popen(args, stdin=PIPE, stdout=PIPE, stderr=stderr)
158
raise PatchInvokeError(e)
160
for patch in patches:
161
process.stdin.write(bytes(patch))
162
process.stdin.close()
165
raise PatchInvokeError(e, process.stderr.read())
167
result = process.wait()
170
out.write(process.stdout.read())
172
process.stdout.read()
179
def iter_patched_from_hunks(orig_lines, hunks):
180
"""Iterate through a series of lines with a patch applied.
181
This handles a single file, and does exact, not fuzzy patching.
183
:param orig_lines: The unpatched lines.
184
:param hunks: An iterable of Hunk instances.
186
This is different from breezy.patches in that it invokes the patch
189
with tempfile.NamedTemporaryFile() as f:
190
f.writelines(orig_lines)
192
# TODO(jelmer): Stream patch contents to command, rather than
193
# serializing the entire patch upfront.
194
serialized = b''.join([hunk.as_bytes() for hunk in hunks])
195
args = ["patch", "-f", "-s", "--posix", "--binary",
196
"-o", "-", f.name, "-r", "-"]
197
stdout, stderr, status = write_to_cmd(args, serialized)
200
raise PatchFailed(stderr)