# HG changeset patch
# User Pulkit Goyal <7895pulkit@gmail.com>
# Date 1527191130 -19800
# Node ID 009aa4af50939faa5f7b45fbae0b2bcee8a50131
# Parent  6f67bfe4b82f1d0f1e11a7092262f961c49fdac9
graft: factor out function to read graft state in separate function

Fatcoring out the logic in a separate function will help us in adding
conditional logic for different versions of graft state files.

Differential Revision: https://phab.mercurial-scm.org/D3651

diff -r 6f67bfe4b82f -r 009aa4af5093 mercurial/commands.py
--- a/mercurial/commands.py	Fri May 25 01:46:06 2018 +0530
+++ b/mercurial/commands.py	Fri May 25 01:15:30 2018 +0530
@@ -2222,7 +2222,7 @@
             raise error.Abort(_("can't specify --continue and revisions"))
         # read in unfinished revisions
         try:
-            nodes = repo.vfs.read('graftstate').splitlines()
+            nodes = _readgraftstate(repo)['nodes']
             revs = [repo[node].rev() for node in nodes]
         except IOError as inst:
             if inst.errno != errno.ENOENT:
@@ -2381,6 +2381,11 @@
 
     return 0
 
+def _readgraftstate(repo):
+    """read the graft state file and return a dict of the data stored in it"""
+    nodes = repo.vfs.read('graftstate').splitlines()
+    return {'nodes': nodes}
+
 @command('grep',
     [('0', 'print0', None, _('end fields with NUL')),
     ('', 'all', None, _('print all revisions that match')),