The Go package simplecsv is a simple mini-library to handle csv files. I'm building it to help me writing small command line scripts. Maybe it's useful to someone else as well.
Some notes:
- all read methods return the value in the csv and a second true/false value that is true if the value exists
- all write methods that change the csv return the changed csv and a true/false value if the operation was sucessful
- all cells are strings
Simplecsv works with comma separated csv files.
Reads file and parses as a SimpleCsv object. fileRead is false if there's an error reading the file or parsing the CSV.
var x simplecsv.SimpleCsv
var fileRead bool
x, fileRead = simplecsv.ReadCsvFile("my1file.csv")Create empty file and define cssv headers:
var u simplecsv.SimpleCsv
u = simplecsv.CreateEmpyCsv([]string{"Age", "Gender", "ID"})Write the SimpleCsv object to my2file.csv. If there's an error, wasWritten is false.
wasWritten := u.WriteCsvFile("my2file.csv")The cells of the first row are considered headers.
Get all headers:
headers := x.GetHeaders()Get header at position one (second position as it starts from 0):
headerName, headerExists := x.GetHeader(1)Get header position: (it returns -1 if the header does not exist)
position := x.GetHeaderPosition("Gender")Rename header: (old header, new header)
x, headerExists := x.RenameHeader("ID", "IDnumber")headerExists is false if the old header does not exist.
Get number of rows:
numberOfRows := x.GetNumberRows()Get second row:
row, rowExists := x.GetRow(1)Get second row as a map:
row, rowExists := x.GetRowAsMap(1)Add a slice to a row. The slice must have the same size as the CSV number of columns. If not wasSucessful is false.
var wasSucessful bool
x, wasSucessful = x.AddRow([]string{"24", "M", "2986732"})Add row from map: (If the map keys don't exist as columns, the value will be discarded. If a key does not exist, it will create empty cells.)
mymap := make(map[string]string)
mymap["Age"] = "62"
mymap["Gender"] = "F"
mymap["ID"] = "6463246"
var wasAdded bool
x, wasAdded = x.AddRowFromMap(mymap)Set second row (1) from a slice. The length of the slice must be the same as the number of columns and the row must already exist. If there’s an error wasSet is false.
var wasSet bool
x, wasSet = x.SetRow(1, []string{"45", "F", "8356138"})Set second row from map: If the map keys don't exist as columns, the value will be discarded. If a key does not exist, it will create empty cells.)
mymap2 := make(map[string]string)
mymap2["Age"] = "62"
mymap2["Gender"] = "F"
mymap2["ID"] = "6463246"
var wasSet bool
x, wasSet = x.SetRowFromMap(1, mymap2)Unlike SetRowFromMap, UpdateRowCellsFromMap does not erase the cells value just because the column names are not keys in the map. It updates the cells that have the column name in the map and maintains the value of all the others.
To update the age in row 1:
mymap3 := make(map[string]string)
mymap3["Age"] = "63"
var wasUpdated bool
x, wasUpdated = x.UpdateRowCellsFromMap(1, mymap3)Delete second row: (If the row number is invalid, wasDeleted is false)
var wasDeleted bool
x, wasDeleted = x.DeleteRow(1)To add a column at the end of the CSV:
var wasSucessful bool
x, wasSucessful = x.AddEmptyColumn("NewColumn")To remove a column at position 1 (second column, because it's zero based):
var wasRemoved bool
x, wasRemoved = x.RemoveColumn(1)To remove a column by name:
var wasRemoved bool
x, wasRemoved = x.RemoveColumnByName("Gender")Get value of the cell in the second column, second row:
cellValue, cellExists := x.GetCell(1, 1)Get the value of the cell in the column Age, second row:
cellValue, cellExists := x.GetCellByField("Age", 1)Changes the value of the cell in the first column (0) and the second row (1) to "27":
var wasChanged bool
x, wasChanged = x.SetCell(0, 1, "27")The same, using the column name instead of the column position:
var wasChanged bool
x, wasChanged = x.SetCellByField("Age", 1, "27")Find the word "27" in the first column (column 0):
rowsWithWord, validColumn := x.FindInColumn(0, "27")It returns a slice of rownumbers (int) where you can find the word in the column position. Please note that in simplecsv all cells are strings.
If it doesn't find the value, it returns an empty slice.
In FindInColumn and in FindInField case does not matter. Foo = foo = FOO.
The same as FindInColumn but using a column/field name instead of position. Please note that FindInField, unlike FindInColumn never includes the header in the search result.
rowsWithWord, validFieldName := x.FindInField("Age", "27")If the field name does not exist, the second value returned (validFieldName) is false.
Find where results match a regular expression in the third column (column 2):
rowsWithWord, areParamsOk := x.MatchInColumn(2, "p([a-z]+)ch$")Use ^ and $ in the regular expression to match exact results.
Same as with MatchInColumn, but with a field (column name). Find where results match a regular expression in the third column (column "ID"):
rowsWithWord, areParamsOk := x.MatchInField("ID", "p([a-z]+)ch$")Use ^ and $ in the regular expression to match exact results.
Please note that MatchInField, unlike MatchInColumn never includes the header in the search result.
Use boolean functions AND, OR and NOT to combine indexes and produce other complex indexes that point to rows in the csv. Very useful to produce search results.
Indexes are slices of row numbers (integrers between 0 or 1 and the length of the csv -1).
var w []int
w = simplecsv.OrIndex(a,b)
w = simplecsv.OrIndex(a,b,c,d,e)Note: The OrIndex function can be used in 2 or more operands.
var p []int
p = simplecsv.AndIndex(a,b)
p = simplecsv.AndIndex(a,b,c,d,e)Note: The AndIndex function can be used in 2 or more operands.
The code bellow returns the negative of the index g, between row 1 and row 4. If g is an index with the values {1, 2} the negative of g is {3, 4}. Because 3 and 4 are the integrers between 1 and 4 that are not in g.
var g []int
min := 1
max := 4
p = simplecsv.NotIndex(g, min, max)Note: For csvs with headers the min value is usually 1 and for csvs without headers the min value is usually 0.
Only are 2 functions to simplify and sort a csv.
It removes rows that are not in the index and reorders a CSV by the index order. If header is true, it starts by the csv header.
newIndex := []int{1,3}
header := true
x, _ = x.OnlyThisRows(newIndex, header)Removes fields that are not in the list of fields, reorders the CSV by the list of fields and adds fields that do not exist as blank fields.
fieldsList := []string{"Age","ID"}
x, _ = x.OnlyThisFields(fieldsList)- Function to sort an index, incorporate that function to all methods that return and index.