Flask是支持文件上传的, 近来做了一个上传SQL文件的功能, SQL中会使用到中文, 泰文, 马来西亚文等多种语言,
我们通过Flask接收到文件, 先把文件保存在后端, 保存后却发现是乱码.
from flask import request
from werkzeug.utils import secure_filename
@app.route('/upload', methods=['GET', 'POST'])
def upload_file():
if request.method == 'POST':
f = request.files['the_file']
f.save('/var/www/uploads/' + secure_filename(f.filename))
...
通过上面一段代码接收并保存中文文件会乱码的.
如何解决呢?
首先考虑乱码是如何产生的呢, 两种不同的编码方式导致的.
通过Python的chardet模块的detect方法可以获取到字符串的编码方式
file = request.files.get('file')
sql_content = file.read()
encoding = chardet.detect(sql_content)['encoding']
我的项目中获取的编码方式是GB2312编码
解决编码错乱的问题就是先正确解码, 再重新编码, 本项目中是先用GB2312解码, 然后用UTF-8编码. 如下:
sql_content = sql_content.decode('GB2312').encode('utf-8')
上面重新编码后的sql_content的再重新写入文件对象, 然后保存就解决了中文乱码的问题.
file.truncate()
file.seek(0)
file.write(sql_content)
file.save('test.sql')
后台之间保存文件的乱码问题解决后, 我们通过MINIO的SDK上传的乱码问题也迎刃而解.
代码全览:
# 文件上传接口
class DBTaskSQLFile(Resource):
# 新建文件
def post(self):
file = request.files.get('file')
file.seek(0)
sql_content = file.read()
encoding = chardet.detect(sql_content)['encoding']
sql_content = sql_content.decode(encoding).encode('utf-8')
file.truncate()
file.seek(0)
file.write(sql_content)
file.seek(0)
# SQL文件上传至文件服务器
try:
pass
except Exception as err:
print('[MINIO]:', err)
return jsonify(code=1, msg='SQL文件上传至文件服务器时失败')
return jsonify(code=0, msg='Success')
