pflaenz.li/pflaenzli/pflaenzli/models.py

45 lines
1.5 KiB
Python
Raw Normal View History

2023-02-19 12:29:08 +01:00
from django.db import models
2023-03-31 18:10:38 +02:00
from django.dispatch import receiver
2023-04-05 23:39:26 +02:00
from django.contrib.auth.models import AbstractUser
2023-02-19 23:56:05 +01:00
from django.utils import timezone
2023-03-31 18:10:38 +02:00
from django.core.files.storage import default_storage
2023-02-19 12:29:08 +01:00
2023-02-19 23:56:05 +01:00
2023-04-05 23:39:26 +02:00
class PflaenzliUser(AbstractUser):
2023-04-09 12:38:05 +02:00
email = models.EmailField(max_length=254)
zipcode = models.PositiveIntegerField(blank=True, null=True)
2023-04-05 18:18:55 +02:00
2023-02-19 23:56:05 +01:00
class Offer(models.Model):
created = models.DateTimeField(default=timezone.now)
2023-04-05 18:18:55 +02:00
user = models.ForeignKey(PflaenzliUser, on_delete=models.CASCADE)
2023-02-19 23:56:05 +01:00
title = models.CharField(max_length=50)
description = models.TextField(max_length=5000)
zipcode = models.IntegerField(blank=True, default=0)
image = models.ImageField(upload_to="uploads/")
class Wish(models.Model):
created = models.DateTimeField(default=timezone.now)
2023-04-05 18:18:55 +02:00
user = models.ForeignKey(PflaenzliUser, on_delete=models.CASCADE)
2023-02-19 23:56:05 +01:00
title = models.CharField(max_length=200)
2023-03-31 18:10:38 +02:00
@receiver(models.signals.post_delete, sender=Offer)
def auto_delete_image_on_delete(sender, instance, **kwargs):
"""
Deletes image from filesystem when corresponding `Offer` object is deleted.
"""
if instance.image and os.path.isfile(instance.image.path):
default_storage.delete(instance.image.path)
@receiver(models.signals.pre_save, sender=Offer)
def delete_old_image(sender, instance, **kwargs):
try:
old_image = sender.objects.get(pk=instance.pk).image
2023-03-31 22:51:53 +02:00
except sender.DoesNotExist:
2023-03-31 18:10:38 +02:00
return False
default_storage.delete(old_image.path)