flask-admin fileadmin 上传文件,中文名的解决方案 重写部分secure_filename

class upload_view(FileAdmin):
    def _save_form_files(self, directory, path, form):
        super()
        filename = self._separator.join([directory, form.upload.data.filename])

        if self.storage.path_exists(filename):
            secure_name = self._separator.join([path, form.upload.data.filename])
            raise Exception(gettext('File "%(name)s" already exists.',
                                    name=secure_name))
        else:
            self.save_file(filename, form.upload.data)
            self.on_file_upload(directory, path, filename)

    @expose('/rename/', methods=('GET', 'POST'))
    def rename(self):
        """
            Rename view method
        """
        form = self.name_form()

        path = form.path.data
        if path:
            base_path, full_path, path = self._normalize_path(path)

            return_url = self._get_dir_url('.index_view', op.dirname(path))
        else:
            return redirect(self.get_url('.index_view'))

        if not self.can_rename:
            flash(gettext('Renaming is disabled.'), 'error')
            return redirect(return_url)

        if not self.is_accessible_path(path):
            flash(gettext('Permission denied.'), 'error')
            return redirect(self._get_dir_url('.index_view'))

        if not self.storage.path_exists(full_path):
            flash(gettext('Path does not exist.'), 'error')
            return redirect(return_url)

        if self.validate_form(form):
            try:
                dir_base = op.dirname(full_path)
                filename = form.name.data
                self.storage.rename_path(full_path, self._separator.join([dir_base, filename]))
                self.on_rename(full_path, dir_base, filename)
                flash(gettext('Successfully renamed "%(src)s" to "%(dst)s"',
                              src=op.basename(path),
                              dst=filename), 'success')
            except Exception as ex:
                flash(gettext('Failed to rename: %(error)s', error=ex), 'error')

            return redirect(return_url)
        else:
            helpers.flash_errors(form, message='Failed to rename: %(error)s')

        if self.rename_modal and request.args.get('modal'):
            template = self.rename_modal_template
        else:
            template = self.rename_template

        return self.render(template, form=form, path=op.dirname(path),
                           name=op.basename(path), dir_url=return_url,
                           header_text=gettext('Rename %(name)s',
                                               name=op.basename(path)))

就是重写FileAdmin中的一个方法和一个视图。

把secure_filename去掉了,虽然不安全,但是目前也没什么快速的方法了。

见到另一个解决方案是把汉字转拼音。使用pypinyin库。

把上传的文件名分解,汉字部分转拼音。字符继续保存。

然后数据库中建立对应关系。

还需要改下载的视图。

麻烦,留着以后加吧。

猜你喜欢

转载自www.cnblogs.com/jackadam/p/9711233.html