1. 程式人生 > >Twitter資料抓取的方法(二)

Twitter資料抓取的方法(二)

In the previous post we covered effectively the theory of how we can search and extract tweets from Twitter without having to use their API.

First, let’s have a quick recap of what we learned in the previous post. We have a URL that we can use to search Twitter with:

This includes the following parameters:

KeyValue
q URL encoded query string
f Type of query (omit for top results or realtime for all)
scroll_cursor Allows to paginate through results. If omitted it returns first page

We also know that Twitter returns the following JSON response:

{ 
    has_more_items: boolean, 
    items_html: "...", 
    is_scrolling_request: boolean, 
    is_refresh_request: boolean, 
    scroll_cursor: "...", 
    refresh_cursor: "...", 
    focused_refresh_interval: int 
}

Finally, we know that we can extract the following information for each tweet:

Embedded Tweet Data
SelectorValue
div.original-tweet[data-tweet-id] The authors twitter handle
div.original-tweet[data-name] The name of the author
div.original-tweet[data-user-id] The user ID of the author
span._timestamp[data-time] Timestamp of the post
span._timestamp[data-time-ms] Timestamp of the post in ms
p.tweet-text  Text of Tweet
span.ProfileTweet-action–retweet > span.ProfileTweet-actionCount[data-tweet-stat-count] Number of Retweets
span.ProfileTweet-action–favorite > span.ProfileTweet-actionCount[data-tweet-stat-count]  Number of Favourites

Ok, recap done, let’s consider some pseudo code to get us started. As the example is going to be in Java, the pseudo code will take on a Java syntax.

searchTwitter(String query, long rateDelay) {
  URL searchURL = createSearchURL(query)
  TwitterResponse twitterResponse
  String scrollCursor
  while ( (twitterResponse = executeSearch(searchURL)) != null && twitterResponse.has_more_items && twitterResponse.scroll_cursor != scrollCursor) {
    List tweets = extractTweest(twitterResponse.items_html)
    saveTweets(tweets)
    searchURL = createSearchURL(query, twitterResponse.scroll_cursor)
    sleep(rateDelay)
 }
}

Firstly, we define a function called searchTwitter, where we pass a query value as a string, and a specified time to pause the thread between calls. Given this string, we then pass to a function that creates our search URL based on our query. Then, in a while loop, we execute the search to return a TwitterResponse object that represents the JSON Twitter returns. Checking that the response is not null, it has more items, and we are not repeating the scroll cursor, we proceed to extract tweets from the items html, save them, and create our next search URL. We finally sleep the thread for however long we choose to with rateDelay, so we are not bombarding Twitter with a stupid amount of requests that could be viewed as a very crap DDOS.

Now that we’ve got an idea of what algorithm we’re going to use, let’s start coding.

I’m going to use Gradle as a the build system, as we are going to use some additional dependencies to make things easier. You can either download it and set it up on your machine if you want, but I’ve also added a Gradle wrapper (gradlew) to the repository so you can run without downloading Gradle. All you’ll need is to make sure that you’re JAVA_HOME Path variable is set up and pointing to wherever Java is located.

Lets take a look at the Gradle file.

apply plugin: 'java'
 
sourceCompatibility = 1.7
version = '1.0'
 
repositories {
  mavenCentral()
}
 
dependencies {
  compile 'org.apache.httpcomponents:httpclient:4.3.6'
  compile 'com.google.code.gson:gson:2.3'
  compile 'org.jsoup:jsoup:1.7.3'
  compile 'log4j:log4j:1.2.17'
 
  testCompile group: 'junit', name: 'junit', version: '4.11'
}

As this is Java project, we’ve applied the java plugin. This will generate our standard directory structure that we get with Gradle and Maven projects: src/main/java src/test/java.

In addition, there are several dependencies I’ve included to help make the task a little easier. HTTPClient provides libraries that make it easier to construct URI’s, GSON is a useful JSON processing library that will allow us to convert the response query from Twitter into a Java object, and finally JSoup is an HTML parsing library that we can use to extract what we need from the inner_html value that Twitter returns to us. Finally, I’ve included JUnit, however I won’t go into unit testing with this example.

