New function SetCellValue added and update godoc example.

This commit is contained in:
Ri Xu 2016-09-09 19:39:41 +08:00
parent 9c14741a6b
commit cbfd657753
2 changed files with 48 additions and 36 deletions

View File

@ -4,7 +4,6 @@ import (
"archive/zip" "archive/zip"
"bytes" "bytes"
"encoding/xml" "encoding/xml"
"fmt"
"strconv" "strconv"
"strings" "strings"
) )
@ -35,6 +34,20 @@ func OpenFile(filename string) (*File, error) {
}, nil }, nil
} }
// SetCellValue provide function to set int or string type value of a cell
func (f *File) SetCellValue(sheet string, axis string, value interface{}) {
switch t := value.(type) {
case int, int8, int16, int32, int64, float32, float64:
f.SetCellInt(sheet, axis, value.(int))
case string:
f.SetCellStr(sheet, axis, t)
case []byte:
f.SetCellStr(sheet, axis, string(t))
default:
f.SetCellStr(sheet, axis, ``)
}
}
// SetCellInt provide function to set int type value of a cell // SetCellInt provide function to set int type value of a cell
func (f *File) SetCellInt(sheet string, axis string, value int) { func (f *File) SetCellInt(sheet string, axis string, value int) {
axis = strings.ToUpper(axis) axis = strings.ToUpper(axis)
@ -56,10 +69,7 @@ func (f *File) SetCellInt(sheet string, axis string, value int) {
xlsx.SheetData.Row[xAxis].C[yAxis].T = "" xlsx.SheetData.Row[xAxis].C[yAxis].T = ""
xlsx.SheetData.Row[xAxis].C[yAxis].V = strconv.Itoa(value) xlsx.SheetData.Row[xAxis].C[yAxis].V = strconv.Itoa(value)
output, err := xml.Marshal(xlsx) output, _ := xml.Marshal(xlsx)
if err != nil {
fmt.Println(err)
}
f.saveFileList(name, replaceRelationshipsID(replaceWorkSheetsRelationshipsNameSpace(string(output)))) f.saveFileList(name, replaceRelationshipsID(replaceWorkSheetsRelationshipsNameSpace(string(output))))
} }
@ -81,13 +91,10 @@ func (f *File) SetCellStr(sheet string, axis string, value string) {
xlsx = completeRow(xlsx, rows, cell) xlsx = completeRow(xlsx, rows, cell)
xlsx = completeCol(xlsx, rows, cell) xlsx = completeCol(xlsx, rows, cell)
xlsx.SheetData.Row[xAxis].C[yAxis].T = "str" xlsx.SheetData.Row[xAxis].C[yAxis].T = `str`
xlsx.SheetData.Row[xAxis].C[yAxis].V = value xlsx.SheetData.Row[xAxis].C[yAxis].V = value
output, err := xml.Marshal(xlsx) output, _ := xml.Marshal(xlsx)
if err != nil {
fmt.Println(err)
}
f.saveFileList(name, replaceRelationshipsID(replaceWorkSheetsRelationshipsNameSpace(string(output)))) f.saveFileList(name, replaceRelationshipsID(replaceWorkSheetsRelationshipsNameSpace(string(output))))
} }
@ -162,24 +169,24 @@ func replaceWorkSheetsRelationshipsNameSpace(workbookMarshal string) string {
// Check XML tags and fix discontinuous case, for example: // Check XML tags and fix discontinuous case, for example:
// //
// <row r="15" spans="1:22" x14ac:dyDescent="0.2"> // <row r="15" spans="1:22" x14ac:dyDescent="0.2">
// <c r="A15" s="2" /> // <c r="A15" s="2" />
// <c r="B15" s="2" /> // <c r="B15" s="2" />
// <c r="F15" s="1" /> // <c r="F15" s="1" />
// <c r="G15" s="1" /> // <c r="G15" s="1" />
// </row> // </row>
// //
// in this case, we should to change it to // in this case, we should to change it to
// //
// <row r="15" spans="1:22" x14ac:dyDescent="0.2"> // <row r="15" spans="1:22" x14ac:dyDescent="0.2">
// <c r="A15" s="2" /> // <c r="A15" s="2" />
// <c r="B15" s="2" /> // <c r="B15" s="2" />
// <c r="C15" s="2" /> // <c r="C15" s="2" />
// <c r="D15" s="2" /> // <c r="D15" s="2" />
// <c r="E15" s="2" /> // <c r="E15" s="2" />
// <c r="F15" s="1" /> // <c r="F15" s="1" />
// <c r="G15" s="1" /> // <c r="G15" s="1" />
// </row> // </row>
// //
// Noteice: this method could be very slow for large spreadsheets (more than 3000 rows one sheet). // Noteice: this method could be very slow for large spreadsheets (more than 3000 rows one sheet).
func checkRow(xlsx xlsxWorksheet) xlsxWorksheet { func checkRow(xlsx xlsxWorksheet) xlsxWorksheet {
@ -223,20 +230,20 @@ func checkRow(xlsx xlsxWorksheet) xlsxWorksheet {
// //
// For example: // For example:
// //
// <row r="19" spans="2:2"> // <row r="19" spans="2:2">
// <c r="B19"> // <c r="B19">
// <f>SUM(Sheet2!D2,Sheet2!D11)</f> // <f>SUM(Sheet2!D2,Sheet2!D11)</f>
// <v>100</v> // <v>100</v>
// </c> // </c>
// </row> // </row>
// //
// to // to
// //
// <row r="19" spans="2:2"> // <row r="19" spans="2:2">
// <c r="B19"> // <c r="B19">
// <f>SUM(Sheet2!D2,Sheet2!D11)</f> // <f>SUM(Sheet2!D2,Sheet2!D11)</f>
// </c> // </c>
// </row> // </row>
func (f *File) UpdateLinkedValue() { func (f *File) UpdateLinkedValue() {
for i := 1; i <= f.SheetCount; i++ { for i := 1; i <= f.SheetCount; i++ {
var xlsx xlsxWorksheet var xlsx xlsxWorksheet

View File

@ -26,6 +26,11 @@ func TestExcelize(t *testing.T) {
file.GetCellValue("Sheet2", "C11") file.GetCellValue("Sheet2", "C11")
file.GetCellValue("Sheet2", "D11") file.GetCellValue("Sheet2", "D11")
file.GetCellValue("Sheet2", "D12") file.GetCellValue("Sheet2", "D12")
// Test SetCellValue function
file.SetCellValue("Sheet2", "F1", "Hello")
file.SetCellValue("Sheet2", "G1", []byte("World"))
file.SetCellValue("Sheet2", "F2", 42)
file.SetCellValue("Sheet2", "G2", nil)
// Test read cell value with given axis large than exists row // Test read cell value with given axis large than exists row
file.GetCellValue("Sheet2", "E13") file.GetCellValue("Sheet2", "E13")