[PATCH 3 of 3] dagop: move lines() out of annotate()

Yuya Nishihara yuya at tcha.org
Sun Mar 11 11:00:41 EDT 2018


# HG changeset patch
# User Yuya Nishihara <yuya at tcha.org>
# Date 1519849241 18000
#      Wed Feb 28 15:20:41 2018 -0500
# Node ID ba4ae5454659806d03b7e4ebceb7238bfe46443e
# Parent  f634f309bed610e84b4ff151044f320461509e78
dagop: move lines() out of annotate()

diff --git a/mercurial/dagop.py b/mercurial/dagop.py
--- a/mercurial/dagop.py
+++ b/mercurial/dagop.py
@@ -369,6 +369,11 @@ class annotateline(object):
     # Whether this annotation was the result of a skip-annotate.
     skip = attr.ib(default=False)
 
+def _countlines(text):
+    if text.endswith("\n"):
+        return text.count("\n")
+    return text.count("\n") + int(bool(text))
+
 def _annotatepair(parents, childfctx, child, skipchild, diffopts):
     r'''
     Given parent and child fctxes and annotate data for parents, for all lines
@@ -436,18 +441,13 @@ def annotate(base, parents, linenumber=F
     `parents(fctx)` is a function returning a list of parent filectxs.
     """
 
-    def lines(text):
-        if text.endswith("\n"):
-            return text.count("\n")
-        return text.count("\n") + int(bool(text))
-
     if linenumber:
         def decorate(text, rev):
             return ([annotateline(fctx=rev, lineno=i)
-                     for i in xrange(1, lines(text) + 1)], text)
+                     for i in xrange(1, _countlines(text) + 1)], text)
     else:
         def decorate(text, rev):
-            return ([annotateline(fctx=rev)] * lines(text), text)
+            return ([annotateline(fctx=rev)] * _countlines(text), text)
 
     # This algorithm would prefer to be recursive, but Python is a
     # bit recursion-hostile. Instead we do an iterative


More information about the Mercurial-devel mailing list