51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
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()
|