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创建的，具体可以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)