952 > from __future__ import absolute_import |
952 > from __future__ import absolute_import |
953 > from mercurial import extensions |
953 > from mercurial import extensions |
954 > def _rebase(orig, ui, repo, *args, **kwargs): |
954 > def _rebase(orig, ui, repo, *args, **kwargs): |
955 > with repo.wlock(): |
955 > with repo.wlock(): |
956 > with repo.lock(): |
956 > with repo.lock(): |
957 > with repo.transaction('wrappedrebase'): |
957 > with repo.transaction(b'wrappedrebase'): |
958 > return orig(ui, repo, *args, **kwargs) |
958 > return orig(ui, repo, *args, **kwargs) |
959 > def wraprebase(loaded): |
959 > def wraprebase(loaded): |
960 > assert loaded |
960 > assert loaded |
961 > rebasemod = extensions.find('rebase') |
961 > rebasemod = extensions.find(b'rebase') |
962 > extensions.wrapcommand(rebasemod.cmdtable, 'rebase', _rebase) |
962 > extensions.wrapcommand(rebasemod.cmdtable, b'rebase', _rebase) |
963 > def extsetup(ui): |
963 > def extsetup(ui): |
964 > extensions.afterloaded('rebase', wraprebase) |
964 > extensions.afterloaded(b'rebase', wraprebase) |
965 > EOF |
965 > EOF |
966 |
966 |
967 $ cat >> .hg/hgrc <<EOF |
967 $ cat >> .hg/hgrc <<EOF |
968 > [extensions] |
968 > [extensions] |
969 > wraprebase=$TESTTMP/wraprebase.py |
969 > wraprebase=$TESTTMP/wraprebase.py |