fix up the migration for encrypted documents and a couple other associated issues.

This commit is contained in:
jonaswinkler
2020-12-16 21:53:11 +01:00
parent 1410bc4ce6
commit 5e156d8124
2 changed files with 18 additions and 6 deletions

View File

@@ -30,12 +30,14 @@ def add_mime_types(apps, schema_editor):
documents = Document.objects.all()
for d in documents:
f = open(source_path(d), "rb")
if d.storage_type == STORAGE_TYPE_GPG:
f = GnuPG.decrypted(open(source_path(d), "rb"))
else:
f = open(source_path(d), "rb")
d.mime_type = magic.from_buffer(f.read(1024), mime=True)
data = GnuPG.decrypted(f)
else:
data = f.read(1024)
d.mime_type = magic.from_buffer(data, mime=True)
d.save()
f.close()