首先是有关该应用程序的一些信息:我希望能够上传一个 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 关系。现在我有两个问题:
- 我可以把上面的代码写得更好吗?(主要是不喜欢
super(Document, self).save(*args, **kwargs) 在代码中间。
- 对于 m2m 更新,我查看了 m2m_changed 信号,但老实说我不明白如何使用它。
编辑:
一旦你了解了参数,这实际上很容易,谢谢