- Add hyperlink and set formula support for cell support;

- Character limits for cells added;
- Update go test and fix typo
This commit is contained in:
Ri Xu 2017-01-19 14:05:32 +08:00
parent 52796f6e58
commit 4a9b39afc6
8 changed files with 217 additions and 112 deletions

View File

@ -106,7 +106,7 @@ func main() {
} }
``` ```
### Add pictures to XLSX files ### Add picture to XLSX files
```go ```go
package main package main
@ -125,6 +125,11 @@ func main() {
fmt.Println(err) fmt.Println(err)
os.Exit(1) os.Exit(1)
} }
err = xlsx.WriteTo("/tmp/Workbook.xlsx")
if err != nil {
fmt.Println(err)
os.Exit(1)
}
} }
``` ```

63
cell.go
View File

@ -6,8 +6,9 @@ import (
"strings" "strings"
) )
// GetCellValue provide function get value from cell by given sheet index and // GetCellValue provides function to get value from cell by given sheet index
// axis in XLSX file. The value of the merged cell is not available currently. // and axis in XLSX file. The value of the merged cell is not available
// currently.
func (f *File) GetCellValue(sheet string, axis string) string { func (f *File) GetCellValue(sheet string, axis string) string {
axis = strings.ToUpper(axis) axis = strings.ToUpper(axis)
var xlsx xlsxWorksheet var xlsx xlsxWorksheet
@ -50,8 +51,8 @@ func (f *File) GetCellValue(sheet string, axis string) string {
return "" return ""
} }
// GetCellFormula provide function get formula from cell by given sheet index // GetCellFormula provides function to get formula from cell by given sheet
// and axis in XLSX file. // index and axis in XLSX file.
func (f *File) GetCellFormula(sheet string, axis string) string { func (f *File) GetCellFormula(sheet string, axis string) string {
axis = strings.ToUpper(axis) axis = strings.ToUpper(axis)
var xlsx xlsxWorksheet var xlsx xlsxWorksheet
@ -84,3 +85,57 @@ func (f *File) GetCellFormula(sheet string, axis string) string {
} }
return "" return ""
} }
// SetCellHyperLink provides function to set cell hyperlink by given sheet index
// and link URL address. Only support external link currently.
func (f *File) SetCellHyperLink(sheet, axis, link string) {
axis = strings.ToUpper(axis)
var xlsx xlsxWorksheet
name := "xl/worksheets/" + strings.ToLower(sheet) + ".xml"
xml.Unmarshal([]byte(f.readXML(name)), &xlsx)
rID := f.addSheetRelationships(sheet, SourceRelationshipHyperLink, link, "External")
hyperlink := xlsxHyperlink{
Ref: axis,
RID: "rId" + strconv.Itoa(rID),
}
if xlsx.Hyperlinks != nil {
xlsx.Hyperlinks.Hyperlink = append(xlsx.Hyperlinks.Hyperlink, hyperlink)
} else {
hyperlinks := xlsxHyperlinks{}
hyperlinks.Hyperlink = append(hyperlinks.Hyperlink, hyperlink)
xlsx.Hyperlinks = &hyperlinks
}
output, _ := xml.Marshal(xlsx)
f.saveFileList(name, replaceWorkSheetsRelationshipsNameSpace(string(output)))
}
// SetCellFormula provides function to set cell formula by given string and
// sheet index.
func (f *File) SetCellFormula(sheet, axis, formula string) {
axis = strings.ToUpper(axis)
var xlsx xlsxWorksheet
col := string(strings.Map(letterOnlyMapF, axis))
row, _ := strconv.Atoi(strings.Map(intOnlyMapF, axis))
xAxis := row - 1
yAxis := titleToNumber(col)
name := "xl/worksheets/" + strings.ToLower(sheet) + ".xml"
xml.Unmarshal([]byte(f.readXML(name)), &xlsx)
rows := xAxis + 1
cell := yAxis + 1
xlsx = completeRow(xlsx, rows, cell)
xlsx = completeCol(xlsx, rows, cell)
if xlsx.SheetData.Row[xAxis].C[yAxis].F != nil {
xlsx.SheetData.Row[xAxis].C[yAxis].F.Content = formula
} else {
f := xlsxF{
Content: formula,
}
xlsx.SheetData.Row[xAxis].C[yAxis].F = &f
}
output, _ := xml.Marshal(xlsx)
f.saveFileList(name, replaceWorkSheetsRelationshipsNameSpace(string(output)))
}

