from pymongo import MongoClient

def get_db():
    #建立連接
    client = MongoClient("localhost", 27017)
    #test,還有其他寫法
    db = client.test
    return db

def get_collection(db):
    #選擇集合(mongo中collection和database都是lazy創(chuàng)建的，具體可以google下)
    collection = db['posts']
    print collection

def insert_one_doc(db):
    #插入一個document
    posts = db.posts
    post = {"name":"lzz", "age":25, "weight":"55"}
    post_id = posts.insert(post)
    print post_id

def insert_mulit_docs(db):
    #批量插入documents,插入一個數組
    posts = db.posts
    post = [ {"name":"nine", "age":28, "weight":"55"},
                 {"name":"jack", "age":25, "weight":"55"}]
    obj_ids = posts.insert(post)
    print obj_ids

##查詢，可以對整個集合查詢，可以根ObjectId查詢，可以根據某個字段查詢等
def get_all_colls(db):
    #獲得一個數據庫中的所有集合名稱
    print db.collection_names()

def get_one_doc(db):
    #有就返回一個，沒有就返回None
    posts = db.posts
    print posts.find_one()
    print posts.find_one({"name":"jack"})
    print posts.find_one({"name":"None"})
    return

def get_one_by_id(db):
    #通過objectid來查找一個doc
    posts = db.posts
    obj = posts.find_one()
    obj_id = obj["_id"]
    print "_id 為ObjectId類型 :"
    print posts.find_one({"_id":obj_id})
    #需要注意這里的obj_id是一個對象，不是一個str，使用str類型作為_id的值無法找到記錄
    print "_id 為str類型 "
    print posts.find_one({"_id":str(obj_id)})

    #可以通過ObjectId方法把str轉成ObjectId類型
    from bson.objectid import ObjectId
    print "_id 轉換成ObjectId類型"
    print posts.find_one({"_id":ObjectId(str(obj_id))})

def get_many_docs(db):
    #mongo中提供了過濾查找的方法，可以通過各
    #種條件篩選來獲取數據集，還可以對數據進行計數，排序等處理
    posts = db.posts
    #所有數據,按年齡排序, -1是倒序
    all =  posts.find().sort("age", -1)

    count = posts.count()
    print "集合中所有數據 %s個"%int(count)
    for i in all:
        print i

    #條件查詢
    count = posts.find({"name":"lzz"}).count()
    print "lzz: %s"%count
    for i in  posts.find({"name":"lzz", "age":{"$lt":20}}):
        print i

def clear_coll_datas(db):
    #清空一個集合中的所有數據
    db.posts.remove({})

if __name__ == "__main__":
    db = get_db()
    obj_id = insert_one_doc(db)
    obj_ids = insert_mulit_docs(db)
    #get_all_colls(db)
    #get_one_doc(db)
    #get_one_by_id(db)
    #get_many_docs(db)
    clear_coll_datas(db)