1. 程式人生 > >Python的MongoDB模組PyMongo

Python的MongoDB模組PyMongo

介面文件:
http://api.mongodb.com/python/current/migrate-to-pymongo3.html#pymongo-2-9
http://api.mongodb.com/python/current/api/pymongo/collation.html

優秀部落格:
https://www.cnblogs.com/zhouxuchen/p/5544227.html


1、Mongo的連線

from pymongo import MongoClient
from bson import ObjectId
from collections import Iterable

MONGO_CLIENT = MongoClient(host=host, port=port, username=conf.MONGO_USERNAME, password=conf.MONGO_PASSWORD, replicaSet=conf.portscan)
#replset的一個目的是保持叢集的健壯性,但是,因此需要通過合理的方式連線整個replset,才能在primary宕機的時候,自動啟用secondary

MONGO_DB = MONGO_CLIENT[conf.dataName]
MONGO_CONN = MONGO_DB[conf.dbName]

2、Mongo的資料插入

data_a = {"name":"Bob"}
data_b = {"name":"Alice"}
1、result = MONGO_CONN.insert(data_a)
print(result)
# 在MongoDB中,每條資料其實都有一個_id屬性來唯一標識,如果沒有顯式指明_id,MongoDB會自動產生一個ObjectId型別的_id屬性。
# insert()方法會在執行後返回_id值。
2、 result = MONGO_CONN.insert_one(data_a)
print(result)
print(result.inserted_id)
#返回的是InsertOneResult物件,我們可以呼叫其inserted_id屬性獲取_id。
3、 result = MONGO_CONN.insert([data_a, data_b])
print(result)
# 返回的結果是對應的_id的集合,執行結果:
# [ObjectId('5932a80115c2606a59e8a048'), ObjectId('5932a80115c2606a59e8a049')]
4、 result = MONGO_CONN.insert_many([data_a, data_b])
print(result)
print(result.inserted_ids)
# insert_many()方法返回的型別是InsertManyResult,呼叫inserted_ids屬性可以獲取插入資料的_id列表

官方推薦使用insert_one()和insert_many()方法將插入單條和多條記錄分開。

3、Mongo的資料查詢

result = MONGO_CONN.find_one({'name': 'Bob'})
print(type(result))
print(result)
# 在這裡我們查詢name為Mike的資料,它的返回結果是字典型別,執行結果:
# <class'dict'>
# {'_id': ObjectId('5932a80238c2606a59e8a050'), 'name': 'Bob'}
# 可以發現它多了一個_id屬性,這就是MongoDB在插入的過程中自動新增的。

# 我們也可以直接根據ObjectId來查詢,這裡需要使用bson庫裡面的ObjectId。
from bson.objectid import ObjectId
result = MONGO_CONN.find_one({'_id': ObjectId('5932a80238c2606a59e8a050')})
print(result)
# 其查詢結果依然是字典型別

results = MONGO_CONN.find({'school': "guangming"})
print(results)
for result in results:
    print(result)
# 執行結果:

# <pymongo.cursor.Cursor object at 0x1032d5128>
# {'_id': ObjectId('5932a80238c2606a59e8a050'), 'name': 'Bob', 'age': 20, 'school': 'guangming'}
# {'_id': ObjectId('593278c815c2602678bb2b8d'), 'name': 'Alice', 'age': 20, 'school': 'guangming'}
# 返回結果是Cursor型別,相當於一個生成器,我們需要遍歷取到所有的結果,每一個結果都是字典型別。

重點:
——比較查詢
$lt小於{'add_time': {'$lt': time.time()}}
$gt大於{'add_time': {'$gt': time.time()}}
$lte小於等於{'add_time': {'$lte': time.time()}}
$gte大於等於{'add_time': {'$gte': time.time()}}
$ne不等於{'add_time': {'$ne': time.time()}}
$in在範圍內{'add_time': {'$in': [datetime.datetime.now()-datetime.timedelta(days=1), datetime.datetime.now()]}}
$nin不在範圍內{'add_time': {'$nin': [datetime.datetime.now()-datetime.timedelta(days=1), datetime.datetime.now()]}}
$exists屬性是否存在{'add_time': {'$exists': True}}name屬性存在

eg:時間查詢
results = MONGO_CONN.find({'add_time':{'$gt': datetime.datetime.now(), '$lte': datetime.datetime.now()+datetime.timedelta(days=1)}})
# datetime.datetime.now()+datetime.timedelta(days=1)表示明天
# datetime.datetime.now()-datetime.timedelta(days=1)表示昨天

4、更新操作

#school欄位的修改
MONGO_CONN.update({'name': Bob}, {'$set': {'school': "yangguang"}})

#列表的新增
MONGO_CONN.update({'name': "Bob"}, {'$push': {'class': "English"}})

5、刪除操作

MONGO_CONN.conn.remove({'name': "Bob"})

注:當資料量大且操作資料庫頻繁時,建立合理索引

from pymongo import ASCENDING, DESCENDING
MONGO_CONN.create_index([("age", DESCENDING), (" weight", ASCENDING)])