forked from edasque/DynamoDBtoCSV
-
Notifications
You must be signed in to change notification settings - Fork 0
/
dynamoDBtoCSV.js
129 lines (103 loc) · 3.02 KB
/
dynamoDBtoCSV.js
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
var program = require('commander');
var AWS = require('aws-sdk');
var unmarshalItem = require('dynamodb-marshaler').unmarshalItem;
var unmarshal = require('dynamodb-marshaler').unmarshal;
var Papa = require('papaparse');
var fs = require('fs');
var headers = [];
var unMarshalledArray = [];
program
.version('0.0.1')
.option('-t, --table [tablename]', 'Add the table you want to output to csv')
.option("-d, --describe")
.option("-r, --region [regionname]")
.option("-e, --endpoint [url]", 'Endpoint URL, can be used to dump from local DynamoDB')
.option("-p, --profile [profile]", 'Use profile from your credentials file')
.option("-f, --file [file]", "Name of the file to be created")
.parse(process.argv);
if (!program.table) {
console.log("You must specify a table");
program.outputHelp();
process.exit(1);
}
if (program.region && AWS.config.credentials) {
AWS.config.update({region: program.region});
} else {
AWS.config.loadFromPath(__dirname + '/config.json');
}
if (program.endpoint) {
AWS.config.update({endpoint: program.endpoint})
}
if (program.profile) {
var newCreds = AWS.config.credentials;
newCreds.profile = program.profile;
AWS.config.update({credentials: newCreds});
}
var dynamoDB = new AWS.DynamoDB();
var query = {
"TableName": program.table,
"Limit": 1000
};
var describeTable = function(query) {
dynamoDB.describeTable({
"TableName": program.table
}, function(err, data) {
if (!err) {
console.dir(data.Table);
} else console.dir(err);
});
}
var scanDynamoDB = function ( query ) {
dynamoDB.scan( query, function ( err, data ) {
if ( !err ) {
unMarshalIntoArray( data.Items ); // Print out the subset of results.
if ( data.LastEvaluatedKey ) { // Result is incomplete; there is more to come.
query.ExclusiveStartKey = data.LastEvaluatedKey;
scanDynamoDB(query);
}
else {
let endData = Papa.unparse( { fields: [ ...headers ], data: unMarshalledArray } );
if(program.file){
writeData(endData)
}else{
console.log(endData);
}
}
}
else {
console.dir(err);
}
});
};
var writeData = function(data)
{
fs.writeFile(program.file, data, (err) => {
if(err) throw err;
console.log('File Saved');
});
}
function unMarshalIntoArray( items ) {
if ( items.length === 0 )
return;
items.forEach( function ( row ) {
let newRow = {};
// console.log( 'Row: ' + JSON.stringify( row ));
Object.keys( row ).forEach( function ( key ) {
if ( headers.indexOf( key.trim() ) === -1 ) {
// console.log( 'putting new key ' + key.trim() + ' into headers ' + headers.toString());
headers.push( key.trim() );
}
let newValue = unmarshal( row[key] );
if ( typeof newValue === 'object' ) {
newRow[key] = JSON.stringify( newValue );
}
else {
newRow[key] = newValue;
}
});
// console.log( newRow );
unMarshalledArray.push( newRow );
});
}
if ( program.describe ) describeTable( query );
else scanDynamoDB( query );