python-mongodb存储文件

来源:互联网 发布:上传小说的软件 编辑:程序博客网 时间:2024/05/29 03:02

python 版本为2.7

mongodb版本2.6.5


使用mongodb存储文件,可以使用两种方式,一种是像存储普通数据那样,将文件转化为二进制数据存入mongodb,另一种使用gridfs,咱们先来说说第一种

先读取文件内容,然后塞进bson.binary.Binary对象里,最后像平常那样写入数据库,是不是很简单呢,获取文件一样的简单,像平时那样查找数据,然后将二进制内容写入文件即可

#coding=utf-8'''Created on 2015-10-8@author: kwsy2015'''import pymongoimport bson.binaryfrom pymongo import MongoClientfrom cStringIO import StringIOdef insertFile():    client = MongoClient('localhost', 27017)    #获得一个database    db = client.MongoFile    #获得一个collection    coll = db.image    filename = 'F:/测试数据/hehe.jpg'.decode('utf-8')    with open (filename,'rb') as myimage:        content = StringIO(myimage.read())        coll.save(dict(        content= bson.binary.Binary(content.getvalue()),        filename = 'hehe.jpg'      ))        def getFile():    client = MongoClient('localhost', 27017)    #获得一个database    db = client.MongoFile    #获得一个collection    coll = db.image    data = coll.find_one({'filename':'hehe.jpg'})    out = open('F:/测试数据/test4.jpg'.decode('utf-8'),'wb')    out.write(data['content'])    out.close()getFile()
因为我的文件路径都带有中文,因此需要用utf-8解码,否则会报错的

使用上述方法存储小文件是很方便的,那么如果是大文件呢,可以使用gridfs

        gridfs会把文件分成若干块来存储,每一块的大小默认为256K,所以,如果是小文件,就不要用gridfs来存储了,不然会浪费空间的,gridfs是MongoDB之上的分布式文件系统,可以使用mongodb的分片和复制机制,因为Mongodb分配数据空间时以2GB为单位,所以gridfs不产生磁盘碎片。

        

#coding=utf-8'''Created on 2015-9-29@author: Administrator'''from pymongo import MongoClientfrom bson.objectid import ObjectIdfrom gridfs import *def insertFile():    client = MongoClient('localhost', 27017)    db = client.Pic    fs = GridFS(db, 'images')    with open ('F:/测试数据/hehe.jpg'.decode('utf-8'),'rb') as myimage:        data=myimage.read()                id = fs.put(data,filename='first')        print iddef getFile():    client = MongoClient('localhost', 27017)    db = client.Pic    fs = GridFS(db, 'images')    file = fs.get_version('first', 0)    data = file.read()    out = open('F:/测试数据/test3.jpg'.decode('utf-8'),'wb')    out.write(data)    out.close()def delFile():    client = MongoClient('localhost', 27017)    db = client.Pic    fs = GridFS(db, 'images')    fs.delete(ObjectId('560a531b0d4eae34a4edbfdd'))def listName():    client = MongoClient('localhost', 27017)    db = client.Pic    fs = GridFS(db, 'images')     print fs.list()   listName()
写入文件时,我们可以设置它的filename,如果多个文件使用同一个filename呢,我们在获取文件时可以使用get_version()函数,第一个参数是filename,第二个参数是版本,从0开始。

此外,我们还可以使用get(),函数,需传入文件的ObjectId

使用python操作gridfs总得来说是很方便的,毕竟所提供的函数就那么几个,稍微用心看看源码就没问题了

0 0
原创粉丝点击