Django在保存图像之前获取ImageField路径

时间:2012-06-08 13:15:37

标签: django django-models

我正在尝试使用chunk方法保存图像,以处理用户尝试上传大图像的情况:

destination = open(incident.Image.path, 'wb+')
for chunk in request.FILES['image'].chunks():
    destination.write(chunk)
destination.close()

我的问题是,如果没有先在Image字段中保存一些内容,我就无法获取文件路径,如下所示:

fileName = str(int(time.time() * 1000))
imageName = fileName + '.jpg'
incident.Image.save(imageName, request.FILES['image'])

我的问题是如何在不首先保存图像的情况下获得相同的incident.Image.path?我不想对路径进行硬编码,而且我被卡住的原因是因为我无法获得完整的文件路径而没有得到ImageField声明的upload_to部分

修改

好的,我已经走得更远,但我再次陷入困境:

imgBuffer = StringIO.StringIO()
for chunk in request.FILES['image'].chunks():
    imgBuffer.write(chunk)

# rotate it
rotate = request.POST['rotation']
im = Image.open(imgBuffer)
im = im.rotate(float(rotate))

incident.Image.save(imageName, ContentFile(imgBuffer))

我收到IOError无法识别im = Image.open(imgBuffer)引发的图像文件,有什么想法吗?

3 个答案:

答案 0 :(得分:0)

看看stringIO还看看this site

答案 1 :(得分:0)

图像(作为文件)保存在模型字段的 pre_save 方法中。这是一个这样的自定义模型字段的工作示例(来自我的项目) - FixedImageField

您可以看到我如何在 pre_save 函数中检索图像路径。

在我的情况下,我有一个复杂的方案(我必须继承图像字段),但你可以根据你的情况简化它。

您可以实现自己的_save_fixed_resolution_image(块,路径) - 在我的情况下,它会将图像转换为最大尺寸为400x400的图像:

from PIL import Image
import StringIO
from yourproject.settings import MEDIA_ROOT
import os
from django.db import models

class ChunksImageField( models.ImageField ):    
    def pre_save( self, model_instance, add ):
        file = super( models.FileField, self ).pre_save(model_instance, add)
        if file and not file._committed:
            if callable( self.upload_to ):
                path = self.upload_to( model_instance, "" )
            else:
                path = self.upload_to
            file.name = path
            full_path = os.path.join( MEDIA_ROOT, path )
            chunks = _get_chunks( file.chunks() )
            self._save_image_func( model_instance, chunks, full_path )

        return file

class FixedImageField( ChunksImageField ):
    def __init__( self, *args, **kwargs ):
        super( FixedImageField, self ).__init__( *args, **kwargs )

    def _save_image_func( self, model_instance, chunks, path ):
        _save_fixed_resolution_image( chunks, path )

def _save_fixed_resolution_image( chunks, out_file_path ):
    image = _get_image( chunks )

    if image.size[ 0 ] > _image_max_size or image.size[ 1 ] > _image_max_size:
        image.thumbnail( ( _image_max_size, _image_max_size, ) )

    save_image( image, out_file_path )

def save_image( image, out_file_path ):
    image.save( out_file_path, "JPEG", quality = 100 )

def _get_chunks( chunks ):
    chunks_ = ""
    for chunk in chunks:
        chunks_ += chunk
    return chunks_

如何在您的模型中使用它:

class YourModel( models.Model ):
    image = FixedImageField( upload_to = image_path ) # substitute your image path here

答案 2 :(得分:0)

我解决了以下主要问题:

myimage = MyImageModel()
upload_to = myimage.myimagefield.field.upload_to
filepath = os.path.join(settings.MEDIA_ROOT, upload_to)

不上传两次的第二个问题是,我做了如下操作:

filename = os.path.basename(url)
myimage.myimagefield = os.path.join(upload_to, filename)
myimage.save()