mongo_tags.py 文件源码

def update_keywords():
    util.print_message('Start updating keywords...', debug=True)
    client = MongoClient(settings.MONGO_HOST, settings.MONGO_PORT)
    db = client[settings.MONGO_DATABASE]

    cursor = db.arch.find()
    for post in cursor:
        id = post['_id']
        tf_idf_dict = util.file2dict(settings.TFIDF_FILE_PATH, str(id) + '.txt')
        tags = []
        for item in sorted(tf_idf_dict.items(), key=operator.itemgetter(1), reverse=True)[:20]:
            tags.append(item[0])
        util.print_message(' '.join(tags))
        db.arch.update_one({'_id': id}, {'$set': {'tags': tags}})