[ADD] BASE: 404 for images

This commit is contained in:
Horilla
2024-07-29 16:42:54 +05:30
parent cf151556e7
commit 1df784dbd6
2 changed files with 18 additions and 0 deletions

View File

@@ -2,11 +2,28 @@ from auditlog.models import AuditlogHistoryField
from auditlog.registry import auditlog
from django.contrib.auth.models import User
from django.db import models
from django.db.models.fields.files import FieldFile
from django.urls import reverse
from django.utils.translation import gettext as _
from horilla.horilla_middlewares import _thread_locals
@property
def url(self: FieldFile):
"""
Custom url attribute/property
"""
try:
self._require_file()
except Exception as e:
return reverse("404")
return self.storage.url(self.name)
setattr(FieldFile, "url", url)
class HorillaModel(models.Model):
created_at = models.DateTimeField(
auto_now_add=True,