mongo_tags.py 文件源码

def test_keywords():
    client = MongoClient(settings.MONGO_HOST, settings.MONGO_PORT)
    db = client[settings.MONGO_DATABASE]

    cursor = db.arch.find().limit(10)
    for post in cursor:
        id = post['_id']
        title = post['title']
        tf_idf_dict = util.file2dict(
            settings.TFIDF_FILE_PATH, str(id) + '.txt')
        util.print_message(title.encode('gbk'))
        for item in sorted(tf_idf_dict.items(), key=operator.itemgetter(1), reverse=True)[:10]:
            util.print_message('{0}:{1}'.format(item[0].decode(
                'utf8').encode('gbk'), item[1]))