import uuid from django.conf import settings from django.db import models from vbv_lernwelt.core.models import User from vbv_lernwelt.files.enums import FileUploadStorage from vbv_lernwelt.files.integrations import s3_delete_file, s3_generate_presigned_url from vbv_lernwelt.files.utils import file_generate_upload_path # Inspired by https://www.hacksoft.io/blog/direct-to-s3-file-upload-with-django # Code https://github.com/HackSoftware/Django-Styleguide-Example/tree/bdadf52b849bb5fa47854a3094f4da6fe9d54d02/styleguide_example/files class UploadFile(models.Model): id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False) file = models.FileField(upload_to=file_generate_upload_path, blank=True, null=True) original_file_name = models.TextField() file_name = models.CharField(max_length=255, unique=True) file_type = models.CharField(max_length=255) # As a specific behavior, # We might want to preserve files after the uploader has been deleted. # In case you want to delete the files too, use models.CASCADE & drop the null=True uploaded_by = models.ForeignKey(User, null=True, on_delete=models.SET_NULL) upload_finished_at = models.DateTimeField(blank=True, null=True) @property def is_valid(self): """ We consider a file "valid" if the datetime flag has value. """ return bool(self.upload_finished_at) @property def url(self): if settings.FILE_UPLOAD_STORAGE == FileUploadStorage.S3.value: return s3_generate_presigned_url(file_path=str(self.file)) return f"{self.file.url}" def delete_file(self): if settings.FILE_UPLOAD_STORAGE == FileUploadStorage.S3.value: return s3_delete_file(file_path=str(self.file)) else: return self.file.delete()