使用pre_保存信号编辑上传的文件(djangos文件字段)
我想在保存上传的文件之前,在字节级别对其进行编辑(即搜索和删除某个字节序列) 我通过以下方式设置了预存信号:使用pre_保存信号编辑上传的文件(djangos文件字段),django,python-3.x,filefield,Django,Python 3.x,Filefield,我想在保存上传的文件之前,在字节级别对其进行编辑(即搜索和删除某个字节序列) 我通过以下方式设置了预存信号: class Snippet(models.Model): name = models.CharField(max_length=256, unique=True) audio_file = models.FileField(upload_to=generate_file_name, blank=True, null=True) @receiver(models.sign
class Snippet(models.Model):
name = models.CharField(max_length=256, unique=True)
audio_file = models.FileField(upload_to=generate_file_name, blank=True, null=True)
@receiver(models.signals.pre_save, sender=Snippet)
def prepare_save(sender, instance, **kwargs):
if instance.audio_file:
remove_headers(instance)
现在,我在实现remove_headers
函数时遇到了一些问题,因为我可以在文件仍在内存中时对其进行编辑,然后将其存储。我尝试了以下方法:
def remove_headers(instance):
byte_sequence = b'bytestoremove'
f = instance.audio_file.read()
file_in_hex = f.hex()
file_in_hex = re.sub(byte_sequence.hex(), '', file_in_hex)
x = b''
x = x.fromhex(file_in_hex)
tmp_file = TemporaryFile()
tmp_file.write(x)
tmp_file.flush()
tmp_file.seek(0)
instance.audio_file.save(instance.audio_file.name, tmp_file, save=True)
首先,这将导致一个无限循环。但这可以通过以下方法来缓解:例如,在create上只调用remove\u headers
方法。但是,它不起作用,文件没有更改。我还尝试将最后一行替换为:
instance.audio_file = File(tmp_file, name=instance.audio_file.name)
但是,这会导致写入/保存一个空文件。
奇怪的是,在编写测试时,这种方法似乎有效:
def test_header_removed(self):
snippet = mommy.make(Snippet)
snippet.audio_file.save('newname.mp3', ContentFile('contentbytestoremovecontent'))
snippet.save()
self.assertEqual(snippet.audio_file.read(), b'contentcontent')
尽管文件最终为零字节,但此测试不会失败。
我在这里遗漏了什么?第二个解决方案几乎是正确的。文件最终为空的原因(实际上,这只发生在较大的文件上)是,有时打开文件后必须查找文件的开头。因此需要更改
remove_headers
的开头:
def remove_headers(instance):
byte_sequence = b'bytestoremove'
instance.audio_file.seek(0)
f = instance.audio_file.read()
file_in_hex = f.hex()