Mongodb可以使用BSON格式来保存大小不超过16M的二进制文件,很适合用来存放web中的图片资源,记录一下操作BSON格式的方法
本地文件写入Pymongo/Pymongo数据写回本地文件
"""
@File : file_demo.py
@Description : 本地文件写入Pymongo/Pymongo数据写回本地文件
@Time : 2020/4/7 23:20
@Author : DexterLien
@Email : [email protected]
@Software : PyCharm
"""
import os
from pymongo import MongoClient
# 连接带有密码认证的Mongodb,格式如下:
# mongodb://用户名:密码@服务器地址/数据库名
con = MongoClient('mongodb://username:[email protected]:8017/dexter')
# 使用名为dexter的数据库
db = con['dexter']
# 使用名为users的collection
users = db['users']
# 本地文件以BSON二进制写入数据库集合
fpath = os.path.join(os.getcwd(), 'pic.png')
with open(fpath, 'rb') as file:
users.insert_one({
'name': 'pic.png',
'data': file.read()
})
# 数据库中的BSON数据写回文件
ret = users.find_one({'name': 'pic.png'})
with open(os.path.join(os.getcwd(), 'new.png'), 'wb') as file:
file.write(ret.get('data'))
POST上传的图片附件写入Mongodb与返回
用Flask随便撸了一个demo,记录一下主要的思路
"""
@File : post_pic.py
@Description : POST上传的图片附件写入Mongodb与返回
@Time : 2020/4/7 23:59
@Author : DexterLien
@Email : [email protected]
@Software : PyCharm
"""
from flask import Flask, request, render_template, jsonify, make_response
from pymongo import MongoClient
from werkzeug.datastructures import FileStorage
app = Flask(__name__)
app.config['JSON_AS_ASCII'] = False
def save_to_mongo(file: FileStorage):
"""
将post请求中的文件对象保存至Mongodb,主要是通过file.stream.read()方法直接将二进制数据读出来,给Mongodb插入用
:param file:
:return:
"""
con = MongoClient('mongodb://username:[email protected]:8017/dexter')
db = con['dexter']
users = db['users']
users.insert_one({
'name': file.filename,
'data': file.stream.read()
})
@app.route('/', methods=['GET', 'POST'])
def index():
if request.method == 'GET':
return render_template('index.html')
if request.method == 'POST':
file = request.files.get('pic')
save_to_mongo(file)
return jsonify({
'code': 0,
'msg': '上传成功'
})
@app.route('/pic/<filename>', methods=['GET'])
def pic(filename):
con = MongoClient('mongodb://username:[email protected]:8017/dexter')
db = con['dexter']
users = db['users']
data = users.find_one({'name': filename})['data']
# 使用Mongodb查询到的BSON数据生成response响应,后面通过设置不同的headers实现不同响应行为
response = make_response(data)
# 在浏览器中直接显示图片
response.headers.set('Content-Type', 'image/png')
# 以文件下载方式返回
# response.headers.set(
# 'Content-Disposition', 'attachement', filename=f'{filename}'
# )
return response
if __name__ == '__main__':
app.run(debug=True)
上传操作前端页面
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<body>
{# 注意:Flask中需要指定form的enctype才能将文件封装发送到post请求中 #}
<form method="post" enctype=multipart/form-data>
<input name="pic" type="file">
<button type="submit">上传</button>
</form>
</body>
</html>
使用Flask直接将图片二进制数据传给jinja2模板渲染
主要是使用base64.b64encode()先将Mongodb中的二进制数据进行Base64编码,然后再对其使用utf-8解码,这样就可以得到图像的Base64类型的字符串数据
import base64
# 省略其他代码
@bp.route('/users')
def users_view():
_users = users.find({})
data = [u for u in _users]
for d in data:
d['mgcz'] = base64.b64encode(d['mgcz']).decode('utf-8')
return render_template('users.html', users=data)
jinja2模板中使用<img src="data:;base64,{{ user.mgcz | safe }}">这样格式的标签来渲染Base64格式字符串数据的图片
<div>
<ul>
{% for user in users %}
<li>
{{ user.xm }}
<img src="data:;base64,{{ user.mgcz | safe }}">
</li>
{% endfor %}
</ul>
</div>
与传统文件方式存储图片占用空间对比
做了一个简单的测试:
传统文件方式存放的图片new.webp文件实际大小:
使用BSON类型向MongoDB中写入该图片文件的前后服务器上数据库文件大小对比(单位byte字节)
lpwm@python-server:~/mongodb$ sudo du -sb
336945978 .
lpwm@python-server:~/mongodb$ sudo du -sb
336947156
相减后服务器上MongoDB存储这个图片数据只用了1,178个字节,比传统文件方式占用的60,362字节大小减少了98%,简直太叼了有木有!!!感觉这样对比是不是有点不准确呢?