/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/patch.py

  • Committer: Jelmer Vernooij
  • Date: 2019-04-06 12:28:50 UTC
  • mto: (7296.6.2 patch-silent)
  • mto: This revision was merged to the branch mainline in revision 7309.
  • Revision ID: jelmer@jelmer.uk-20190406122850-gegiroz6jkv1n9tl
Import patch code from bzrtools.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/patch.py
1
1
# Copyright (C) 2005, 2006 Canonical Ltd
 
2
# Copyright (C) 2005, 2008 Aaron Bentley, 2006 Michael Ellerman
2
3
#
3
4
# This program is free software; you can redistribute it and/or modify
4
5
# it under the terms of the GNU General Public License as published by
16
17
 
17
18
from __future__ import absolute_import
18
19
 
 
20
"""Diff and patch functionality"""
 
21
 
19
22
import errno
20
23
import os
21
24
from subprocess import Popen, PIPE
 
25
import sys
22
26
 
23
 
from .errors import NoDiff3
 
27
from .errors import NoDiff3, BzrError
24
28
from .textfile import check_text_path
25
29
 
26
 
"""Diff and patch functionality"""
27
 
 
28
 
__docformat__ = "restructuredtext"
 
30
class PatchFailed(BzrError):
 
31
 
 
32
    _fmt = """Patch application failed"""
 
33
 
 
34
 
 
35
class PatchInvokeError(BzrError):
 
36
 
 
37
    _fmt = """Error invoking patch: %(errstr)s%(stderr)s"""
 
38
    internal_error = False
 
39
 
 
40
    def __init__(self, e, stderr=''):
 
41
        self.exception = e
 
42
        self.errstr = os.strerror(e.errno)
 
43
        self.stderr = '\n' + stderr
29
44
 
30
45
 
31
46
_do_close_fds = True
102
117
    finally:
103
118
        f.close()
104
119
    return status
 
120
 
 
121
 
 
122
def patch_multi(tree, location, strip, quiet=False):
 
123
    """Apply a patch to a branch, using patch(1).  URLs may be used."""
 
124
    my_file = None
 
125
    if location is None:
 
126
        my_file = sys.stdin
 
127
    else:
 
128
        my_file = open_from_url(location)
 
129
    patches = [my_file.read()]
 
130
    return run_patch(tree.basedir, patches, strip, quiet=quiet)
 
131
 
 
132
 
 
133
def run_patch(directory, patches, strip=0, reverse=False, dry_run=False,
 
134
              quiet=False, _patch_cmd='patch', target_file=None):
 
135
    args = [_patch_cmd, '-d', directory, '-s', '-p%d' % strip, '-f']
 
136
    if quiet:
 
137
        args.append('--quiet')
 
138
 
 
139
    if sys.platform == "win32":
 
140
        args.append('--binary')
 
141
 
 
142
    if reverse:
 
143
        args.append('-R')
 
144
    if dry_run:
 
145
        if sys.platform.startswith('freebsd'):
 
146
            args.append('--check')
 
147
        else:
 
148
            args.append('--dry-run')
 
149
        stderr = PIPE
 
150
    else:
 
151
        stderr = None
 
152
    if target_file is not None:
 
153
        args.append(target_file)
 
154
 
 
155
    try:
 
156
        process = Popen(args, stdin=PIPE, stdout=PIPE, stderr=stderr)
 
157
    except OSError as e:
 
158
        raise PatchInvokeError(e)
 
159
    try:
 
160
        for patch in patches:
 
161
            process.stdin.write(str(patch))
 
162
        process.stdin.close()
 
163
 
 
164
    except IOError as e:
 
165
        raise PatchInvokeError(e, process.stderr.read())
 
166
 
 
167
    result = process.wait()
 
168
    if not dry_run:
 
169
        sys.stdout.write(process.stdout.read())
 
170
    if result != 0:
 
171
        raise PatchFailed()
 
172
 
 
173
    return result