bloombase/plants/models.py
Stephan Kerkman b88ba03884 feat: location FK, multi-photo gallery, hide pruning history
- Location model: plants now FK to Location instead of free-text CharField
- Data migration 0004 converts existing location strings to Location objects
- PlantForm uses ModelChoiceField dropdown + inline "add new location" text field
- PlantPhoto model: multiple photos per plant with is_thumbnail flag
- Plant.thumbnail_url property: picks marked thumbnail, falls back to first photo then legacy photo field
- Photo gallery partial: upload, set thumbnail (★), delete (✕) with HTMX swaps on #photo-gallery
- plant_detail: uses thumbnail_url for hero image, pruning history section removed
- plant_list search updated: location__name__icontains instead of location__icontains
- Three new URL routes: upload_photo, set_thumbnail, delete_photo
- All 53 tests updated and passing

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-05-28 18:27:45 +02:00

94 lines
2.9 KiB
Python

from django.db import models
class Location(models.Model):
name = models.CharField(max_length=200, unique=True)
class Meta:
ordering = ['name']
def __str__(self):
return self.name
class Species(models.Model):
common_name = models.CharField(max_length=200)
scientific_name = models.CharField(max_length=200, blank=True)
description = models.TextField(blank=True)
watering = models.CharField(max_length=100, blank=True)
sunlight = models.CharField(max_length=200, blank=True)
shadow_tolerance = models.CharField(max_length=100, blank=True)
max_height_cm = models.PositiveIntegerField(null=True, blank=True)
growth_rate = models.CharField(max_length=100, blank=True)
growth_season = models.CharField(max_length=100, blank=True)
pruning_months = models.JSONField(default=list)
bloom_months = models.JSONField(default=list)
api_image_url = models.URLField(blank=True)
perenual_id = models.IntegerField(unique=True, null=True, blank=True)
class Meta:
ordering = ['common_name']
verbose_name_plural = 'species'
def __str__(self):
return self.common_name
class Plant(models.Model):
name = models.CharField(max_length=200)
species = models.ForeignKey(
Species, null=True, blank=True,
on_delete=models.SET_NULL, related_name='plants',
)
location = models.ForeignKey(
Location, null=True, blank=True,
on_delete=models.SET_NULL, related_name='plants',
)
is_indoor = models.BooleanField(default=False)
pruning_months = models.JSONField(default=list)
bloom_months = models.JSONField(default=list)
photo = models.ImageField(upload_to='plants/', blank=True)
notes = models.TextField(blank=True)
date_added = models.DateField(auto_now_add=True)
class Meta:
ordering = ['name']
def __str__(self):
return self.name
@property
def thumbnail_url(self):
thumb = self.photos.filter(is_thumbnail=True).first()
if thumb:
return thumb.image.url
first = self.photos.first()
if first:
return first.image.url
if self.photo:
return self.photo.url
return None
class PlantPhoto(models.Model):
plant = models.ForeignKey(Plant, on_delete=models.CASCADE, related_name='photos')
image = models.ImageField(upload_to='plants/photos/')
is_thumbnail = models.BooleanField(default=False)
uploaded_at = models.DateTimeField(auto_now_add=True)
class Meta:
ordering = ['uploaded_at']
class PruningLog(models.Model):
plant = models.ForeignKey(
Plant, on_delete=models.CASCADE, related_name='pruning_logs',
)
pruned_on = models.DateField()
notes = models.TextField(blank=True)
class Meta:
ordering = ['-pruned_on']
def __str__(self):
return f'{self.plant.name} pruned on {self.pruned_on}'