Python 操作 MongoDB

龙行于水 發表於 2016-9-4 03:06:00

<p>原文</p>
<p>这篇文章主要介绍了使用Python脚本操作MongoDB的教程,MongoDB作为非关系型数据库得到了很大的宣传力度,而市面上的教程一般都是讲解JavaScript的脚本操作,本文则是基于Python,需要的朋友可以参考下</p>
<h4 id="0启动mongodb数据库">0.启动mongodb数据库</h4>
<pre><code>$ mongod.exe --dbpath D:\mongodb\data\db
</code></pre>
<h4 id="1连接数据库">1.连接数据库</h4>
<p><strong>MongoClient VS Connection</strong></p>
<p>class MongoClient(pymongo.common.BaseObject)<br>
| Connection to MongoDB.<br>
|<br>
| Method resolution order:<br>
| MongoClient<br>
| pymongo.common.BaseObject<br>
| <strong>builtin</strong>.object<br>
|</p>
<p>class Connection(pymongo.mongo_client.MongoClient)<br>
| Connection to MongoDB.<br>
|<br>
| Method resolution order:<br>
| Connection<br>
| pymongo.mongo_client.MongoClient<br>
| pymongo.common.BaseObject<br>
| <strong>builtin</strong>.object</p>
<p>从这两个类的继承来看，connection是继承了MongoClient的，建议使用MongoClient而不是使用Connection。（也就是说，MongoClient可以使用方法Connection都可以使用）</p>
<pre><code class="language-python">from pymongo import MongoClient

client = MongoClient('192.168.40.87', 27037)
db_name = 'TCL_Useraction'
db = client
collection_useraction = db['useraction']
</code></pre>
<p>这里是通过字典的方式访问数据库和集合，同时你也可以通过.（点号）的方式访问</p>
<h4 id="2插入数据">2.插入数据</h4>
<p>save() VS insert()</p>
<p>mongodb的save和insert函数都可以向collection里插入数据，但两者是有两个区别：</p>
<p>一、save函数实际就是根据参数条件,调用了insert或update函数.如果想插入的数据对象存在,insert函数会报错,而save函数是改变原来的对象;如果想插入的对象不存在,那么它们执行相同的插入操作.这里可以用几个字来概括它们两的区别,即所谓"有则改之,无则加之".</p>
<p>二、insert可以一次性插入一个列表，而不用遍历，效率高， save则需要遍历列表，一个个插入。</p>
<h4 id="3更新数据">3.更新数据</h4>
<p>对于单个数据来说，可以更新后使用save方法</p>
<p>update(criteria, objNew, upsert, mult)</p>
<ul>
<li>criteria: 需要被更新的条件表达式</li>
<li>objNew: 更新表达式</li>
<li>upsert: 如目标记录不存在，是否插入新文档。</li>
<li>multi: 是否更新多个文档。</li>
</ul>
<pre><code class="language-python">collection_useraction.update({'gid':last_gid, 'time':l_date},
                        {'$set':{'gid':last_gid},
                           '$set':{'time':l_date},
                           '$addToSet':{'categories':category_data}},
                        upsert=True)
</code></pre>
<h4 id="4删除数据">4.删除数据</h4>
<pre><code class="language-python">db.users.drop() # 删除集合

# remove(self, spec_or_id=None, safe=None, multi=True, **kwargs)
# remove() 用于删除单个或全部文档，删除后的文档无法恢复。
id = db.users.find_one({"name":"user2"})["_id"]
db.users.remove(id) # 根据 id 删除一条记录
db.users.remove() # 删除集合里的所有记录
db.users.remove({'yy':5}) # 删除yy=5的记录
</code></pre>
<h4 id="5查询">5.查询</h4>
<pre><code class="language-python"># 查询 age 小于 15 的
for u in db.users.find({"age":{"$lt":15}}):
print(u)
</code></pre>
<h5 id="51-查询一条记录">5.1 查询一条记录</h5>
<pre><code class="language-python"># 查询 name 等于 user8 的
for u in db.users.find({"name":"user8"}):
print(u)

# 获取查询的一个
u2 = db.users.find_one({"name":"user9"}) # 查不到时返回 None
print(u2)
</code></pre>
<h5 id="52-查询特定键-fields">5.2 查询特定键 (fields)</h5>
<p><strong>特别说明</strong>：<br>
在3.0版本中，这个参数已经改名为<code>projection</code>，若用<code>fields</code>会报错</p>
<pre><code class="language-python"># select name, age from users where age = 21
for u in db.users.find({"age":21}, ["name", "age"]):
print(u)
for u in db.users.find(fields = ["name", "age"]):
print(u)
</code></pre>
<h5 id="53-排序sort">5.3 排序(SORT)</h5>
<pre><code class="language-python">pymongo.ASCENDING # 也可以用 1 来代替
pymongo.DESCENDING # 也可以用 -1 来代替

