..
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86 | import { Frame } from '../index';
import { writeFileSync } from 'fs';
/**
* toJSON - converts frame to JSON file, takes a path to save the file. Path is optional. If no path is provided, the file is saved in the current directory
* @param frame: the frame to be converted
* @param path: @optional the path to save the file
* @param filename: @optional the name of the file
*/
export function toJSON(this: Frame, path?: string, filename?: string): void {
const data = this.rowdata;
const header = this.columns;
// create a json object, with keys as the header and values as the row data
const json = data.map((row: any[]) => {
const obj: any = {};
for (let i = 0; i < header.length; i++) {
if (row[i]) {
obj[header[i]] = row[i];
}
}
return obj;
});
// save the json object to a file
if (path) {
writeFileSync(
`${path}/${filename ? `${filename}` : 'data'}.json`,
JSON.stringify(json)
);
} else {
writeFileSync(
`${filename ? `${filename}` : 'data'}.json`,
JSON.stringify(json)
);
}
// console.log the file path
console.log(
`${filename ? `${filename}` : 'data'}.json saved at ${
path ? path : 'root level'
}.`
);
}
/**
* toCSV - converts frame to CSV file, takes a path to save the file. Path is optional. If no path is provided, the file is saved in the current directory
* @param frame: the frame to be converted
* @param path: @optional the path to save the file
* @param filename: @optional the name of the file
*/
export function toCSV(this: Frame, path?: string, filename?: string): void {
const data = this.rowdata;
const header = this.columns;
// create a csv string, with keys as the header and values as the row data
const csv = data.map((row: any[]) => {
const csvRow = header.map((col: string) => {
if (row[header.indexOf(col)]) {
return row[header.indexOf(col)];
}
return '';
});
return csvRow.join(',');
});
// add the header to the csv string
csv.unshift(header.join(','));
// save the csv string to a file
if (path) {
writeFileSync(
`${path}/${filename ? `${filename}` : 'data'}.csv`,
csv.join('\n')
);
} else {
writeFileSync(`${filename ? `${filename}` : 'data'}.csv`, csv.join('\n'));
}
// console.log the file path
console.log(
`${filename ? `${filename}` : 'data'}.csv saved at ${
path ? path : 'root level'
}.`
);
}
|
|
