rubyでTwitterから画像urlを取得する

2015-08-15 Sat.

gem twitter を使用してツイートから画像の URL だけを抽出します。

app.rb
1require 'twitter'
2
3@client = Twitter::REST::Client.new do |config|
4  config.consumer_key = CONSUMER_KEY
5  config.consumer_secret = CONSUMER_SECRET
6  config.access_token = ACCESS_TOKEN
7  config.access_token_secret = ACCESS_TOKEN_SECRET
8end
9
10tweets = @client.home_timeline(count: 200)
11imgs = tweets.flat_map { |s| s.media}.map { |m| m.media_url.to_s}

以上のように media を取得するとpic.twitter.com の画像しか取得できないため、動画の判定処理に加え、url からも画像であるか判定します。今回は url が画像であるかどうかの判定に fastimage を使用します。 fastimage は本来画像のサイズを取得するためのものですが、(url の文字列からではなく)実際に url にアクセスしてファイル形式を判別できるため採用しました。

app.rb
1require 'twitter'
2require 'fastimage'
3
4# 省略
5
6imgs = []
7tweets = @client.home_timeline(count: 200)
8imgs.concat tweets.flat_map { |s| s.media }.flat_map { |m|
9              case m
10              when Twitter::Media::AnimatedGif
11                m.video_info.variants.map { |v| v.url.to_s }
12              when Twitter::Media::Photo
13                m.media_url.to_s
14              else
15                []
16              end
17            }
18imgs.concat tweets.flat_map { |s| s.urls }.flat_map { |u|
19              case FastImage.type(u.url.to_s)
20              when :bmp, :gif, :jpeg, :png, :webm
21                u.expanded_url.to_s
22              else
23                []
24              end
25            }

画像をファイルとして保存する場合は以下のようになります。

app.rb
1require 'open-uri'
2
3imgs.each do |url|
4  path = "./#{File.basename(url)}"
5  File.open(path, 'w') do |f|
6    f.write open(url).read
7  end
8end

url で動画を投稿している場合(vine とか)の処理は書いていません。画像だけでいいかな……