for u in db.users.find().sort([("age", pymongo.ASCENDING)]):
print(u)# select * from 集合名 order by 键1

for u in db.users.find().sort([("age", pymongo.DESCENDING)]):
print(u) # select * from 集合名 order by 键1 desc

for u in db.users.find().sort([("键1", pymongo.ASCENDING), ("键2", pymongo.DESCENDING)]):
print(u) # select * from 集合名 order by 键1 asc, 键2 desc

for u in db.users.find(sort = [("键1", pymongo.ASCENDING), ("键2", pymongo.DESCENDING)]):
print(u) # sort 的另一种写法

for u in db.users.find({"name":"user9"}, sort=[['name',1],['sex',1]], fields = ["name", "age", 'sex']):
print(u) # 组合写法
</code></pre>
<h5 id="54-从第几行开始读取slice读取多少行limit">5.4 从第几行开始读取(SLICE)，读取多少行(LIMIT)</h5>
<pre><code class="language-python"># select * from 集合名 skip 2 limit 3
# MySQL 的写法： select * from 集合名 limit 2, 3
for u in db.users.find().skip(2).limit(3):
print(u)
for u in db.users.find(skip = 2, limit = 3):
print(u)

# 可以用切片代替 skip & limit (mongo 中的 $slice 貌似有点问题)。
for u in db.users.find():
print(u)

# 单独的写
for u in db.users.find().skip(2):
print(u)
for u in db.users.find(skip=1):
print(u)
for u in db.users.find().limit(5):
print(u)
for u in db.users.find(limit = 3):
print(u)
</code></pre>
<h5 id="55-多条件查询conditional-operators--like-的可使用正则表达式查询">5.5 多条件查询(Conditional Operators) # like 的可使用正则表达式查询</h5>
<pre><code class="language-python"># select * from users where name = 'user3' and age > 12 and age < 15
for u in db.users.find({'age': {'$gt': 12, '$lt': 15}, 'name': 'user3'}):
print(u)
# select * from users where name = 'user1' and age = 21
for u in db.users.find({"age":21, "name":"user1"}):
print(u)
</code></pre>
<h5 id="56-in">5.6 IN</h5>
<pre><code class="language-python">for u in db.users.find({"age":{"$in":(23, 26, 32)}}):
print(u)# select * from users where age in (23, 26, 32)
for u in db.users.find({"age":{"$nin":(23, 26, 32)}}):
print(u) # select * from users where age not in (23, 26, 32)
</code></pre>
<h5 id="57-统计总数count">5.7 统计总数(COUNT)</h5>
<pre><code class="language-python">print(db.users.count()) # select count(*) from users
print(db.users.find({"age":{"$gt":30}}).count()) # select count(*) from users where age > 30
</code></pre>
<h5 id="58-or">5.8 OR</h5>
<pre><code class="language-python">for u in db.users.find({"$or":[{"age":25}, {"age":28}]}):
print(u) # select * from 集合名 where 键1 = 值1 or 键1 = 值2
for u in db.users.find({"$or":[{"age":{"$lte":23}}, {"age":{"$gte":33}}]}):
print(u) # select * from 集合名 where 键1 <= 值1 or 键1 >= 值2
</code></pre>
<h5 id="6-是否存在-exists">6. 是否存在 (exists)</h5>
<pre><code class="language-python">db.users.find({'sex':{'$exists':True}}) # select * from 集合名 where exists 键1
db.users.find({'sex':{'$exists':False}}) # select * from 集合名 where not exists 键1
</code></pre>
<h5 id="7-正则表达式查询">7. 正则表达式查询</h5>
<pre><code class="language-python">for u in db.users.find({"name" : {"$regex" : r"(?i)user"}}, ["name"]):
print(u) # 查询出 name 为 user1, user3, user5
</code></pre>
<h5 id="8-多级路径的元素值匹配">8. 多级路径的元素值匹配</h5>
<p>Document 采取 JSON-like 这种层级结构，因此我们可以直接用嵌入(Embed)代替传统关系型数据库的关联引用(Reference)。<br>
MongoDB 支持以 "." 分割的 namespace 路径，条件表达式中的多级路径须用引号</p>
<pre><code class="language-python"># 如果键里面包含数组，只需简单匹配数组属性是否包含该元素即可查询出来
db.集合名.find_one({'address':"address1"}) # address 是个数组，匹配时仅需包含有即可
# 查询结果如：{"_id" : ObjectId("4c479885089df9b53474170a"), "name" : "user1", "address" : ["address1", "address2"]}