View File

@ -85,9 +85,13 @@ func (f *File) SetCellInt(sheet string, axis string, value int) {
f.saveFileList(name, replaceWorkSheetsRelationshipsNameSpace(string(output))) f.saveFileList(name, replaceWorkSheetsRelationshipsNameSpace(string(output)))
} }
// SetCellStr provides function to set string type value of a cell. // SetCellStr provides function to set string type value of a cell. Total number
// of characters that a cell can contain 32767 characters.
func (f *File) SetCellStr(sheet string, axis string, value string) { func (f *File) SetCellStr(sheet string, axis string, value string) {
axis = strings.ToUpper(axis) axis = strings.ToUpper(axis)
if len(value) > 32767 {
value = value[0:32767]
}
var xlsx xlsxWorksheet var xlsx xlsxWorksheet
col := string(strings.Map(letterOnlyMapF, axis)) col := string(strings.Map(letterOnlyMapF, axis))
row, _ := strconv.Atoi(strings.Map(intOnlyMapF, axis)) row, _ := strconv.Atoi(strings.Map(intOnlyMapF, axis))

View File

@ -7,108 +7,119 @@ import (
func TestOpenFile(t *testing.T) { func TestOpenFile(t *testing.T) {
// Test update a XLSX file. // Test update a XLSX file.
f1, err := OpenFile("./test/Workbook1.xlsx") xlsx, err := OpenFile("./test/Workbook1.xlsx")
if err != nil { if err != nil {
t.Log(err) t.Log(err)
} }
// Test get all the rows in a not exists sheet. // Test get all the rows in a not exists sheet.
rows := f1.GetRows("Sheet4") rows := xlsx.GetRows("Sheet4")
// Test get all the rows in a sheet. // Test get all the rows in a sheet.
rows = f1.GetRows("Sheet2") rows = xlsx.GetRows("Sheet2")
for _, row := range rows { for _, row := range rows {
for _, cell := range row { for _, cell := range row {
t.Log(cell, "\t") t.Log(cell, "\t")
} }
t.Log("\r\n") t.Log("\r\n")
} }
f1.UpdateLinkedValue() xlsx.UpdateLinkedValue()
f1.SetCellDefault("SHEET2", "A1", strconv.FormatFloat(float64(100.1588), 'f', -1, 32)) xlsx.SetCellDefault("SHEET2", "A1", strconv.FormatFloat(float64(100.1588), 'f', -1, 32))
f1.SetCellDefault("SHEET2", "A1", strconv.FormatFloat(float64(-100.1588), 'f', -1, 64)) xlsx.SetCellDefault("SHEET2", "A1", strconv.FormatFloat(float64(-100.1588), 'f', -1, 64))
f1.SetCellInt("SHEET2", "A1", 100) xlsx.SetCellInt("SHEET2", "A1", 100)
f1.SetCellStr("SHEET2", "C11", "Knowns") xlsx.SetCellStr("SHEET2", "C11", "Knowns")
f1.NewSheet(3, ":\\/?*[]Maximum 31 characters allowed in sheet title.") // Test max characters in a cell.
var s = "c"
for i := 0; i < 32768; i++ {
s += "c"
}
xlsx.SetCellStr("SHEET2", "D11", s)
xlsx.NewSheet(3, ":\\/?*[]Maximum 31 characters allowed in sheet title.")
// Test set sheet name with illegal name. // Test set sheet name with illegal name.
f1.SetSheetName("Maximum 31 characters allowed i", "[Rename]:\\/?* Maximum 31 characters allowed in sheet title.") xlsx.SetSheetName("Maximum 31 characters allowed i", "[Rename]:\\/?* Maximum 31 characters allowed in sheet title.")
f1.SetCellInt("Sheet3", "A23", 10) xlsx.SetCellInt("Sheet3", "A23", 10)
f1.SetCellStr("SHEET3", "b230", "10") xlsx.SetCellStr("SHEET3", "b230", "10")
f1.SetCellStr("SHEET10", "b230", "10") xlsx.SetCellStr("SHEET10", "b230", "10")
f1.SetActiveSheet(2) xlsx.SetActiveSheet(2)
f1.GetCellFormula("Sheet1", "B19") // Test get cell formula with given rows number. xlsx.GetCellFormula("Sheet1", "B19") // Test get cell formula with given rows number.
f1.GetCellFormula("Sheet2", "B20") // Test get cell formula with illegal sheet index. xlsx.GetCellFormula("Sheet2", "B20") // Test get cell formula with illegal sheet index.
f1.GetCellFormula("Sheet1", "B20") // Test get cell formula with illegal rows number. xlsx.GetCellFormula("Sheet1", "B20") // Test get cell formula with illegal rows number.
// Test read cell value with given illegal rows number. // Test read cell value with given illegal rows number.
f1.GetCellValue("Sheet2", "a-1") xlsx.GetCellValue("Sheet2", "a-1")
// Test read cell value with given lowercase column number. // Test read cell value with given lowercase column number.
f1.GetCellValue("Sheet2", "a5") xlsx.GetCellValue("Sheet2", "a5")
f1.GetCellValue("Sheet2", "C11") xlsx.GetCellValue("Sheet2", "C11")
f1.GetCellValue("Sheet2", "D11") xlsx.GetCellValue("Sheet2", "D11")
f1.GetCellValue("Sheet2", "D12") xlsx.GetCellValue("Sheet2", "D12")
// Test SetCellValue function. // Test SetCellValue function.
f1.SetCellValue("Sheet2", "F1", "Hello") xlsx.SetCellValue("Sheet2", "F1", "Hello")
f1.SetCellValue("Sheet2", "G1", []byte("World")) xlsx.SetCellValue("Sheet2", "G1", []byte("World"))
f1.SetCellValue("Sheet2", "F2", 42) xlsx.SetCellValue("Sheet2", "F2", 42)
f1.SetCellValue("Sheet2", "F2", int8(42)) xlsx.SetCellValue("Sheet2", "F2", int8(42))
f1.SetCellValue("Sheet2", "F2", int16(42)) xlsx.SetCellValue("Sheet2", "F2", int16(42))
f1.SetCellValue("Sheet2", "F2", int32(42)) xlsx.SetCellValue("Sheet2", "F2", int32(42))
f1.SetCellValue("Sheet2", "F2", int64(42)) xlsx.SetCellValue("Sheet2", "F2", int64(42))
f1.SetCellValue("Sheet2", "F2", float32(42.65418)) xlsx.SetCellValue("Sheet2", "F2", float32(42.65418))
f1.SetCellValue("Sheet2", "F2", float64(-42.65418)) xlsx.SetCellValue("Sheet2", "F2", float64(-42.65418))
f1.SetCellValue("Sheet2", "F2", float32(42)) xlsx.SetCellValue("Sheet2", "F2", float32(42))
f1.SetCellValue("Sheet2", "F2", float64(42)) xlsx.SetCellValue("Sheet2", "F2", float64(42))
f1.SetCellValue("Sheet2", "G2", nil) xlsx.SetCellValue("Sheet2", "G2", nil)
// Test completion column. // Test completion column.
f1.SetCellValue("Sheet2", "M2", nil) xlsx.SetCellValue("Sheet2", "M2", nil)
// Test read cell value with given axis large than exists row. // Test read cell value with given axis large than exists row.
f1.GetCellValue("Sheet2", "E231") xlsx.GetCellValue("Sheet2", "E231")
// Test get active sheet of XLSX and get sheet name of XLSX by given sheet index. // Test get active sheet of XLSX and get sheet name of XLSX by given sheet index.
f1.GetSheetName(f1.GetActiveSheetIndex()) xlsx.GetSheetName(xlsx.GetActiveSheetIndex())
// Test get sheet name of XLSX by given invalid sheet index. // Test get sheet name of XLSX by given invalid sheet index.
f1.GetSheetName(4) xlsx.GetSheetName(4)
// Test get sheet map of XLSX. // Test get sheet map of XLSX.
f1.GetSheetMap() xlsx.GetSheetMap()
for i := 1; i <= 300; i++ { for i := 1; i <= 300; i++ {
f1.SetCellStr("SHEET3", "c"+strconv.Itoa(i), strconv.Itoa(i)) xlsx.SetCellStr("SHEET3", "c"+strconv.Itoa(i), strconv.Itoa(i))
} }
err = f1.Save() err = xlsx.Save()
if err != nil {
t.Log(err)
}
// Test add picture to sheet.
err = f1.AddPicture("Sheet2", "I1", "L10", "./test/images/excel.jpg")
if err != nil {
t.Log(err)
}
err = f1.AddPicture("Sheet1", "F21", "G25", "./test/images/excel.png")
if err != nil {
t.Log(err)
}
err = f1.AddPicture("Sheet2", "L1", "O10", "./test/images/excel.bmp")
if err != nil {
t.Log(err)
}
err = f1.AddPicture("Sheet1", "G21", "H25", "./test/images/excel.ico")
if err != nil {
t.Log(err)
}
// Test add picture to sheet with unsupport file type.
err = f1.AddPicture("Sheet1", "G21", "H25", "./test/images/excel.icon")
if err != nil {
t.Log(err)
}
// Test add picture to sheet with invalid file path.
err = f1.AddPicture("Sheet1", "G21", "H25", "./test/Workbook1.xlsx")
if err != nil {
t.Log(err)
}
// Test write file to given path.
err = f1.WriteTo("./test/Workbook_2.xlsx")
if err != nil { if err != nil {
t.Log(err) t.Log(err)
} }
// Test write file to not exist directory. // Test write file to not exist directory.
err = f1.WriteTo("") err = xlsx.WriteTo("")
if err != nil {
t.Log(err)
}
}
func TestAddPicture(t *testing.T) {
xlsx, err := OpenFile("./test/Workbook1.xlsx")
if err != nil {
t.Log(err)
}
// Test add picture to sheet.
err = xlsx.AddPicture("Sheet2", "I1", "L10", "./test/images/excel.jpg")
if err != nil {
t.Log(err)
}
err = xlsx.AddPicture("Sheet1", "F21", "G25", "./test/images/excel.png")
if err != nil {
t.Log(err)
}
err = xlsx.AddPicture("Sheet2", "L1", "O10", "./test/images/excel.bmp")
if err != nil {
t.Log(err)
}
err = xlsx.AddPicture("Sheet1", "G21", "H25", "./test/images/excel.ico")
if err != nil {
t.Log(err)
}
// Test add picture to sheet with unsupport file type.
err = xlsx.AddPicture("Sheet1", "G21", "H25", "./test/images/excel.icon")
if err != nil {
t.Log(err)
}
// Test add picture to sheet with invalid file path.
err = xlsx.AddPicture("Sheet1", "G21", "H25", "./test/Workbook1.xlsx")
if err != nil {
t.Log(err)
}
// Test write file to given path.
err = xlsx.WriteTo("./test/Workbook_2.xlsx")
if err != nil { if err != nil {
t.Log(err) t.Log(err)
} }
@ -116,13 +127,13 @@ func TestOpenFile(t *testing.T) {
func TestBrokenFile(t *testing.T) { func TestBrokenFile(t *testing.T) {
// Test write file with broken file struct. // Test write file with broken file struct.
f2 := File{} xlsx := File{}
err := f2.Save() err := xlsx.Save()
if err != nil { if err != nil {
t.Log(err) t.Log(err)
} }
// Test write file with broken file struct with given path. // Test write file with broken file struct with given path.
err = f2.WriteTo("./test/Workbook_3.xlsx") err = xlsx.WriteTo("./test/Workbook_3.xlsx")
if err != nil { if err != nil {
t.Log(err) t.Log(err)
} }
@ -143,35 +154,63 @@ func TestBrokenFile(t *testing.T) {
func TestCreateFile(t *testing.T) { func TestCreateFile(t *testing.T) {
// Test create a XLSX file. // Test create a XLSX file.
f4 := CreateFile() xlsx := CreateFile()
f4.NewSheet(2, "XLSXSheet2") xlsx.NewSheet(2, "XLSXSheet2")
f4.NewSheet(3, "XLSXSheet3") xlsx.NewSheet(3, "XLSXSheet3")
f4.SetCellInt("Sheet2", "A23", 56) xlsx.SetCellInt("Sheet2", "A23", 56)
f4.SetCellStr("SHEET1", "B20", "42") xlsx.SetCellStr("SHEET1", "B20", "42")
f4.SetActiveSheet(0) xlsx.SetActiveSheet(0)
// Test add picture to sheet. // Test add picture to sheet.
err := f4.AddPicture("Sheet1", "H2", "K12", "./test/images/excel.gif") err := xlsx.AddPicture("Sheet1", "H2", "K12", "./test/images/excel.gif")
if err != nil { if err != nil {
t.Log(err) t.Log(err)
} }
err = f4.AddPicture("Sheet1", "C2", "F12", "./test/images/excel.tif") err = xlsx.AddPicture("Sheet1", "C2", "F12", "./test/images/excel.tif")
if err != nil { if err != nil {
t.Log(err) t.Log(err)
} }
err = f4.WriteTo("./test/Workbook_3.xlsx") err = xlsx.WriteTo("./test/Workbook_3.xlsx")
if err != nil { if err != nil {
t.Log(err) t.Log(err)
} }
} }
func TestSetColWidth(t *testing.T) { func TestSetColWidth(t *testing.T) {
f5, err := OpenFile("./test/Workbook1.xlsx") xlsx, err := OpenFile("./test/Workbook1.xlsx")
if err != nil { if err != nil {
t.Log(err) t.Log(err)
} }
f5.SetColWidth("sheet1", "B", "A", 12) xlsx.SetColWidth("sheet1", "B", "A", 12)
f5.SetColWidth("sheet1", "A", "B", 12) xlsx.SetColWidth("sheet1", "A", "B", 12)
err = f5.Save() err = xlsx.Save()
if err != nil {
t.Log(err)
}
}
func TestSetCellHyperLink(t *testing.T) {
xlsx, err := OpenFile("./test/Workbook1.xlsx")
if err != nil {
t.Log(err)
}
// Test set cell hyperlink in a work sheet already have hyperlinks.
xlsx.SetCellHyperLink("sheet1", "B19", "https://github.com/Luxurioust/excelize")
// Test add first hyperlink in a work sheet.
xlsx.SetCellHyperLink("sheet2", "C1", "https://github.com/Luxurioust/excelize")
err = xlsx.Save()
if err != nil {
t.Log(err)
}
}
func TestSetCellFormula(t *testing.T) {
xlsx, err := OpenFile("./test/Workbook1.xlsx")
if err != nil {
t.Log(err)
}
xlsx.SetCellFormula("sheet1", "B19", "SUM(Sheet2!D2,Sheet2!D11)")
xlsx.SetCellFormula("sheet1", "C19", "SUM(Sheet2!D2,Sheet2!D9)")
err = xlsx.Save()
if err != nil { if err != nil {
t.Log(err) t.Log(err)
} }

View File

@ -53,7 +53,7 @@ func (f *File) AddPicture(sheet string, xAxis string, yAxis string, picture stri
drawingXML = strings.Replace(sheetRelationshipsDrawingXML, "..", "xl", -1) drawingXML = strings.Replace(sheetRelationshipsDrawingXML, "..", "xl", -1)
} else { } else {
// Add first picture for given sheet. // Add first picture for given sheet.
rID := f.addSheetRelationships(sheet, SourceRelationshipDrawingML, sheetRelationshipsDrawingXML) rID := f.addSheetRelationships(sheet, SourceRelationshipDrawingML, sheetRelationshipsDrawingXML, "")
f.addSheetDrawing(sheet, rID) f.addSheetDrawing(sheet, rID)
} }
drawingRID = f.addDrawingRelationships(drawingID, SourceRelationshipImage, "../media/image"+strconv.Itoa(pictureID)+ext) drawingRID = f.addDrawingRelationships(drawingID, SourceRelationshipImage, "../media/image"+strconv.Itoa(pictureID)+ext)
@ -66,7 +66,7 @@ func (f *File) AddPicture(sheet string, xAxis string, yAxis string, picture stri
// addSheetRelationships provides function to add // addSheetRelationships provides function to add
// xl/worksheets/_rels/sheet%d.xml.rels by given sheet name, relationship type // xl/worksheets/_rels/sheet%d.xml.rels by given sheet name, relationship type
// and target. // and target.
func (f *File) addSheetRelationships(sheet string, relType string, target string) int { func (f *File) addSheetRelationships(sheet, relType, target, targetMode string) int {
var rels = "xl/worksheets/_rels/" + strings.ToLower(sheet) + ".xml.rels" var rels = "xl/worksheets/_rels/" + strings.ToLower(sheet) + ".xml.rels"
var sheetRels xlsxWorkbookRels var sheetRels xlsxWorkbookRels
var rID = 1 var rID = 1
@ -85,6 +85,7 @@ func (f *File) addSheetRelationships(sheet string, relType string, target string
ID: ID.String(), ID: ID.String(),
Type: relType, Type: relType,
Target: target, Target: target,
TargetMode: targetMode,
}) })
output, err := xml.Marshal(sheetRels) output, err := xml.Marshal(sheetRels)
if err != nil { if err != nil {

View File

@ -8,10 +8,9 @@ import (
"strings" "strings"
) )
// Sprint formats using the default formats for its operands and returns the // NewSheet provice function to greate a new sheet by given index, when creating
// resulting string. NewSheet provice function to greate a new sheet by given // a new XLSX file, the default sheet will be create, when you create a new
// index, when creating a new XLSX file, the default sheet will be create, when // file, you need to ensure that the index is continuous.
// you create a new file, you need to ensure that the index is continuous.
func (f *File) NewSheet(index int, name string) { func (f *File) NewSheet(index int, name string) {
// Update docProps/app.xml // Update docProps/app.xml
f.setAppXML() f.setAppXML()
@ -126,7 +125,7 @@ func replaceRelationshipsNameSpace(workbookMarshal string) string {
return strings.Replace(workbookMarshal, oldXmlns, newXmlns, -1) return strings.Replace(workbookMarshal, oldXmlns, newXmlns, -1)
} }
// SetActiveSheet provide function to set default active sheet of XLSX by given // SetActiveSheet provides function to set default active sheet of XLSX by given
// index. // index.
func (f *File) SetActiveSheet(index int) { func (f *File) SetActiveSheet(index int) {
var content xlsxWorkbook var content xlsxWorkbook

View File

@ -5,6 +5,7 @@ const (
SourceRelationship = "http://schemas.openxmlformats.org/officeDocument/2006/relationships" SourceRelationship = "http://schemas.openxmlformats.org/officeDocument/2006/relationships"
SourceRelationshipImage = "http://schemas.openxmlformats.org/officeDocument/2006/relationships/image" SourceRelationshipImage = "http://schemas.openxmlformats.org/officeDocument/2006/relationships/image"
SourceRelationshipDrawingML = "http://schemas.openxmlformats.org/officeDocument/2006/relationships/drawing" SourceRelationshipDrawingML = "http://schemas.openxmlformats.org/officeDocument/2006/relationships/drawing"
SourceRelationshipHyperLink = "http://schemas.openxmlformats.org/officeDocument/2006/relationships/hyperlink"
SourceRelationshipWorkSheet = "http://schemas.openxmlformats.org/officeDocument/2006/relationships/worksheet" SourceRelationshipWorkSheet = "http://schemas.openxmlformats.org/officeDocument/2006/relationships/worksheet"
NameSpaceDrawingML = "http://schemas.openxmlformats.org/drawingml/2006/main" NameSpaceDrawingML = "http://schemas.openxmlformats.org/drawingml/2006/main"
NameSpaceSpreadSheetDrawing = "http://schemas.openxmlformats.org/drawingml/2006/spreadsheetDrawing" NameSpaceSpreadSheetDrawing = "http://schemas.openxmlformats.org/drawingml/2006/spreadsheetDrawing"

View File

@ -16,11 +16,12 @@ type xlsxWorkbookRels struct {
Relationships []xlsxWorkbookRelation `xml:"Relationship"` Relationships []xlsxWorkbookRelation `xml:"Relationship"`
} }
// xmlxWorkbookRelation maps sheet id and xl/worksheets/sheet%d.xml // xmlxWorkbookRelation maps sheet id and xl/worksheets/_rels/sheet%d.xml.rels
type xlsxWorkbookRelation struct { type xlsxWorkbookRelation struct {
ID string `xml:"Id,attr"` ID string `xml:"Id,attr"`
Target string `xml:",attr"` Target string `xml:",attr"`
Type string `xml:",attr"` Type string `xml:",attr"`
TargetMode string `xml:",attr,omitempty"`
} }
// xlsxWorkbook directly maps the workbook element from the namespace // xlsxWorkbook directly maps the workbook element from the namespace