如何使用Bottle框架和Mongoengine将图像文件上传到MongoDB?

时间:2014-05-12 03:40:51

标签: python mongodb bottle mongoengine

我需要使用mongoengine和Bottle框架将图像文件上传到MongoDB。 有我的python代码:

from bottle import Bottle, run, route, post, debug, request, template, response
from mongoengine import *


connect('imgtestdb')


app = Bottle()


class Image(Document):
    img_id = IntField()
    img_src = ImageField()


@app.route('/img/<image>')
def get_img(image):
    img = Image.objects(img_id=image)[0].img_src
    response.content_type = 'image/jpeg'

    return img


@app.route('/new')
def new_img_form():
    return template('new.tpl')


@app.post('/new')
def new_img():
    img_id = request.forms.get('id')
    img_src = request.files.get('upload')

    img = Image()
    img.img_id = img_id
    img.img_src.put(img_src, content_type = 'image/jpeg')
    img.save()



app.run(host='localhost', port=8080, debug=True, reloader=True)

模板:

<!DOCTYPE html>
<html>
<head>
    <title></title>
</head>
<body>
<form action="/new" method="post" enctype="multipart/form-data">
  Image ID: <input type="text" name="id" />
  Select a file: <input type="file" name="upload" />
  <input type="submit" value="Start upload" />
</form>
</body>
</html>

当我尝试上传图片时,会出错:

Traceback (most recent call last):
  File "/usr/local/lib/python3.4/dist-packages/mongoengine/fields.py", line 1311, in put
    img = Image.open(file_obj)
  File "/usr/lib/python3/dist-packages/PIL/Image.py", line 2000, in open
    prefix = fp.read(16)
AttributeError: 'FileUpload' object has no attribute 'read'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/usr/local/lib/python3.4/dist-packages/bottle.py", line 862, in _handle
    return route.call(**args)
  File "/usr/local/lib/python3.4/dist-packages/bottle.py", line 1728, in wrapper
    rv = callback(*a, **ka)
  File "./imgtest.py", line 38, in new_img
    img.img_src.put(img_src, content_type = 'image/jpeg')
  File "/usr/local/lib/python3.4/dist-packages/mongoengine/fields.py", line 1314, in put
    raise ValidationError('Invalid image: %s' % e)
mongoengine.errors.ValidationError: Invalid image: 'FileUpload' object has no attribute 'read'

是否可以将图像文件从Bottle请求上传到MongoDB?

我只是想保存文件:

@app.route('/upload', method='POST')
def do_upload():
    img_id = request.forms.get('imgid')
    upload = request.files.get('upload')

    upload.save('tmp/{0}'.format(img_id))

它返回错误:

ValueError('I/O operation on closed file',)

然后我在保存上传之前尝试打开文件:

@app.route('/upload', method='POST')
def do_upload():
    upload = request.files.get('upload')

    with open('tmp/1.jpg', 'w') as open_file:
        open_file.write(upload.file.read())

错误:

ValueError('read of closed file',)

我做错了什么?

2 个答案:

答案 0 :(得分:0)

首先,所有img_src都是类bottle.FileUpload的实例。它没有read()。基本上保存文件并再次打开。在下面做测试。祝你好运!

import os

from bottle import route, run, template, request
from mongoengine import Document, IntField, ImageField, connect

connect('bottle')

class Image(Document):
    img_id = IntField()
    img_src = ImageField()

@route('/home')
def home():
    return template('upload')

@route('/upload', method='POST')
def do_upload():
    img_id = request.forms.get('id')
    img_src = request.files.get('upload')
    name = img_src.filename

    # your class ...
    img = Image()
    img.img_id = img_id
    # img_src is class bottle.FileUpload it not have read()
    # you need save the file
    img_src.save('.') # http://bottlepy.org/docs/dev/_modules/bottle.html#FileUpload

    # mongoengine uses PIL to open and read the file
    # https://github.com/python-imaging/Pillow/blob/master/PIL/Image.py#L2103
    # open temp file
    f = open(img_src.filename, 'r')

    # saving in GridFS...
    img.img_src.put(f)
    img.save()

    return 'OK'

run(host='localhost', port=8080, debug=True)

答案 1 :(得分:0)

是的,如果img_src是一个bottle.FileUpload实例,它没有.read()方法

但是img_src.file是一个_io.BufferedRandom,所以它有.read()方法。

所以你可以直接做:

file_h = img_src.file
img = Image.open(file_h)

仍然,试图用Pillow 2.5 Image.open()打开它给了我一个&#34; ValueError:读取已关闭的文件&#34; (异常来自Image.open()和fp.read(16)调用)

将瓶子从0.12升级到0.12.7就可以了,这个例外就消失了。 希望它有所帮助。