django - 使用 pre_save 信号编辑上传的文件(djangos FileField)
问题描述
我想在保存之前在字节级别编辑上传的文件(即搜索和删除某个字节序列)。
我通过以下方式设置了 pre_save 信号:
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)
这首先会导致无限循环。但这可以通过例如仅remove_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()
推荐阅读
- spring-boot - 需要在 Spring Boot 中动态生成 url 以进行招摇
- javascript - 如何确定 jquery ajax 调用何时完成?
- sql - 将数值转换为 VARCHAR2 会引发 ORA-03113
- teradata - 如何检查 Teradata 中的列是 CASESPECIFIC 还是 NOT CASESPECIFIC
- vue.js - Vue v-model 命名最佳实践
- python - spaCy中的token和span(来自文档的切片)有什么区别?
- java - Java Spark获取kafka解析json
- ssl - WebSocket 握手:意外的响应代码:200 - React 客户端
- python - Python datetime.datetime 不在 pandas._libs.tslibs.timestamps.Timestamp 系列中
- stm32 - STM32 Nucleo F103RB:LD1 使用外部电源缓慢闪烁