aoshiman.org

留意事項

• 完全にオレオレクライアントになりますので、OAuth用consumer_key、secret（及びaccess_token、secret）は御自分で発行して下さい。 私の場合は、tweepyに付属しているgetaccesstoken.pyというtoken発行のサンプルを使用しています。すごく手軽なので使ってみて下さい。 consumer_key、secret等の発行方法はこちらの方々がとっても詳しいです。参考にさせて頂きました。ありがとうございます！
  • TwitterのOAuthを使って90日以上発言してない人をremoveするスクリプト書いた - YAMAGUCHI::weblog
  • Twitter API を OAuth で認証するスクリプトを 0 から書いてみた - trial and error
  
  
• access_token、secret等はpitを使って登録、管理しています。pitに関する素敵な解説はこちら
Pythonでの、Pitの使い方　世界一わかりやすいムービー付き - a2c.get.diary

使用方法

• scriptと同じ場所等にdbを作成します。db名はtwit.dbでtable名はtwitlogです。内容はこんな感じでお願いします。
  

  $ sqlite3 twit.db
  sqlite> .schema twitlog
  CREATE TABLE twitlog
                  (id INTEGER PRIMARY KEY,
                  date INTEGER,
                  repid INTEGER,
                  screen TEXT, body TEXT);
  sqlite> 
  

  
  
• 使用方法は殆どオリジナルと変更ありません。起動後、取得したい情報のkeyや発言内容を入力してリターンキーを押して下さい
• in_reply_to_id付きreplyは、replyしたい内容の番号を入力してから @screen_name のように入力します（下図ご参照）

#!/usr/bin/env python
# coding: utf-8
# thanks yoshiori: http://d.hatena.ne.jp/jYoshiori/20081201/1228123623
# thanks a2c: http://d.hatena.ne.jp/a2c/20081221/1229799599

import readline
from pit import Pit
from datetime import timedelta
import sqlite3

# KeyConfig
getFriendsTimeLineKey = ['friends', 'f']
getRepliesKey = ['res', 'r', '']
sendPostKey = ['post', 'p']
searchKey = ['search', 's']
exitKey = ['x', 'ZZ', 'exit', 'bye']

readline.parse_and_bind('tab: complete')

friends = set()

def complete(text, status):
    results = [x for x in friends if x.startswith(text)] + [None]
    return results[status]

def date_jst(utf_date):
    # UTC -> JST
    d = utf_date + timedelta(hours=9)
    return d

def twitPost(input):
    # Post twitterr
    api.update_status(input)
    print 'update : ' + input

def select_repid(num):
    con = sqlite3.connect(db_name)
    sql = 'SELECT repid FROM twitlog WHERE id = ?'
    stid = con.execute(sql, (num,)).fetchone()[0] 
    con.close()
    return stid
    
def twitRepost(input, stid):
    # Repost twitterr
    api.update_status(input, stid)
    print 'update : ' + input
    
def getReplies(maxid):
    # Get twitter friends Replies
    insert_data(api.mentions(count=maxid))
    con = sqlite3.connect(db_name)
    cur = con.cursor()
    cur.execute('select * from twitlog')

    print '\n\ttwitter replies\t'
    for row in cur:
        print ('%s :%s : %s %s' % (row[0], row[1], row[3], row[4]))
        friends.add(row[3])
    cur.close()
    con.close()

def getFriensTimeLine(maxid):
    # Get FriendsTimeLine
    insert_data(api.friends_timeline(count=maxid))
    con = sqlite3.connect(db_name)
    cur = con.cursor()
    cur.execute('select * from twitlog')

    print '\n  -----  twitter Friends Time Line  -----'
    for row in cur:
        print ('%s :%s : %s %s' % (row[0], row[1], row[3], row[4]))
        friends.add(row[3])
    cur.close()
    con.close()

def twitSearch(word):
    # Twitter Search
    insert_search_data(api.search(word))
    con = sqlite3.connect(db_name)
    cur = con.cursor()
    cur.execute('select * from twitlog')

    print '\n  -----  twitter search  -----'
    for row in cur:
        print ('%s :%s : %s %s' % (row[0], row[1], row[3], row[4]))
        friends.add(row[3])
    cur.close()
    con.close()

def insert_data(api):
    sql = 'insert into twitlog values(null,?,?,?,?)' 
    con = sqlite3.connect(db_name)
    con.execute('delete from twitlog')       
    for data in reversed(api):
        con.execute(sql, (date_jst(data.created_at), data.id, data.user.screen_name, data.text))
    con.commit()
    con.close()

def insert_search_data(api):
    sql = 'insert into twitlog values(null,?,?,?,?)' 
    con = sqlite3.connect(db_name)
    con.execute('delete from twitlog')       
    for data in reversed(api):
        con.execute(sql, (date_jst(data.created_at), data.id, data.from_user, data.text))
    con.commit()
    con.close()

def create_twitdb():
    # DBがぶっ壊れた時用
    db = sqlite3.connect(db_name)
    sql = (
            '''CREATE TABLE twitlog
                (id INTEGER PRIMARY KEY,
                date INTEGER,
                repid INTEGER,
                screen TEXT, body TEXT);'''
          )
    db.execute(sql)
    db.commit()
    db.close()

if __name__ == "__main__":
    db_name = 'twit.db'
   # create_twitdb()
    try:
        import tweepy
    except:
        print "error:\tcan't import tweepy module, you must install the module:\n"
        print "\tto install: 'easy_install tweepy'\n"
        exit()
    # OAuth setting
    consumer_config = Pit.get('twoauth')
    auth = tweepy.OAuthHandler(consumer_config['consumer_key'], consumer_config['consumer_secret'])
    access_config = Pit.get('access_key')
    auth.set_access_token(access_config['access_token'], access_config['access_secret'])

    readline.set_completer(complete)
    prompt = '\n cmd: Friendstimeline[f] Search[s] Post[p] Repost[num + @] Replies[r] Exit[x] \n> '
    maxid = 20
    api = tweepy.API(auth)

    while True:
        all_input = raw_input(prompt)
        input = all_input.split(" ")
        if input[0] != '':
            if input[0] in getFriendsTimeLineKey: 
                getFriensTimeLine(maxid)
                continue
            elif input[0] in sendPostKey:
                twitPost(unicode(' '.join(input[1:]), 'utf-8'))
                continue
            elif input[0].isdigit():
                sid = select_repid(input[0])
                #print sid
                twitRepost(unicode(' '.join(input[1:]), 'utf-8'), sid)
                continue
            elif input[0] in getRepliesKey:
                getReplies(maxid)
                continue
            elif input[0] in searchKey:
                twitSearch(' '.join(input[1:]))
                continue
            elif input[0] in exitKey:
                print 'bye ;-)'
                break
            else:
                twitPost(unicode('%s'%(all_input), 'utf-8'))
                continue
        else: 
            getReplies(maxid)