Exemplo n.º 1
0
def _save_follower_relation_to_db(user, follower_list):
    follower_id_list = []
    refined_follower_list = []
    rejected_fields = ['domain','firstname','lastname','cover_url','thumbnail_background_url','avatars']

    for follower in follower_list:
        follower_id_list.append(follower['id'])
        if not user_manager.user_seen(follower['id']):
            user_manager.add_user(follower['id'])
            bfs_queue.put(follower['id'])
            for field in rejected_fields:
                follower.pop(field, None)
            refined_follower_list.append(follower)

    #save users into users collection
    user_collection = mydb.users
    if len(refined_follower_list)>0:
        user_collection.insert_many(refined_follower_list)

    #add the following relation of the given user into user relation collection
    user_relation_collection = mydb.user_relation
    relation = {}
    relation['user'] = user
    relation['followers'] = follower_id_list
    user_relation_collection.insert(relation)
Exemplo n.º 2
0
def test():
    user_manager.clear_all()
    print user_manager.user_seen(5)
    user_manager.add_user(5)
    print user_manager.user_seen(5)