# HG changeset patch
# User Pierre-Yves David <pierre-yves.david@fb.com>
# Date 1395701460 25200
# Node ID ec7fc110faee7302b8af69c5fc584c9bdf057f0f
# Parent  deed5edb72de8f7fa442d2a97506ddea9c2d5ec4
bundle2: introduce a `parthandler` decorator

Simple syntax sugar to register an handler for a new part type.

diff -r deed5edb72de -r ec7fc110faee mercurial/bundle2.py
--- a/mercurial/bundle2.py	Mon Mar 24 12:25:33 2014 -0700
+++ b/mercurial/bundle2.py	Mon Mar 24 15:51:00 2014 -0700
@@ -148,8 +148,23 @@
     """
     return '>'+('BB'*nbparams)
 
+parthandlermapping = {}
 
-parthandlermapping = {}
+def parthandler(parttype):
+    """decorator that register a function as a bundle2 part handler
+
+    eg::
+
+        @parthandler('myparttype')
+        def myparttypehandler(...):
+            '''process a part of type "my part".'''
+            ...
+    """
+    def _decorator(func):
+        assert parttype not in parthandlermapping
+        parthandlermapping[parttype] = func
+        return func
+    return _decorator
 
 def processbundle(repo, stream):
     """This function process a bundle, apply effect to/from a repo
diff -r deed5edb72de -r ec7fc110faee tests/test-bundle2.t
--- a/tests/test-bundle2.t	Mon Mar 24 12:25:33 2014 -0700
+++ b/tests/test-bundle2.t	Mon Mar 24 15:51:00 2014 -0700
@@ -20,14 +20,13 @@
   > Emana Karassoli, Loucra Loucra Ponponto, Pata Pata, Ko Ko Ko."""
   > assert len(ELEPHANTSSONG) == 178 # future test say 178 bytes, trust it.
   > 
+  > @bundle2.parthandler('test:song')
   > def songhandler(repo, part):
   >     """handle a "test:song" bundle2 part, printing the lyrics on stdin"""
   >     repo.ui.write('The choir start singing:\n')
   >     for line in part.data.split('\n'):
   >         repo.ui.write('    %s\n' % line)
   > 
-  > bundle2.parthandlermapping['test:song'] = songhandler
-  > 
   > @command('bundle2',
   >          [('', 'param', [], 'stream level parameter'),
   >           ('', 'parts', False, 'include some arbitrary parts to the bundle'),],