From 8e75a646d1f8bb64379a67b1160d681768f3ec5f Mon Sep 17 00:00:00 2001 From: Nick White Date: Wed, 28 Aug 2019 19:30:53 +0100 Subject: Add standalone graph tool; confgraph --- bookpipeline/graph.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'bookpipeline/graph.go') diff --git a/bookpipeline/graph.go b/bookpipeline/graph.go index a4fdee0..4386218 100644 --- a/bookpipeline/graph.go +++ b/bookpipeline/graph.go @@ -23,12 +23,16 @@ type GraphConf struct { Pgnum, Conf float64 } -func Graph(confs map[string]*Conf, bookname string, w io.Writer) (error) { +func Graph(confs map[string]*Conf, bookname string, w io.Writer) error { // Organise confs to sort them by page var graphconf []GraphConf for _, conf := range confs { name := filepath.Base(conf.Path) - numend := strings.Index(name, "_") + var numend int + numend = strings.Index(name, "_") + if numend == -1 { + numend = strings.Index(name, ".") + } pgnum, err := strconv.ParseFloat(name[0:numend], 64) if err != nil { continue -- cgit v1.2.1-24-ge1ad