#!/usr/bin/env python
# coding: utf-8

# # サナララツイート自動収集の基礎的検討(1)
# 
# このドキュメントは、[ぴょこりんクラスタ Advent Calendar 2015](http://www.adventar.org/calendars/960)のために書いたものです。
# 
# ## 要約
# * twitter公式が用意してくれた枠組みでサナララツイートを収集してみた
# * 神の用意したお試しサナララツイート収集は1週間限定だった
# * 別の方法でもっと古いツイートを収集する（神の作りしルールへの挑戦）
# * 負けた
#  
# ## はじめに
# 　皆さん師走はどうお過ごしでしょうか。なかなか忙しく、ツイッターに張り付いていられない日々を送っているのではないかと思います。自分が見ていないうちに興味のある内容のツイートが流れてしまった・・・なんてこともしばしば。実際のところ僕もすべてのサナララツイートを追うことが出来ず、日々心を痛めています。そこで、多くのサナララツイートを自動的に収集、僕に見やすいように可視化する枠組みを構築することを決意、今回はツイート収集に関する基礎的な検討を行いました。
# 　
# 
# ## 公式が提供している検索の仕様確認（お試し）
# 　とりあえずサクッと叩いてみます。以下のを参考に。
#  * [Python で Twitter API にアクセス](http://qiita.com/yubais/items/dd143fe608ccad8e9f85)
#  * [GET search/tweets](https://dev.twitter.com/rest/reference/get/search/tweets)

# In[1]:


from requests_oauthlib import OAuth1Session

CKey,CSecret, AToken, ASecret=open('./TWOauth','rb').read().strip().split('\n')
knatsuno=OAuth1Session(CKey,CSecret, AToken, ASecret)
query=u'"サナララ"'
uri = 'https://api.twitter.com/1.1/search/tweets.json?q=%s&count=100'%query
req = knatsuno.get(uri)
print uri


# 諸々のキーでOAuth認証キメて、上記URLをgetすると結果が返ってくる感じ。ちなみに認証しないと
# >{"errors":[{"code":215,"message":"Bad Authentication data."}]}  
# 
# って感じで返ってくる。
# 少しだけ注釈。
# * twitterから提供されるConsumerKey, ConsumerSecretなどなど計４つの文字列が各行に記述されたファイルが'./TWOauth'に置いてあるという前提で書いてます。ベタ書きダメ絶対！  
# * サナララの部分について、普通のクオテーション('')の中にさらにダブルクオテーション("")を入れていますが、これは完全一致するものを見つけてくるための検索クエリ用のアレです。
#  - サナララなら特に類似する全然関係ない単語ってあんまりないような気もするので、ここでは完全一致でなくても良さげ。
#  - 気を付けるべき例として、例えばu'椎名希未'で検索を書けると全然関係ない椎名さんも引っかかる。具体的には椎名ゆなさんって名前がひっかった（AV女優さんでした）。u'"椎名希未"'とすると、このような問題が解決する、という感じ。
# * count=100は100個ツイート取ってくるよ、の意味。デフォルトだと15個とのこと。
#  
# 中身はjson形式で格納されていて、以下のような感じで色々見れる。100個並べるとドキュメントが長くなりすぎてつらいので3個だけ。

# In[2]:


for i in req.json()[u'statuses'][:3]:
    print i[u'created_at']
    print i['user'][u'name'] + '(@' + i['user'][u'screen_name'] + u')さんは' + u'言いました。'
    print u'『'+i['text']+u' 』と。\n'


# 　と、こんな感じ。さて、100個取ったはずなので本当に100個あるのか確認してみましょう。

# In[3]:


len(req.json()[u'statuses'])


# なるほど78個、なるほど。  
# 
# 
# 　先ほどリンクを張った公式ドキュメントを再確認。クエリのパラメータとしてqやcountの他にツイートの時期を指定するuntilなんてのがあるのですが、そこの説明にこんな注意書きがありました。
# >Keep in mind that the search index has a 7-day limit.  In other words, no tweets will be found for a date older than one week.
# 
# 　過去一週間分しか検索出来ない仕様だと。使えねぇな。検索結果のうしろの方見てみましょう。

# In[4]:


for i in req.json()[u'statuses'][:-4:-1]:
    print i[u'created_at']
    print i['user'][u'name'] + '(@' + i['user'][u'screen_name'] + u')さんは' + u'言いました。'
    print u'『'+i['text']+u' 』と。\n'


