changeset 47510:94c58f3aab56

dirstate: add a `in_merge` property This factor the "p2 is not null" check and is fairly simpler to read. Differential Revision: https://phab.mercurial-scm.org/D10952
author Pierre-Yves David <pierre-yves.david@octobus.net>
date Sat, 03 Jul 2021 20:12:46 +0200
parents 80dc1d452993
children eaae39894312
files mercurial/dirstate.py
diffstat 1 files changed, 9 insertions(+), 4 deletions(-) [+]
line wrap: on
line diff
--- a/mercurial/dirstate.py	Sat Jul 03 04:01:17 2021 +0200
+++ b/mercurial/dirstate.py	Sat Jul 03 20:12:46 2021 +0200
@@ -337,6 +337,11 @@
     def p2(self):
         return self._validate(self._pl[1])
 
+    @property
+    def in_merge(self):
+        """True if a merge is in progress"""
+        return self._pl[1] != self._nodeconstants.nullid
+
     def branch(self):
         return encoding.tolocal(self._branch)
 
@@ -513,7 +518,7 @@
 
     def normallookup(self, f):
         '''Mark a file normal, but possibly dirty.'''
-        if self._pl[1] != self._nodeconstants.nullid:
+        if self.in_merge:
             # if there is a merge going on and the file was either
             # in state 'm' (-1) or coming from other parent (-2) before
             # being removed, restore that state.
@@ -535,7 +540,7 @@
 
     def otherparent(self, f):
         '''Mark as coming from the other parent, always dirty.'''
-        if self._pl[1] == self._nodeconstants.nullid:
+        if not self.in_merge:
             msg = _(b"setting %r to other parent only allowed in merges") % f
             raise error.Abort(msg)
         if f in self and self[f] == b'n':
@@ -556,7 +561,7 @@
         self._dirty = True
         oldstate = self[f]
         size = 0
-        if self._pl[1] != self._nodeconstants.nullid:
+        if self.in_merge:
             entry = self._map.get(f)
             if entry is not None:
                 # backup the previous state
@@ -572,7 +577,7 @@
 
     def merge(self, f):
         '''Mark a file merged.'''
-        if self._pl[1] == self._nodeconstants.nullid:
+        if not self.in_merge:
             return self.normallookup(f)
         return self.otherparent(f)