root/lang/python/misc/early-hatena-bookmarkers/early-hatena-bookmarkers.py

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

import sys
import codecs
import feedparser
import re
import time

#sys.stdout = codecs.getwriter('utf-8')(sys.stdout)


# 定数
LIMIT = 100; # 結果表示するユーザ数の最大数
PAGE  = 5;   # 処理するページ数
ROW   = 20;  # 1ページあたりのブックマーク数
SLEEP = 3;   # ウェイト秒


# id をチェックし、不正なら 1 を返します
def check_id(id):

    regexp = re.compile('[a-zA-Z][-a-zA-Z0-9_]{1,30}[a-zA-Z0-9]$')

    if regexp.match(id):
        return
    else:
        return 1


# idのブックマークRSSページのURLをリストで返します
def get_my_rss(id):

    my_rss = []

    for p in range(PAGE):

        of = p * ROW + (p==0)
        my_rss.append( "http://b.hatena.ne.jp/" + id + "/rss?of=" + str(of) )

    return my_rss;


# RSSページからブックマークしたエントリーURLをリストで返します
def get_my_bookmarks(my_rss):

    my_bookmarks = []

    for rss_url in my_rss:

        feed = feedparser.parse(rss_url)

        for item in feed['entries']:
            my_bookmarks.append('http://b.hatena.ne.jp/entry/rss/' + item.link)

        time.sleep(SLEEP);

    return my_bookmarks;


# url がはてなのものでユーザを含む場合、ユーザ名を返します
def get_commenter_id_from_url(url):

    regexp = re.compile('http://b\.hatena\.ne\.jp/([^/]+)/')

    m = regexp.match(url)

    if m:
        return m.group(1)
    else:
        return


# target(正規表現) が url にマッチすれば 1 を返します
def check_target_in_url(url, target):

    regexp = re.compile(target)

    m = regexp.match(url)

    if m:
        return 1
    else:
        return


# 各ブックマークページのブックマーカーから
# 自分(id)よりも早くブックマークしているユーザをカウントします
def get_commenters(id, bookmarks):

    users = {};
    for url in bookmarks:

        feed = feedparser.parse(url)

        me = 0

        for item in feed['entries']:

            link = item.link

            if not link:
                continue

            if me:

                user = get_commenter_id_from_url(link)

                if user:
                    if users.has_key(user):
                        users[user] += 1
                    else:
                        users[user] = 1

            else:

                if check_target_in_url(link, 'http://b\.hatena\.ne\.jp/' + id + "/"):
                    me = 1

        time.sleep(SLEEP)

    return users;


#--- メインルーチン


# ID入力とチェック
print "input hatena id, and push enter"
id = raw_input()

if check_id( id ):
    print "ID [" + id + "] is wrong!"
    sys.exit(1)

# 集計
users = get_commenters(id, get_my_bookmarks( get_my_rss(id) ) )

# 結果表示
count = 1
for key,value in sorted(users.items(), key=lambda (k, v): (v, k), reverse=True):
    if count < LIMIT + 1:
        print "%d : %d : %s http://b.hatena.ne.jp/%s/" %(count , value, key, key)
    count += 1



'''
    early-hatena-bookmarkers.py - お気に入りをふやそう！

    DESCRIPTION

    このスクリプトは、あるIDがブックマークしているエントリーを
    IDよりも早くブックマークしているユーザを集計表示します。

    SYNOPSIS

    # python early-hatena-bookmarkers.py

    上記コマンドで実行すると、IDを入力してください。
    結果が表示されるまでにはかなり時間がかかりますので、気長にお待ちください。

    SEE ALSO

    http://coderepos.org/share/browser/lang/perl/misc/early-hatena-bookmarkers

    http://b.hatena.ne.jp/

    AUTHOR

    id:bayashi_net

    LICENSE

    This module is free software; you can redistribute it and/or
    modify it under the same terms as Perl itself. See perlartistic.
'''