github.com/mattdotmatt/gauge@v0.3.2-0.20160421115137-425a4cdccb62/parser/tableParser.go (about) 1 // Copyright 2015 ThoughtWorks, Inc. 2 3 // This file is part of Gauge. 4 5 // Gauge is free software: you can redistribute it and/or modify 6 // it under the terms of the GNU General Public License as published by 7 // the Free Software Foundation, either version 3 of the License, or 8 // (at your option) any later version. 9 10 // Gauge is distributed in the hope that it will be useful, 11 // but WITHOUT ANY WARRANTY; without even the implied warranty of 12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13 // GNU General Public License for more details. 14 15 // You should have received a copy of the GNU General Public License 16 // along with Gauge. If not, see <http://www.gnu.org/licenses/>. 17 18 package parser 19 20 import ( 21 "encoding/csv" 22 "strings" 23 24 "github.com/getgauge/gauge/gauge" 25 "github.com/getgauge/gauge/gauge_messages" 26 ) 27 28 func TableFrom(protoTable *gauge_messages.ProtoTable) *gauge.Table { 29 table := &gauge.Table{} 30 table.AddHeaders(protoTable.GetHeaders().GetCells()) 31 for _, row := range protoTable.GetRows() { 32 table.AddRowValues(row.GetCells()) 33 } 34 return table 35 } 36 37 func convertCsvToTable(csvContents string) (*gauge.Table, error) { 38 r := csv.NewReader(strings.NewReader(csvContents)) 39 r.Comment = '#' 40 lines, err := r.ReadAll() 41 if err != nil { 42 return nil, err 43 } 44 table := new(gauge.Table) 45 for i, line := range lines { 46 if i == 0 { 47 table.AddHeaders(line) 48 } else { 49 table.AddRowValues(line) 50 } 51 } 52 return table, nil 53 }