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

This commit is contained in:
jonaswinkler committed 2020-12-16 21:53:11 +01:00
1 parent 5c310c51d4
commit aa8789ae31
2 files changed
+18 -6

No files matched your search

+6 -4
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()