16
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
18
from subprocess import Popen, PIPE
19
from bzrlib.transport import get_transport
20
from urlparse import urlsplit, urlunsplit
21
def patch(branch, location, strip):
22
"""Apply a patch to a branch, using patch(1). URLs may be used."""
19
def patch(branch, filename, strip):
20
"""Apply a patch to a branch, using patch(1)."""
25
22
my_file = sys.stdin
27
for prefix in ('http://', 'sftp://', 'file://'):
28
if not location.startswith(prefix):
30
(scheme, loc, path, query, fragment) = urlsplit(location)
31
loc_start = urlunsplit((scheme, loc, '/', '', ''))
32
my_file = get_transport(loc_start).get(path[1:])
34
my_file = file(location, 'rb')
24
my_file = file(filename, 'rb')
35
25
cmd = ['patch', '--directory', branch.base, '--strip', str(strip)]
36
26
child_proc = Popen(cmd, stdin=PIPE)
37
27
for line in my_file: