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')