java-imdb-db-parser/ImdbReaderImpl.java at master · ns/java-imdb-db-parser · GitHub

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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
importjava.util.List;

importjavax.xml.parsers.*;
importjavax.xml.transform.*;
importjavax.xml.transform.dom.*;
importjavax.xml.transform.stream.*;
importorg.w3c.dom.*;
importjava.util.ArrayList;
importjava.io.BufferedOutputStream;
importjava.io.FileOutputStream;
importjava.io.OutputStream;

classImdbReaderImplimplementsImdbReader
{
privateArrayList<MovieData> movies = newArrayList<MovieData>();
privateArrayList<ActorData> actors = newArrayList<ActorData>();
privateArrayList<ActorData> actresses = newArrayList<ActorData>();


publicvoidnewActors( finalList<ActorData> actorList )
 {
for ( ActorDataactorData : actorList )
 {
if (!actorData.isActress()) {
actors.add(actorData);
 }
else {
actresses.add(actorData);
 }
 }
 }

publicvoidnewMovies( finalList<MovieData> movieList )
 {
for ( MovieDatamovieData : movieList )
 {
movies.add(movieData);
 }
 }

publicvoidwriteMoviesXML(finalStringpath) throwsException {
DocumentBuilderFactorydocumentBuilderFactory = DocumentBuilderFactory.newInstance();
DocumentBuilderdocumentBuilder = documentBuilderFactory.newDocumentBuilder();
Documentdocument = documentBuilder.newDocument();
ElementrootElement = document.createElement("movies");
document.appendChild(rootElement);

for (MovieDatam : movies) {
Stringelement = "movie";
Elementem = document.createElement(element);

StringelementTitle = "title";
Elementtitle = document.createElement(elementTitle);
title.appendChild(document.createTextNode(m.getTitle()));

StringelementYear = "year";
Elementyear = document.createElement(elementYear);
year.appendChild(document.createTextNode(m.getYear() + ""));

em.appendChild(title);
em.appendChild(year);
rootElement.appendChild(em);
 }

TransformerFactorytransformerFactory = TransformerFactory.newInstance();
Transformertransformer = transformerFactory.newTransformer();
DOMSourcesource = newDOMSource(document);
StreamResultresult = newStreamResult(path);
transformer.transform(source, result);
 }

publicvoidwriteActorsXML(finalStringpath) throwsException {
OutputStreamout = null;
try {
OutputStreamoutFile = newFileOutputStream(path);
out = newBufferedOutputStream(outFile);
out.write("<actors>".getBytes());


for (inti = 0; i < actors.size(); i++) {
ActorDataa = actors.get(i);

out.write("<actor>".getBytes());

out.write("<name>".getBytes());
out.write(a.getName().getBytes());
out.write("</name>".getBytes());

out.write("<roles>".getBytes());
for ( RoleDatamovieRole : a.getMovieRoles() )
 {

if (movieRole == null || movieRole.getRole() == null) {

 }
else {
out.write("<role>".getBytes());
out.write(movieRole.getRole().getBytes());
out.write("</role>".getBytes());
 }


out.write("<movie>".getBytes());
out.write(movieRole.getTitle().getBytes());
out.write("</movie>".getBytes());
 }
out.write("</roles>".getBytes());

out.write("</actor>".getBytes());

intj = 1000;
if (i % j == 0) {
System.out.println(i + "");
 }
 }

out.write("</actors>".getBytes());
 } finally {
if (out != null) {
out.close();
 }
 }
 }

publicvoidwriteActressesXML(finalStringpath) throwsException {
OutputStreamout = null;
try {
OutputStreamoutFile = newFileOutputStream(path);
out = newBufferedOutputStream(outFile);

out.write("<actresses>".getBytes());


for (inti = 0; i < actresses.size(); i++) {
ActorDataa = actresses.get(i);

out.write("<actress>".getBytes());

out.write("<name>".getBytes());
out.write(a.getName().getBytes());
out.write("</name>".getBytes());

out.write("<roles>".getBytes());
for ( RoleDatamovieRole : a.getMovieRoles() )
 {
if (movieRole == null || movieRole.getRole() == null) {

 }
else {
out.write("<role>".getBytes());
out.write(movieRole.getRole().getBytes());
out.write("</role>".getBytes());
 }


out.write("<movie>".getBytes());
out.write(movieRole.getTitle().getBytes());
out.write("</movie>".getBytes());
 }
out.write("</roles>".getBytes());

out.write("</actress>".getBytes());

intj = 1000;
if (i % j == 0) {
System.out.println(i + "");
 }
 }

out.write("</actresses>".getBytes());
 } finally {
if (out != null) {
out.close();
 }
 }
 }
}