Comments
Patch
@@ -181,31 +181,22 @@ def parthandler(parttype):
assert lparttype not in parthandlermapping
parthandlermapping[lparttype] = func
return func
return _decorator
-def processbundle(repo, stream):
+def processbundle(repo, unbundler):
"""This function process a bundle, apply effect to/from a repo
- Currently it:
- - parse a stream into an unbundle20 object
- - iterate over each parts then search and use the proper handling code to
- process the part.
-
- Parts are processes in order.
+ It iterate over each parts then search and use the proper handling code to
+ process the part. Parts are processes in order.
This is very early version of this function that will be strongly reworked
before final usage.
Unknown Mandatory part will abort the process.
"""
ui = repo.ui
- # Extraction of the unbundler object will most likely change. It may be
- # done outside of this function, the unbundler would be passed as argument.
- # in all case the unbundler will eventually be created by a
- # `changegroup.readbundle` style function.
- unbundler = unbundle20(ui, stream)
# todo:
# - replace this is a init function soon.
# - exception catching
unbundler.params
iterparts = iter(unbundler)
@@ -71,11 +71,12 @@ Create an extension to test bundle2 API
> @command('unbundle2', [], '')
> def cmdunbundle2(ui, repo):
> """process a bundle2 stream from stdin on the current repo"""
> try:
> try:
- > bundle2.processbundle(repo, sys.stdin)
+ > unbundler = bundle2.unbundle20(ui, sys.stdin)
+ > bundle2.processbundle(repo, unbundler)
> except KeyError, exc:
> raise util.Abort('missing support for %s' % exc)
> finally:
> remains = sys.stdin.read()
> ui.write('%i unread bytes\n' % len(remains))