init_db.py 文件源码

def add_share_category():
    collection = Connection.get_database(pymongo=True).share_category

    category_list = [
        u'??', u'??', u'??', u'??', u'??', u'??', u'??', u'??']

    for i, category in enumerate(category_list):
        document = {
            'name': category,
            'sort': i
        }

        existed = collection.find_one({"name": category})
        if not existed:
            collection.insert(document)

    raise gen.Return()