コード例 #1
0
ファイル: app.py プロジェクト: pombredanne/github-timeline
def user(id, rdb, mongodb):
    id = id.lower()
    user = mongodb.users_stats.find_one({'_id': id})
    if not user:
        return abort(404)
    user['rank'] = {}
    langs = [lang for lang in user.get('contrib', {})]
    pipe = rdb.pipeline()
    if user.get('loc', {}).get('country') == 'China':
        for lang in langs:
            key = _format("country:{0}.lang:{1}:user".format(user.get('loc', {}).get('country', 'China'), lang))
            pipe.zrevrank(key, id)
        user['rank']['China'] = {lang: rank + 1 for lang, rank in zip(langs, pipe.execute()) if rank is not None}
    user['loc_zh'] = {key: translate(text) for key, text in user.get('loc', {}).items() if key in ['country', 'state', 'city']}
    for lang in langs:
        key = _format("lang:{0}:user".format(lang))
        pipe.zrevrank(key, id)
    user['rank']['World'] = {lang: rank + 1 for lang, rank in zip(langs, pipe.execute()) if rank is not None}
    return user
コード例 #2
0
ファイル: app.py プロジェクト: pombredanne/github-timeline
def _rank(lang, country, page, page_count, rdb, mongodb):
    key = _format("country:{0}.lang:{1}:user".format(country, lang))
    total = rdb.zcard(key)
    pages = total/page_count + (total % page_count and 1)
    users = rdb.zrevrange(key, page*page_count, (page+1)*page_count - 1)
    w_key = _format("lang:{0}:user".format(lang))
    pipe = rdb.pipeline()
    for u in users:
        pipe.zrevrank(w_key, u)
    w_ranks = pipe.execute()
    data = [mongodb.users_stats.find_one({'_id': u}, {'info': 1, 'contrib': 1}) for u in users]
    for i, (user, wr) in enumerate(zip(data, w_ranks)):
        user['rank'] = {country: page_count*page + i + 1, 'world': wr + 1}
    return {
        'pages': pages,
        'page': page,
        'page_count': page_count,
        'language': lang,
        'country': country,
        'data': data
    }