-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathTest_CSVWriter.java
119 lines (91 loc) · 4.44 KB
/
Test_CSVWriter.java
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
import java.io.FileWriter;
import java.io.IOException;
import java.time.Instant;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
public class Test_CSVWriter {
//Delimiter used in CSV file
private static final String COMMA_DELIMITER = ",";
private static final String NEW_LINE_SEPARATOR = "\n";
//CSV file header
private static final String FILE_HEADER = "responsecode,message,verbosemsg,resource,scanid,md5,sha1,sha256,scandate,permalink,positive,total,percent,instant";
public static void writeCsvFile(String fileName) {
//Create new objects
// load all these fields into VirustotalResponse as params
VirustotalResponse resp1 = new VirustotalResponse( Test_RestAPIClient01.responsecode, Test_RestAPIClient01.message, Test_RestAPIClient01.verbosemsg, Test_RestAPIClient01.resource,
Test_RestAPIClient01.scanid, Test_RestAPIClient01.md5, Test_RestAPIClient01.sha1, Test_RestAPIClient01.sha256, Test_RestAPIClient01.scandate,
Test_RestAPIClient01.permalink, Test_RestAPIClient01.positive,
Test_RestAPIClient01.total, Test_RestAPIClient01.percent, Test_RestAPIClient01.instant, Test_RestAPIClient01.avs);
Iterator<Map.Entry> itr1 = Test_RestAPIClient01.avs.entrySet().iterator();
while (itr1.hasNext()) {
Map.Entry pair = itr1.next();
System.out.println(pair.getKey() + " : " + pair.getValue());
}
//CustomResponse response1 = new CustomResponse(1, true);
// Student student1 = new Student(1, "Ahmed", "Mohamed", "M", 25);
// Student student2 = new Student(2, "Sara", "Said", "F", 23);
//Create a new list of student objects
// List students = new ArrayList();
// students.add(student1);
// students.add(student2);
List responses = new ArrayList();
responses.add(resp1);
//
FileWriter fileWriter = null;
try {
fileWriter = new FileWriter(fileName);
//Write the CSV file header
fileWriter.append(FILE_HEADER.toString());
//Add a new line separator after the header
fileWriter.append(NEW_LINE_SEPARATOR);
//Write a new student object list to the CSV file
for (Object response : responses) {
fileWriter.append(String.valueOf(((VirustotalResponse) response).getResponsecode()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getMessage()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getverbosemsg()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).resource()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getscanid()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getmd5()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getsha1()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getsha256()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getscandate()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getpl()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getpos()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).gettotal()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getperc()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getinst()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(String.valueOf(((VirustotalResponse) response).getavs()));
fileWriter.append(COMMA_DELIMITER);
fileWriter.append(NEW_LINE_SEPARATOR);
}
System.out.println("CSV file was created successfully !!!");
} catch (Exception e) {
System.out.println("Error in CsvFileWriter !!!");
e.printStackTrace();
} finally {
try {
fileWriter.flush();
fileWriter.close();
} catch (IOException e) {
System.out.println("Error while flushing/closing fileWriter !!!");
e.printStackTrace();
}
}
}
}