#encoding=utf-8
'''

Mongo Conn連接類
'''

import pymongo

class DBConn:
    conn = None
    servers = "mongodb://localhost:27017"

    def connect(self):
        self.conn = pymongo.Connection(self.servers)

    def close(self):
        return self.conn.disconnect()

    def getConn(self):
        return self.conn

#encoding=utf-8
'''

Mongo操作Demo
Done:
'''
import MongoDBConn

dbconn = MongoDBConn.DBConn()
conn = None
lifeba_users = None

def process():
    #建立連接
    dbconn.connect()
    global conn
    conn = dbconn.getConn()

    #列出server_info信息
    print conn.server_info()

    #列出全部數(shù)據(jù)庫(kù)
    databases = conn.database_names()
    print databases

    #刪除庫(kù)和表
    dropTable()
    #添加數(shù)據(jù)庫(kù)lifeba及表(collections)users
    createTable()
    #插入數(shù)據(jù)
    insertDatas()
    #更新數(shù)據(jù)
    updateData()
    #查詢數(shù)據(jù)
    queryData()
    #刪除數(shù)據(jù)
    deleteData()

    #釋放連接
    dbconn.close()

def insertDatas():
    datas=[{"name":"steven1","realname":"測(cè)試1","age":25},
           {"name":"steven2","realname":"測(cè)試2","age":26},
           {"name":"steven1","realname":"測(cè)試3","age":23}]
    lifeba_users.insert(datas)

def updateData():
    '''只修改最后一條匹配到的數(shù)據(jù)
           第3個(gè)參數(shù)設(shè)置為True,沒找到該數(shù)據(jù)就添加一條
           第4個(gè)參數(shù)設(shè)置為True,有多條記錄就不更新
    '''
    lifeba_users.update({'name':'steven1'},{'$set':{'realname':'測(cè)試1修改'}}, False,False)

def deleteData():
    lifeba_users.remove({'name':'steven1'})

def queryData():
    #查詢?nèi)繑?shù)據(jù)
    rows = lifeba_users.find()
    printResult(rows)
    #查詢一個(gè)數(shù)據(jù)
    print lifeba_users.find_one()
    #帶條件查詢
    printResult(lifeba_users.find({'name':'steven2'}))
    printResult(lifeba_users.find({'name':{'$gt':25}}))

def createTable():
    '''創(chuàng)建庫(kù)和表'''
    global lifeba_users
    lifeba_users = conn.lifeba.users

def dropTable():
    '''刪除表'''
    global conn
    conn.drop_database("lifeba")

def printResult(rows):
    for row in rows:
        for key in row.keys():#遍歷字典
            print row[key], #加, 不換行打印
        print ''

if __name__ == '__main__':
    process()