# 　確かに一週間前っぽい。そうか、だいたいわかった。
#  
#  ## 神のシステムに挑む
#  
# 　神（twitter）によれば一週間しかダメと言うことでした。どこかで聞いたことがある話です。でも一般的にこういうのには抜け道があるはずです。例えばほら、[ここ](https://twitter.com/search-home)で検索するともっと昔のツイートが見つけられる。こんなの納得いかないよ。『観測できれば干渉できる』って[胡散臭い小動物的な何か](https://www.google.com/search?q=QB&safe=off&espv=2&biw=2560&bih=993&source=lnms&tbm=isch&sa=X&ved=0ahUKEwjhxaa6o8nJAhWHoYMKHa4MAf0Q_AUIBigB)も言ってました。僕は観測できれば何でも干渉できるとは思わないのですが、今回のケースはきっと何とかなります。  
# 
# ### 検索結果のスクレイピング
# 　サクッと落としてサクッとスクレイピングしてみましょう。これだけなら簡単。

# In[5]:


req2 = knatsuno.get(u'https://twitter.com/search?f=tweets&vertical=default&q=%s&src=typd'%query)


# In[6]:


from bs4 import BeautifulSoup as bs
import bs4
def strongstrip(s):
    if type(s)!=bs4.element.NavigableString and type(s)!=str:
        if len(s.contents)==0: s=''
        else: 
            s= s.contents[0]
            s=strongstrip(s)
    return s

t=bs(req2.content, "lxml").findAll("div", { "class" : "content" })
for text in t[1:4]:
    print text.findAll('a')[0]['href'].strip('/')
    print text.findAll('strong')[0].contents[0]
    print text.findAll('a')[1]['title']
    print ''.join([strongstrip(i) for i in text.findAll('p')[0]])
    print ''


# 先ほどと同じように上位３ツイートを拾ってみる。スクリーンネーム、大体似たものが取れてる。微妙に時刻が違うんだけどさっきのはGMTで今回表示されてるのはGMT-8だからちょっと注意。ちなみに取れたツイートの個数は、

# In[7]:


len(t)-1


# 　20個、さっきより得られた結果が減ってる。Web上で表示されるときは最初はちょこっと出てて、スクロールしていくと検索結果が増えてくからまぁこれくらいかもしれない。さすが神のシステム、隙が無いぜ。ただ、仕様上はこっちなら1週間縛りは無いはずなので、何とかここで出た検索結果を出さずに、古い結果だけを出すような検索クエリが投げられれば、古いツイートも手に入るはず。
#  
# ### あるツイートより過去のツイートを探す
# 
# 　多分上位20件が表示されるという話だから、それより古いツイートを検索するって機能があれば話は簡単。で、さっきの1週間限定のヤツにはそれがある。
#  >max_id(Optional): Returns results with an ID less than (that is, older than) or equal to the specified ID.
#  
#  これって実はブラウザ上での検索でも使えるんじゃないの？という話。[高度な検索](https://twitter.com/search-advanced)なんかを見てもそんな機能は見当たらないんだけど、もしもこれが使えたら、検索結果のツイートよりさらに古いツイートを検索するのを結果が出なくなるまで続ければかなり古くまでツイートを漁るコトが出きるはず。

# In[8]:


results=[]

t=bs(req2.content, "lxml").findAll("div", { "class" : "content" })
for i in range(10):
    for text in t[1:]:
        results.append([text.findAll('a')[0]['href'].strip('/'),text.findAll('strong')[0].contents[0], \
                        text.findAll('a')[1]['title'], ''.join([strongstrip(i) for i in text.findAll('p')[0]]), \
                        text.findAll('a')[1]['href'].split('/')[-1]])
    req3= knatsuno.get(u'https://twitter.com/search?f=tweets&q=%s max_id%%3A%s&src=typd'%(query,str(int(results[-1][-1])-1)))
    t=bs(req3.content, "lxml").findAll("div", { "class" : "content" })
    if len(t)==1:
        break

for i in results[-1][:len(results[-1])]:
    print i


# 表示してるのは得られたツイートの中で一番古いツイート。max_idを使ったクエリが投げれた・・・のはいいんだけど何で一番古いツイートが一週間以内なの・・・。個数も、

# In[10]:


len(results)


# たった66個、ブラウザで検索してるときのネットワークのやりとり見てるとなんとなくhttps://twitter.com/i/toast_poll?oldest_unread_id=0 にもgetしてるっぽくて怪しいんだけど今日は眠いからここまで。悔しい。