[PATCH 3 of 7] rebase: migrate to context manager for changing dirstate parents

[Augie Fackler]

# HG changeset patch
# User Augie Fackler <augie at google.com>
# Date 1495141861 14400
#      Thu May 18 17:11:01 2017 -0400
# Node ID 045240d198722437267f7d2dadb1c1dbfae971ee
# Parent  d6560d3e5c56e4ae529ec58069aa217791f3168f
rebase: migrate to context manager for changing dirstate parents

diff --git a/hgext/rebase.py b/hgext/rebase.py
--- a/hgext/rebase.py
+++ b/hgext/rebase.py
@@ -427,9 +427,8 @@ class rebaseruntime(object):
                         mergemod.mergestate.clean(repo)
                 else:
                     # Skip commit if we are collapsing
-                    repo.dirstate.beginparentchange()
-                    repo.setparents(repo[p1].node())
-                    repo.dirstate.endparentchange()
+                    with repo.dirstate.parentchange():
+                        repo.setparents(repo[p1].node())
                     newnode = None
                 # Update the state
                 if newnode is not None: