Squash merge feature/email-notifications into beta
This commit is contained in:
@@ -1,4 +1,7 @@
|
||||
import uuid
|
||||
|
||||
from django.conf import settings
|
||||
from django.core.files.storage import default_storage
|
||||
from django.db import models
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
|
||||
@@ -90,6 +93,22 @@ class Claim(models.Model):
|
||||
performed_by=performed_by,
|
||||
)
|
||||
|
||||
def save(self, *args, **kwargs):
|
||||
if self.receipt and not kwargs.pop("skip_receipt_rename", False):
|
||||
original_name = self.receipt.name or ""
|
||||
ext = original_name.rsplit(".", 1)[-1] if "." in original_name else "dat"
|
||||
new_name = self._generate_unique_receipt_name(ext)
|
||||
self.receipt.name = new_name
|
||||
super().save(*args, **kwargs)
|
||||
|
||||
def _generate_unique_receipt_name(self, ext):
|
||||
ext = ext.lower()
|
||||
for _ in range(10):
|
||||
candidate = f"receipts/{uuid.uuid4().hex}.{ext}"
|
||||
if not default_storage.exists(candidate):
|
||||
return candidate
|
||||
return f"receipts/{uuid.uuid4().hex}.{ext}"
|
||||
|
||||
|
||||
class ClaimLog(models.Model):
|
||||
class Action(models.TextChoices):
|
||||
|
||||
Reference in New Issue
Block a user