Lets start writing our code. Again, if you’re not familiar with gradle, the root for your packages should be in src/main/java. If the folders are not already there, you can auto generate, although feel free to look at the example code if you’re still unclear.

package uk.co.tomkdickinson.twitter.search;
import java.util.Date;
 
public class Tweet {
 
    private String id;
    private String text;
    private String userId;
    private String userName;
    private String userScreenName;
    private Date createdAt;
    private int retweets;
    private int favourites;
 
    public Tweet() {
    }
 
    public Tweet(String id, String text, String userId, String userName, String userScreenName, Date createdAt, int retweets, int favourites) {
        this.id = id;
        this.text = text;
        this.userId = userId;
        this.userName = userName;
        this.userScreenName = userScreenName;
        this.createdAt = createdAt;
        this.retweets = retweets;
        this.favourites = favourites;
    }
 
    public String getId() {
        return id;
    }
 
    public void setId(String id) {
        this.id = id;
    }
 
    public String getText() {
        return text;
    }
 
    public void setText(String text) {
        this.text = text;
    }
 
    public String getUserId() {
        return userId;
    }
 
    public void setUserId(String userId) {
        this.userId = userId;
    }
 
    public String getUserName() {
        return userName;
    }
 
    public void setUserName(String userName) {
        this.userName = userName;
    }
 
    public String getUserScreenName() {
        return userScreenName;
    }
 
    public void setUserScreenName(String userScreenName) {
        this.userScreenName = userScreenName;
    }
 
    public Date getCreatedAt() {
        return createdAt;
    }
 
    public void setCreatedAt(Date createdAt) {
        this.createdAt = createdAt;
    }
 
    public int getRetweets() {
        return retweets;
    }
 
    public void setRetweets(int retweets) {
        this.retweets = retweets;
    }
 
    public int getFavourites() {
        return favourites;
    }
 
    public void setFavourites(int favourites) {
        this.favourites = favourites;
    }
}
package uk.co.tomkdickinson.twitter.search;
 
import java.util.ArrayList;
import java.util.List;
 
public class TwitterResponse {
 
    private boolean has_more_items;
    private String items_html;
    private boolean is_scrolling_request;
    private boolean is_refresh_request;
    private String scroll_cursor;
    private String refresh_cursor;
    private long focused_refresh_interval;
 
    public TwitterResponse() {
    }
 
    public TwitterResponse(boolean has_more_items, String items_html, boolean is_scrolling_request, boolean is_refresh_request, String scroll_cursor, String refresh_cursor, long focused_refresh_interval) {
        this.has_more_items = has_more_items;
        this.items_html = items_html;
        this.is_scrolling_request = is_scrolling_request;
        this.is_refresh_request = is_refresh_request;
        this.scroll_cursor = scroll_cursor;
        this.refresh_cursor = refresh_cursor;
        this.focused_refresh_interval = focused_refresh_interval;
    }
 
    public boolean isHas_more_items() {
        return has_more_items;
    }
 
    public void setHas_more_items(boolean has_more_items) {
        this.has_more_items = has_more_items;
    }
 
    public String getItems_html() {
        return items_html;
    }
 
    public void setItems_html(String items_html) {
        this.items_html = items_html;
    }
 
    public boolean isIs_scrolling_request() {
        return is_scrolling_request;
    }
 
    public void setIs_scrolling_request(boolean is_scrolling_request) {
        this.is_scrolling_request = is_scrolling_request;
    }
 
    public boolean isIs_refresh_request() {
        return is_refresh_request;
    }
 
    public void setIs_refresh_request(boolean is_refresh_request) {
        this.is_refresh_request = is_refresh_request;
    }
 
    public String getScroll_cursor() {
        return scroll_cursor;
    }
 
    public void setScroll_cursor(String scroll_cursor) {
        this.scroll_cursor = scroll_cursor;
    }
 
