Make b'-'/b'+' constants within HunkLine

andrewshadura · Mar 24, 2022 · e4089ac · e4089ac
1 parent 0b384ba
commit e4089ac
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/git_crecord/crpatch.py b/git_crecord/crpatch.py
@@ -427,6 +427,9 @@ def allchildren(self) -> Sequence['Hunk']:
 class HunkLine(PatchNode):
     """Represents a changed line in a hunk"""
 
+    DELETE = b'-'
+    INSERT = b'+'
+
     linetext: bytes
     hunk: 'Hunk'
 
@@ -586,10 +589,10 @@ def allchildren(self) -> Sequence[PatchNode]:
     def countchanges(self) -> tuple[int, int]:
         """changedlines -> (n+,n-)"""
         add = len(
-            [line for line in self.changedlines if line.applied and line.diffop == b"+"]
+            [line for line in self.changedlines if line.applied and line.diffop == HunkLine.INSERT]
         )
         rem = len(
-            [line for line in self.changedlines if line.applied and line.diffop == b"-"]
+            [line for line in self.changedlines if line.applied and line.diffop == HunkLine.DELETE]
         )
         return add, rem