# 条件表达式中的多级路径须用引号,以 "." 分割
u = db.集合名.find_one({"im.qq":12345678})
# 查询结果如：{"_id" : ObjectId("4c479885089df9b53474170a"), "name" : "user1", "im" : {"msn" : "", "qq" : 12345678}}

print(u)['im']['msn'] #显示：

# 多级路径的更新
db.集合名.update({"im.qq":12345678}, {'$set':{"im.qq":12345}})

# 查询包含特定键的
for u in db.users.find({"im.qq":{'$exists':True}}, {"im.qq":1}):
print(u)
# 显示如： { "_id" : ObjectId("4c479885089df9b53474170a"), "im" : { "qq" : 12345 } }

for u in db.users.find({'data':"abc"}):
print(u)
# 显示如： { "_id" : ObjectId("4c47a481b48cde79c6780df5"), "name" : "user8", "data" : [ { "a" : 1, "b" : 10 }, 3, "abc" ] }
for u in db.users.find({'data':{'$elemMatch':{'a':1, 'b':{'$gt':5}}}}):
print(u)
# 显示如： { "_id" : ObjectId("4c47a481b48cde79c6780df5"), "name" : "user8", "data" : [ { "a" : 1, "b" : 10 }, 3, "abc" ] }

{data:"abc"} 仅简单匹配数组属性是否包含该元素。$elemMatch 则可以处理更复杂的元素查找条件。当然也可以写成如下方式：
db.集合名.find({"data.a":1, "data.b":{'$gt':5}})

对数组, 还可以直接使用序号进行操作：
db.集合名.find({"data.1":3}) # 序号从0开始

# 如集合的一列内容
{"classifyid":"test1",
"keyword":[
{"name":'test1', # 将修改此值为 test5 (数组下标从0开始,下标也是用点)
"frequence":21,
},
{"name":'test2', # 子表的查询，会匹配到此值
"frequence":50,
},
]
}
# 子表的修改(子表的其它内容不变)
db.集合名.update({"classifyid":"test1"}, {"$set":{"keyword.0.name":'test5'}})
# 子表的查询
db.集合名.find({"classifyid":"test1", "keyword.0.name":"test2"})
</code></pre>
<h4 id="6操作">6.操作</h4>
<p>(1) <code>$all</code>: 判断数组属性是否包含全部条件。</p>
<pre><code class="language-python">db.users.insert({'name':"user3", 'data':})
db.users.insert({'name':"user4", 'data':})

for u in db.users.find({'data':{'$all':}}):
print(u)
# 显示： { "_id" : ObjectId("4c47a133b48cde79c6780df0"), "name" : "user3", "data" : [ 1, 2, 3, 4, 5, 6, 7 ] }
</code></pre>
<p>注意和 <code>$in</code> 的区别。<code>$in</code> 是检查目标属性值是条件表达式中的一员，而 <code>$all</code> 则要求属性值包含全部条件元素。</p>
<p>(2) <code>$size</code>: 匹配数组属性元素数量。</p>
<pre><code class="language-python">for u in db.users.find({'data':{'$size':3}}):
print(u)
# 只显示匹配此数组数量的： { "_id" : ObjectId("4c47a13bb48cde79c6780df1"), "name" : "user4", "data" : [ 1, 2, 3 ] }
</code></pre>
<p>(3) <code>$type</code>: 判断属性类型。</p>
<pre><code class="language-python">for u in db.users.find({'t':{'$type':1}}):
print(u) # 查询数字类型的
for u in db.users.find({'t':{'$type':2}}):
print(u) # 查询字符串类型的
</code></pre>
<p>类型值:<br>
double:1<br>
string: 2<br>
object: 3<br>
array: 4<br>
binary data: 5<br>
object id: 7<br>
boolean: 8<br>
date: 9<br>
null: 10<br>
regular expression: 11<br>
javascript code: 13<br>
symbol: 14<br>
javascript code with scope: 15<br>
32-bit integer: 16<br>
timestamp: 17<br>
64-bit integer: 18<br>
min key: 255<br>
max key: 127</p>
<p>(4) <code>$not</code>: 取反，表示返回条件不成立的文档。<br>
似乎只能跟正则和 <code>$mod</code> 一起使用？？？？<br>
还不知如何使用</p>
<p>(5) <code>$unset</code>: 和 <code>$set</code> 相反，表示移除文档属性。</p>
<pre><code class="language-python">for u in db.users.find({'name':"user1"}):
print(u)
# 显示如： { "_id" : ObjectId("4c479885089df9b53474170a"), "name" : "user1", "age" : 15, "address" : [ "address1", "address2" ] }