    public String getRefresh_cursor() {
        return refresh_cursor;
    }
 
    public void setRefresh_cursor(String refresh_cursor) {
        this.refresh_cursor = refresh_cursor;
    }
 
    public long getFocused_refresh_interval() {
        return focused_refresh_interval;
    }
 
    public void setFocused_refresh_interval(long focused_refresh_interval) {
        this.focused_refresh_interval = focused_refresh_interval;
    } 
 
    public List getTweets() {
        return new ArrayList();
    }
}

You’ll notice the additional method getTweets() in TwitterResponse. For now, just return an empty ArrayList, but we will revisit this later.
In addition to these bean classes, we also want to consider an edge case where people might use this to search for an empty, null string, or the query contains characters not allowed in a URL. Therefore to handle this, we will also create a small Exception class called InvalidQueryException.

package uk.co.tomkdickinson.twitter.search;
 
public class InvalidQueryException extends Exception{
 
    public InvalidQueryException(String query) {
        super("Query string '"+query+"' is invalid");
    }
}

Next, we need to create a TwitterSearch class and it’s basic structure. An important thing to consider here is we are interested in making the code reusable, so in the example I have made this abstract with an abstract method called saveTweets. The nice thing about this is it decouples the saving logic from the extraction logic. In other words, this will allow you to implement your own save solution without having to rewrite any of the TwitterSearch code. Additionally, you might also note that I’ve specified that the saveTweets method returns a boolean. This will allow anyone extending this to provide their own exit condition, for example once a certain number of tweets have been extracted. By returning false, we can indicate in our code to stop extracting tweets from Twitter.

package uk.co.tomkdickinson.twitter.search;
 
import java.net.URL;
import java.util.List;
 
public abstract class TwitterSearch {
 
    public TwitterSearch() {
 
    }
 
    public abstract boolean saveTweets(List tweets);
 
    public void search(final String query, final long rateDelay) throws InvalidQueryException {
 
    }
 
    public static TwitterResponse executeSearch(final URL url) {
        return null;
    }
 
    public static URL constructURL(final String query, final String scrollCursor) throws InvalidQueryException {
        return null;
    }
}

Finally, lets also create a TwitterSearchImpl. This will contain a small implementation of TwitterSearch so we can test our code as we go along.

package uk.co.tomkdickinson.twitter.search;
 
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;
 
public class TwitterSearchImpl extends TwitterSearch {
 
    private final AtomicInteger counter = new AtomicInteger();
 
    @Override
    public boolean saveTweets(List tweets) {
        if(tweets!=null) {
            for (Tweet tweet : tweets) {
                System.out.println(counter.getAndIncrement() + 1 + "[" + tweet.getCreatedAt() + "] - " + tweet.getText());
                if (counter.get() >= 500) {
                    return false;
                }
            }
        }
        return true;
    }
 
    public static void main(String[] args) throws InvalidQueryException {
        TwitterSearch twitterSearch = new TwitterSearchImpl();
        twitterSearch.search("babylon 5", 2000);
    }
}

All this implementation does is print out our tweets date and text, collecting up to a maximum of 500 where the program should terminate.

Now we have the skeleton of our project set up, lets start implementing some of the functionality. Considering our pseudo code from earlier Let’s start with TwitterSearch.class:

