robosats/api/models/notification.py

# We use custom seeded UUID generation during testing
import uuid

from decouple import config
from api.models import Order, Robot
from django.db import models
from django.utils import timezone

if config("TESTING", cast=bool, default=False):
    import random
    import string

    random.seed(1)
    chars = string.ascii_lowercase + string.digits

    def custom_uuid():
        return uuid.uuid5(uuid.NAMESPACE_DNS, "".join(random.choices(chars, k=20)))

else:
    custom_uuid = uuid.uuid4


class Notification(models.Model):
    # notification info
    created_at = models.DateTimeField(default=timezone.now)

    robot = models.ForeignKey(Robot, on_delete=models.CASCADE, default=None)
    order = models.ForeignKey(Order, on_delete=models.CASCADE, default=None)

    # notification details
    title = models.CharField(max_length=240, null=False, default=None)
    description = models.CharField(max_length=240, default=None, blank=True)

    def __str__(self):
        return f"{self.title} {self.description}"