db.users.update({'name':"user1"}, {'$unset':{'address':1, 'age':1}})
for u in db.users.find({'name':"user1"}):
print(u)
# 显示如： { "_id" : ObjectId("4c479885089df9b53474170a"), "name" : "user1" }
</code></pre>
<p>(6) <code>$push</code>: 和 <code>$pushAll</code> 都是向数组属性添加元素。好像两者没啥区别</p>
<pre><code class="language-python">for u in db.users.find({'name':"user1"}):
print(u)
# 显示如： { "_id" : ObjectId("4c479885089df9b53474170a"), "age" : 15, "name" : "user1" }

db.users.update({'name':"user1"}, {'$push':{'data':1}})
for u in db.users.find({'name':"user1"}):
print(u)
# 显示如： { "_id" : ObjectId("4c479885089df9b53474170a"), "age" : 15, "data" : [ 1 ], "name" : "user1" }

db.users.update({'name':"user1"}, {'$pushAll':{'data':}})
for u in db.users.find({'name':"user1"}):
print(u)
# 显示如： { "_id" : ObjectId("4c479885089df9b53474170a"), "age" : 15, "data" : [ 1, 2, 3, 4, 5 ], "name" : "user1" }
</code></pre>
<p>(7) <code>$addToSet</code>: 和 <code>$push</code> 类似，不过仅在该元素不存在时才添加 (Set 表示不重复元素集合)。</p>
<pre><code class="language-python">db.users.update({'name':"user2"}, {'$unset':{'data':1}})
db.users.update({'name':"user2"}, {'$addToSet':{'data':1}})
db.users.update({'name':"user2"}, {'$addToSet':{'data':1}})
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： { "_id" : ObjectId("4c479896089df9b53474170b"), "data" : [ 1 ], "name" : "user2" }

db.users.update({'name':"user2"}, {'$push':{'data':1}})
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： { "_id" : ObjectId("4c479896089df9b53474170b"), "data" : [ 1, 1 ], "name" : "user2" }

要添加多个元素，使用 $each。
db.users.update({'name':"user2"}, {'$addToSet':{'data':{'$each':}}})
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： {'age': 12, '_id': ObjectId('4c479896089df9b53474170b'), 'data': , 'name': 'user2'}
# 貌似不会自动删除重复
</code></pre>
<p>(8) <code>$each</code> 添加多个元素用。</p>
<pre><code class="language-python">db.users.update({'name':"user2"}, {'$unset':{'data':1}})
db.users.update({'name':"user2"}, {'$addToSet':{'data':1}})
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： { "_id" : ObjectId("4c479896089df9b53474170b"), "data" : [ 1 ], "name" : "user2" }

db.users.update({'name':"user2"}, {'$addToSet':{'data':{'$each':}}})
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： {'age': 12, '_id': ObjectId('4c479896089df9b53474170b'), 'data': , 'name': 'user2'}

db.users.update({'name':"user2"}, {'$addToSet':{'data':}})
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： { "_id" : ObjectId("4c479896089df9b53474170b"), "data" : [ 1, 2, 3, 4, [ 1, 2, 3, 4 ] ], "name" : "user2" }

db.users.update({'name':"user2"}, {'$unset':{'data':1}})
db.users.update({'name':"user2"}, {'$addToSet':{'data':}})
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： { "_id" : ObjectId("4c47a133b48cde79c6780df0"), "data" : [ ], "name" : "user2" }
</code></pre>
<p>(9) <code>$pop</code>: 移除数组属性的元素(按数组下标移除)，<code>$pull</code> 按值移除，<code>$pullAll</code> 移除所有符合提交的元素。</p>
<pre><code class="language-python">db.users.update({'name':"user2"}, {'$unset':{'data':1}})
db.users.update({'name':"user2"}, {'$addToSet':{'data':{'$each':}}})
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： { "_id" : ObjectId("4c47a133b48cde79c6780df0"), "data" : [ 1, 2, 3, 4, 5, 6, 7, 2, 3 ], "name" : "user2" }

db.users.update({'name':"user2"}, {'$pop':{'data':1}}) # 移除最后一个元素
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： { "_id" : ObjectId("4c47a133b48cde79c6780df0"), "data" : [ 1, 2, 3, 4, 5, 6, 7, 2 ], "name" : "user2" }

db.users.update({'name':"user2"}, {'$pop':{'data':-1}}) # 移除第一个元素
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： { "_id" : ObjectId("4c47a133b48cde79c6780df0"), "data" : [ 2, 3, 4, 5, 6, 7, 2 ], "name" : "user2" }

