0

首先是有关该应用程序的一些信息:我希望能够上传一个 pdf 文件,该 pdf 文件将被转换为图像(对于每个 pdf 页一个图像)。然后这些图像将显示在网站上,并且可以下载 pdf: 到目前为止,我有这个:

def upload_to(path):
    def upload_callback(instance, filename):
        return '%s' % os.path.join(settings.MEDIA_ROOT, path, str(instance), filename)
    return upload_callback

class Document(models.Model):
    name = models.TextField()
    pdf_file = models.FileField(upload_to=upload_to('pdfs'))
    images = models.ManyToManyField('Image', null=True, blank=True)

class Image(models.Model):
    image = models.ImageField(upload_to='pdfimages')

现在要创建图像和图像对象,我重写了 Document 类的保存函数:我看起来像这样:

def save(self, *args, **kwargs):
    path = os.path.join(settings.MEDIA_ROOT, 'pdfs', self.name)
    imagepath = os.path.join(settings.MEDIA_ROOT, 'pdfimages')

    if os.path.exists(path):
        for file_ in os.listdir(path):
            if file_ == self.pdf_file.name:
                continue

    super(Document, self).save(*args, **kwargs)

    save_to = os.path.join(path, os.path.splitext(self.pdf_file.name)[0] +
                           '_page.png')
    pdffile = os.path.join(path, self.pdf_file.name)

    args = ['convert', '-density', '100', '-depth', 
            '8', '-quality', '85', pdffile, save_to ]

    subprocess.check_call(args, stdout=sys.stdout, stderr=sys.stderr)
    for file_ in os.listdir(path):
        if file_.endswith('png'):
            try:
                shutil.move((os.path.join(settings.MEDIA_ROOT,
                                          'pdfs', self.name, file_)),
                            imagepath)

            except Exception:
                pass

     for file_ in os.listdir(imagepath):
        i, created = Image.objects.get_or_create(image="pdfimages/%s" % file_)

首先代码不是很好,其次最重要的操作仍然缺失。将图像添加到文档意味着更新 m2m 关系。现在我有两个问题:

  1. 我可以把上面的代码写得更好吗?(主要是不喜欢

    super(Document, self).save(*args, **kwargs) 在代码中间。

  2. 对于 m2m 更新,我查看了 m2m_changed 信号,但老实说我不明白如何使用它。
任何帮助表示赞赏。

编辑:

一旦你了解了参数,这实际上很容易,谢谢

4

1 回答 1

-1

查看这篇关于 m2m 信号使用的帖子

您可以将所有后期保存处理逻辑移动到信号

于 2012-09-13T15:49:29.127 回答