-
Notifications
You must be signed in to change notification settings - Fork 11
/
tablib_json.go
72 lines (62 loc) · 1.62 KB
/
tablib_json.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package tablib
import (
"encoding/json"
)
// LoadJSON loads a dataset from a YAML source.
func LoadJSON(jsonContent []byte) (*Dataset, error) {
var input []map[string]interface{}
if err := json.Unmarshal(jsonContent, &input); err != nil {
return nil, err
}
return internalLoadFromDict(input)
}
// LoadDatabookJSON loads a Databook from a JSON source.
func LoadDatabookJSON(jsonContent []byte) (*Databook, error) {
var input []map[string]interface{}
var internalInput []map[string]interface{}
if err := json.Unmarshal(jsonContent, &input); err != nil {
return nil, err
}
db := NewDatabook()
for _, d := range input {
b, err := json.Marshal(d["data"])
if err != nil {
return nil, err
}
if err := json.Unmarshal(b, &internalInput); err != nil {
return nil, err
}
if ds, err := internalLoadFromDict(internalInput); err == nil {
db.AddSheet(d["title"].(string), ds)
} else {
return nil, err
}
}
return db, nil
}
// JSON returns a JSON representation of the Dataset as an Exportable.
func (d *Dataset) JSON() (*Exportable, error) {
back := d.Dict()
b, err := json.Marshal(back)
if err != nil {
return nil, err
}
return newExportableFromBytes(b), nil
}
// JSON returns a JSON representation of the Databook as an Exportable.
func (d *Databook) JSON() (*Exportable, error) {
b := newBuffer()
b.WriteString("[")
for _, s := range d.sheets {
b.WriteString("{\"title\": \"" + s.title + "\", \"data\": ")
js, err := s.dataset.JSON()
if err != nil {
return nil, err
}
b.Write(js.Bytes())
b.WriteString("},")
}
by := b.Bytes()
by[len(by)-1] = ']'
return newExportableFromBytes(by), nil
}