python 使用mongodb 数据库操作

做了个项目，需要用mongodb ，查了许多资料，现在整理一下，做个笔记。
1、安装mongodb再网上可以查到，这个就不做记录了（懒得打字）
2、python 使用mongodb
    mongodb 可以保存文件和文档，具体的结构请看上一篇博文。
    现在上代码
    python导入mongodb 的库
from pymongo import MongoClient #mongo客户端
from pymongo.errors import ConnectionFailure #错误类型
from gridfs import GridFS #文件储存
from gridfs.errors import FileExists #出错类型    建立一个类，再类里有连接和处理方法
    
class MongoDB(object): #建立mongo类（名字随便）

def connect(self): #连接到mongo的方法
try:
self.conn=MongoClient("127.0.0.1",27017) #连接到mongo
self.db=self.conn["crack"] #建立crack 数据库（名字随意），mongodb里面没有此库的话就新建
self.coll_set=self.db["report"] #建立report 集合，同上，（这个集合和下面的fs集合看需求选择，因本人都需要所以建立两个集合，存储大型文件的话需要下面的就够了）
self.fs=GridFS(self.db) #在crack 数据库里建立GridFS文件集合（可以这么理解吧，专业名词不知道怎么说）产生两个集合，fs.files,fs.chunks
   
    def store_file(self,file_obj,report,filename=""):#存储文件方法，file_obj是mongodb处理文件的类，因为mongo存如文件需要文件的MD5等数据
if not filename:
filename=file_obj.get_name()

#store_file

#print file_obj.file_path
#print file_obj.get_md5()
existing = self.db.fs.files.find_one({"md5":file_obj.get_md5()})#到crack数据库的fs.files文件集合找对应的数据
if existing:

if self.coll_set.find_one({"file_path":file_obj.file_path})["crack_status"] == "cracked":#到report集合找数据
return existing["_id"]

new = self.fs.new_file(filename=filename,
contenType=file_obj.get_content_type(),
md5=file_obj.get_md5())#fs.files文件集合建新文件数据

for chunk in file_obj.get_chunks():#fs.chunks文件集合存文件
new.write(chunk)

self.coll_set.insert({"file_id":new._id,"crack_status":"uncrack","file_path":file_obj.file_path,
"crack_info":"","md5":file_obj.get_md5(),"deep_info":report})#report集合插入自定义数据

try:
new.close()
#print "new file"
return new._id
except FileExists:
to_find={"md5":file_obj.get_md5()}
return self.db.fs.files.find_one(to_find)["_id"]

def find_report(self,md5):#查找report集合数据
file_report=self.coll_set.find_one({"md5":md5})
if not file_report:
return None
if file_report["crack_status"] == "uncrack":
return None
return file_report

def find_file(self,md5):#查找fs.files数据
existing = self.db.fs.files.file_one({"md5":md5})
if existing:
return existing["_id"]
else:
return None

def find_uncrack(self):#查找report集合里满足条件的数据
uncrack_list=[]
for i in self.coll_set.find({"crack_status":"uncrack"}):
if not i :
return None
uncrack_list.append(i)
return uncrack_list

def update_cracked(self,report,crack_info):#更新数据库数据
self.coll_set.update(report,{"$set":{"crack_status":"cracked"}})
self.coll_set.update({"file_id":report.get("file_id")},{"$set":{"crack_info":crack_info}})
print self.coll_set.find_one({"file_path":report.get("file_path")})

def update_test(self,report,info):#同上，测试使用
self.coll_set.update(report,{"$set":{"crack_info":info}})
print self.coll_set.find_one({"file_path":report.get("file_path")})

def run(self,file_path,report):#连接mongo
if not HAVE_MONGO:
print (
"Unable to import pymongo (install with "
"`pip install pymongo`)"
)
file_md5=""
self.connect()
file = File(file_path)#这个类是上文说到的文件处理类，这个就不贴出来了，上面需要的get_md5，get_name等方法都是再这个类里
if file.valid():
id=self.store_file(file,report)

self.conn.close()

def clear_files(self):#清空report集合所有数据
self.coll_set.remove()

def clear_file(self,report):#清空符合条件的数据
self.coll_set.remove(report)

def upload_file(self,md5,upload_path=os.getcwd()):#取出保存的文件
report=self.file_report(md5)
if not report:
return None
else:
with open(upload_path,"wb") as fd:
fd.write(self.fs.get(ObjectId(report["files_id"])).read())

#结合其他两篇博文慢慢摸索，就会明白怎么使用了