public void search(final String query, final long rateDelay) {
    TwitterResponse response;
    String scrollCursor = null;
    URL url = constructURL(query, scrollCursor);
    boolean continueSearch = true;
    while((response = executeSearch(url))!=null && response.isHas_more_items() && continueSearch) {
        continueSearch = saveTweets(response.getTweets());
        scrollCursor = response.getScroll_cursor();
        try {
            Thread.sleep(rateDelay);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        url = constructURL(query, scrollCursor);
    }
}

As you can probably tell, that is pretty much most of our main pseudo code implemented. Running it will have no effect, as we haven’t implemented any of the actual steps yet, but it is a good start.

Lets implement some of our other methods starting with constructURL.

public final static String TYPE_PARAM = "f";
public final static String QUERY_PARAM = "q";
public final static String SCROLL_CURSOR_PARAM = "scroll_cursor";
public final static String TWITTER_SEARCH_URL = "https://twitter.com/i/search/timeline";
 
public static URL constructURL(final String query, final String scrollCursor) throws InvalidQueryException {
        if(query==null || query.isEmpty()) {
            throw new InvalidQueryException(query);
        }
        try {
            URIBuilder uriBuilder;
            uriBuilder = new URIBuilder(TWITTER_SEARCH_URL);
            uriBuilder.addParameter(QUERY_PARAM, query);
            uriBuilder.addParameter(TYPE_PARAM, "realtime");
            if (scrollCursor != null) {
                uriBuilder.addParameter(SCROLL_CURSOR_PARAM, scrollCursor);
            }
            return uriBuilder.build().toURL();
        } catch(MalformedURLException | URISyntaxException e) {
            e.printStackTrace();
            throw new InvalidQueryException(query);
        }
    }

First, we make a check to see if the query is valid. If not, we’re going to throw that InvalidQuery exception from earlier. Additionally, we may throw a MalformedURLException or URISyntaxexception, both caused by an invalid query string, so when caught we shall throw a new InvalidQuery exception. Next, using a URIBuilder, we build our URL using some constants we specify as variables, and the query and scroll_cursor value we pass. With our initial queries, we will have a null scroll cursor, so we also check for that. Finally, we build the URI and return as a URL, so we can use it to open up an InputStream later on.

Lets implement our executeSearch function. This is where we actually call Twitter and parse its response.

public static TwitterResponse executeSearch(final URL url) {
        BufferedReader reader = null;
        try {
            reader = new BufferedReader(new InputStreamReader(url.openConnection().getInputStream()));
            Gson gson = new Gson();
            return gson.fromJson(reader, TwitterResponse.class);
        } catch(IOException e) {
            e.printStackTrace();
        } finally {
            try {
                reader.close();
            } catch(NullPointerException | IOException e) {
                e.printStackTrace();
            }
        }
        return null;
    }

This is a fairly simple method. All we’re doing is opening up a URLConnection for our Twitter query, then parsing that response using Gson as a TwitterResponse object, serializing the JSON into a Java object that we can use. As we’ve already implemented the logic earlier for using the scroll cursor, if we were to run this now, rather than the program terminating after a few seconds, it will keep running till there is no longer a valid response from Twitter. However, we haven’t quite finished yet as we have yet to extract any information from the tweets.

The TwitterResponse object is currently holding all the twitter data in it’s items_html variable, so what we now need to do is go back to TwitterResponse and add in some code that lets us extract that data. If you remember from earlier, we added a getTweets() method to the TwitterResponse object, however it’s returning an empty list. We’re going to fully implement that method so that when called, it builds up a list of tweets from the response inner_html.

To do this, we are going to be using JSoup, and we can even refer to some of those CSS queries that we noted earlier.

public List getTweets() {
    final List tweets = new ArrayList<>();
    Document doc = Jsoup.parse(items_html);
    for(Element el : doc.select("li.js-stream-item")) {
        String id = el.attr("data-item-id");
        String text = null;
        String userId = null;
        String userScreenName = null;
        String userName = null;
        Date createdAt = null;
        int retweets = 0;
        int favourites = 0;
        try {
            text = el.select("p.tweet-text").text();
        } catch (NullPointerException e) {
            e.printStackTrace();
        }
        try {
            userId = el.select("div.tweet").attr("data-user-id");
        } catch (NullPointerException e) {
            e.printStackTrace();
        }
        try {
            userName = el.select("div.tweet").attr("data-name");
        } catch (NullPointerException e) {
            e.printStackTrace();
        }
        try {
            userScreenName = el.select("div.tweet").attr("data-screen-name");
        } catch (NullPointerException e) {
            e.printStackTrace();
        }
        try {
            final String date = el.select("span._timestamp").