#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()