将文件从一个模型复制到另一个模型

时间:2011-04-23 14:38:13

标签: django copy filefield

我有两个简单的模型:

class UploadImage(models.Model):
   Image = models.ImageField(upload_to="temp/")

class RealImage(models.Model):
   Image = models.ImageField(upload_to="real/")

一种形式

class RealImageForm(ModelForm):
    class Meta:
        model = RealImage 

我需要将文件从UploadImage保存到RealImage中。我怎么能这样做 下面的代码不起作用

realform.Image=UploadImage.objects.get(id=image_id).Image 
realform.save()

Tnx寻求帮助。

5 个答案:

答案 0 :(得分:28)

受Gerard解决方案的启发,我想出了以下代码:

from django.core.files.base import ContentFile

#...

class Example(models.Model):
    file = models.FileField()

    def duplicate(self):
        """
        Duplicating this object including copying the file
        """
        new_example = Example()
        new_file = ContentFile(self.file.read())
        new_file.name = self.file.name
        new_example.file = new_file
        new_example.save()

这实际上会通过在文件名中添加“_1”来重命名文件,以便原始文件和文件的新副本可以同时存在于磁盘上。

答案 1 :(得分:5)

虽然这已经很晚了,但我会解决这个问题,

class UploadImage(models.Model):
    Image = models.ImageField(upload_to="temp/")

    # i need to delete the temp uploaded file from the file system when i delete this model      
    # from the database
    def delete(self, using=None):
        name = self.Image.name
        # i ensure that the database record is deleted first before deleting the uploaded 
        # file from the filesystem.
        super(UploadImage, self).delete(using)
        self.Image.storage.delete(name)


class RealImage(models.Model):
   Image = models.ImageField(upload_to="real/")


# in my view or where ever I want to do the copying i'll do this
import os
from django.core.files import File

uploaded_image = UploadImage.objects.get(id=image_id).Image
real_image = RealImage()
real_image.Image = File(uploaded_image, uploaded_image.name)
real_image.save()
uploaded_image.close()
uploaded_image.delete()

如果我使用模型表来处理这个过程,我只会做

# django model forms provides a reference to the associated model via the instance property
form.instance.Image = File(uploaded_image, os.path.basename(uploaded_image.path))
form.save()
uploaded_image.close()
uploaded_image.delete()

请注意,我确保upload_image文件已关闭,因为调用real_image.save()将打开文件并读取其内容。这是由ImageField实例使用的存储系统

处理的

答案 2 :(得分:2)

尝试不使用表单即可。在不知道你得到的确切错误的情况下,我只能推测表单的clean()方法由于upload_to参数不匹配而引发错误。

这让我想到了下一点,如果你试图将图像从'temp /'复制到'real /',你将不得不做一些文件处理来自己移动文件(如果你有PIL则更容易) ):

import Image
from django.conf import settings

u = UploadImage.objects.get(id=image_id)
im = Image.open(settings.MEDIA_ROOT + str(u.Image))
newpath = 'real/' + str(u.Image).split('/', 1)[1]
im.save(settings.MEDIA_ROOT + newpath)
r = RealImage.objects.create(Image=newpath)

希望有帮助...

答案 3 :(得分:1)

我有同样的问题并且像这样解决了,希望它可以帮助任何人:

# models.py
class A(models.Model):
    # other fields...
    attachment = FileField(upload_to='a')

class B(models.Model):
    # other fields...
    attachment = FileField(upload_to='b')

# views.py or any file you need the code in
try:
    from cStringIO import StringIO
except ImportError:
    from StringIO import StringIO
from django.core.files.base import ContentFile
from main.models import A, B

obj1 = A.objects.get(pk=1)

# You and either copy the file to an existent object
obj2 = B.objects.get(pk=2)

# or create a new instance
obj2 = B(**some_params)

tmp_file = StringIO(obj1.attachment.read())
tmp_file = ContentFile(tmp_file.getvalue())
url = obj1.attachment.url.split('.')
ext = url.pop(-1)
name = url.pop(-1).split('/')[-1]  # I have my files in a remote Storage, you can omit the split if it doesn't help you
tmp_file.name = '.'.join([name, ext])
obj2.attachment = tmp_file

# Remember to save you instance
obj2.save()

答案 4 :(得分:-2)

更新Gerard的解决方案以通用方式处理它:

try:
    from cStringIO import StringIO
except ImportError:
    from StringIO import StringIO

from django.core.files.base import ContentFile

init_str = "src_obj." + src_field_name + ".read()"
file_name_str = "src_obj." + src_field_name + ".name"

try:
    tmp_file = StringIO(eval(str(init_str)))
    tmp_file = ContentFile(tmp_file.getvalue())
    tmp_file.name = os.path.basename(eval(file_name_str))
except AttributeError:
    tmp_file = None

if tmp_file:
    try:
        dest_obj.__dict__[dest_field_name] = tmp_file
        dest_obj.save()
    except KeyError:
        pass

使用变量:

  1. src_obj =来源附件对象。
  2. src_field_name =源附件对象的FileField名称。
  3. dest_obj =目标附件对象。
  4. dest_field_name =目标附件对象的FileField名称。