- Added a new attribute on Tweet model: geo (Location name)

Jefferson-Henrique · Nov 11, 2015 · 669798c · 669798c
1 parent 8ce970e
commit 669798c
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 4 deletions.
diff --git a/src/me/jhenrique/main/Exporter.java b/src/me/jhenrique/main/Exporter.java
@@ -57,12 +57,12 @@ public static void main(String[] args) {
 
 			try {
 				BufferedWriter bw = new BufferedWriter(new FileWriter("output_got.csv"));
-				bw.write("username;date;retweets;favorites;text");
+				bw.write("username;date;retweets;favorites;text;geo");
 				bw.newLine();
 
 				System.out.println("Searching... \n");
 				for (Tweet t : TweetManager.getTweets(criteria)) {
-					bw.write(String.format("%s;%s;%d;%d;\"%s\"", t.getUsername(), sdf.format(t.getDate()), t.getRetweets(), t.getFavorites(), t.getText()));
+					bw.write(String.format("%s;%s;%d;%d;\"%s\";%s", t.getUsername(), sdf.format(t.getDate()), t.getRetweets(), t.getFavorites(), t.getText(), t.getGeo()));
 					bw.newLine();
 				}
 

diff --git a/src/me/jhenrique/manager/TweetManager.java b/src/me/jhenrique/manager/TweetManager.java
@@ -94,9 +94,16 @@ public static List<Tweet> getTweets(TwitterCriteria criteria) {
 					int retweets = Integer.valueOf(tweet.select("span.ProfileTweet-action--retweet span.ProfileTweet-actionCount").attr("data-tweet-stat-count").replaceAll(",", ""));
 					int favorites = Integer.valueOf(tweet.select("span.ProfileTweet-action--favorite span.ProfileTweet-actionCount").attr("data-tweet-stat-count").replaceAll(",", ""));
 					long dateMs = Long.valueOf(tweet.select("small.time span.js-short-timestamp").attr("data-time-ms"));
+
+					String geo = "";
+					Elements geoElement = tweet.select("span.Tweet-geo");
+					if (geoElement.size() > 0) {
+						geo = geoElement.attr("title");
+					}
+
 					Date date = new Date(dateMs);
 
-					Tweet t = new Tweet(usernameTweet, txt, date, retweets, favorites);
+					Tweet t = new Tweet(usernameTweet, txt, date, retweets, favorites, geo);
 					results.add(t);
 
 					if (criteria.getMaxTweets() > 0 && results.size() >= criteria.getMaxTweets()) {

diff --git a/src/me/jhenrique/model/Tweet.java b/src/me/jhenrique/model/Tweet.java
@@ -19,16 +19,19 @@ public class Tweet {
 
 	private int favorites;
 
+	private String geo;
+
 	public Tweet() {
 	}
 
 	public Tweet(String username, String text, Date date, int retweets,
-			int favorites) {
+			int favorites, String geo) {
 		this.username = username;
 		this.text = text;
 		this.date = date;
 		this.retweets = retweets;
 		this.favorites = favorites;
+		this.geo = geo;
 	}
 
 	public String getUsername() {
@@ -71,4 +74,12 @@ public void setFavorites(int favorites) {
 		this.favorites = favorites;
 	}
 
+	public String getGeo() {
+		return geo;
+	}
+
+	public void setGeo(String geo) {
+		this.geo = geo;
+	}
+
 }