db.users.update({'name':"user2"}, {'$pull':{'data':2}}) # 移除全部 2
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： { "_id" : ObjectId("4c47a133b48cde79c6780df0"), "data" : [ 3, 4, 5, 6, 7 ], "name" : "user2" }

db.users.update({'name':"user2"}, {'$pullAll':{'data':}}) # 移除 3,5,6
for u in db.users.find({'name':"user2"}):
print(u)
# 显示： { "_id" : ObjectId("4c47a133b48cde79c6780df0"), "data" : [ 4, 7 ], "name" : "user2" }
</code></pre>
<p>(10) <code>$where</code>: 用 JS 代码来代替有些丑陋的 <code>$lt</code>、<code>$gt</code>。<br>
MongoDB 内置了 Javascript Engine (SpiderMonkey)。可直接使用 JS Expression，甚至使用 JS Function 写更复杂的 Code Block。</p>
<pre><code class="language-python">db.users.remove() # 删除集合里的所有记录
for i in range(10):
db.users.insert({'name':"user" + str(i), 'age':i})
for u in db.users.find():
print(u)
</code></pre>
<p 9="">显示如下：<br>
{ "_id" : ObjectId("4c47b3372a9b2be866da226e"), "name" : "user0", "age" : 0 }<br>
{ "_id" : ObjectId("4c47b3372a9b2be866da226f"), "name" : "user1", "age" : 1 }<br>
{ "_id" : ObjectId("4c47b3372a9b2be866da2270"), "name" : "user2", "age" : 2 }<br>
{ "_id" : ObjectId("4c47b3372a9b2be866da2271"), "name" : "user3", "age" : 3 }<br>
{ "_id" : ObjectId("4c47b3372a9b2be866da2272"), "name" : "user4", "age" : 4 }<br>
{ "_id" : ObjectId("4c47b3372a9b2be866da2273"), "name" : "user5", "age" : 5 }<br>
{ "_id" : ObjectId("4c47b3372a9b2be866da2274"), "name" : "user6", "age" : 6 }<br>
{ "_id" : ObjectId("4c47b3372a9b2be866da2275"), "name" : "user7", "age" : 7 }<br>
{ "_id" : ObjectId("4c47b3372a9b2be866da2276"), "name" : "user8", "age" : 8 }</p>
<pre><code class="language-python">for u in db.users.find({"$where":"this.age > 7 || this.age < 3"}):
print(u)
</code></pre>
<p 9.0,="">显示如下：<br>
{'age': 0.0, '_id': ObjectId('4c47b3372a9b2be866da226e'), 'name': 'user0'}<br>
{'age': 1.0, '_id': ObjectId('4c47b3372a9b2be866da226f'), 'name': 'user1'}<br>
{'age': 2.0, '_id': ObjectId('4c47b3372a9b2be866da2270'), 'name': 'user2'}<br>
{'age': 8.0, '_id': ObjectId('4c47b3372a9b2be866da2276'), 'name': 'user8'}</p>
<pre><code class="language-python">for u in db.users.find().where("this.age > 7 || this.age < 3"):
print(u)
</code></pre>
<p 9.0,="">显示如下：<br>
{'age': 0.0, '_id': ObjectId('4c47b3372a9b2be866da226e'), 'name': 'user0'}<br>
{'age': 1.0, '_id': ObjectId('4c47b3372a9b2be866da226f'), 'name': 'user1'}<br>
{'age': 2.0, '_id': ObjectId('4c47b3372a9b2be866da2270'), 'name': 'user2'}<br>
{'age': 8.0, '_id': ObjectId('4c47b3372a9b2be866da2276'), 'name': 'user8'}</p>
<pre><code class="language-python"># 使用自定义的 function, javascript语法的
for u in db.users.find().where("function() { return this.age > 7 || this.age < 3;}"):
print(u)
</code></pre>
<p 9.0,="">显示如下：<br>
{'age': 0.0, '_id': ObjectId('4c47b3372a9b2be866da226e'), 'name': 'user0'}<br>
{'age': 1.0, '_id': ObjectId('4c47b3372a9b2be866da226f'), 'name': 'user1'}<br>
{'age': 2.0, '_id': ObjectId('4c47b3372a9b2be866da2270'), 'name': 'user2'}<br>
{'age': 8.0, '_id': ObjectId('4c47b3372a9b2be866da2276'), 'name': 'user8'}</p><br><br>
来源：https://www.cnblogs.com/hhh5460/p/5838516.html

頁: [1]

琼殿技术论坛's Archiver

Python 操作 MongoDB