From b3f0a1cd405df585a1d484c4a0744e560def74c6 Mon Sep 17 00:00:00 2001
From: Stephan Kerkman <stephan@kerkman.io>
Date: Thu, 28 May 2026 09:48:06 +0200
Subject: [PATCH] feat: add PlantForm and PruningLogForm

---
 plants/forms.py | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 plants/forms.py

diff --git a/plants/forms.py b/plants/forms.py
new file mode 100644
index 0000000..2a3f165
--- /dev/null
+++ b/plants/forms.py
@@ -0,0 +1,47 @@
+from django import forms
+from .models import Plant, PruningLog
+
+MONTH_CHOICES = [
+    ('1', 'Jan'), ('2', 'Feb'), ('3', 'Mar'), ('4', 'Apr'),
+    ('5', 'May'), ('6', 'Jun'), ('7', 'Jul'), ('8', 'Aug'),
+    ('9', 'Sep'), ('10', 'Oct'), ('11', 'Nov'), ('12', 'Dec'),
+]
+
+
+class PlantForm(forms.ModelForm):
+    pruning_months = forms.MultipleChoiceField(
+        choices=MONTH_CHOICES,
+        widget=forms.CheckboxSelectMultiple,
+        required=False,
+        label='Pruning months',
+    )
+
+    class Meta:
+        model = Plant
+        fields = ['name', 'location', 'is_indoor', 'pruning_months', 'photo', 'notes']
+        widgets = {
+            'notes': forms.Textarea(attrs={'rows': 3}),
+            'name': forms.TextInput(attrs={'class': 'form-control'}),
+            'location': forms.TextInput(attrs={'class': 'form-control'}),
+        }
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        # JSONField stores ints; CheckboxSelectMultiple expects strings matching choice values
+        if self.instance.pk and self.instance.pruning_months:
+            self.initial['pruning_months'] = [str(m) for m in self.instance.pruning_months]
+        elif not self.instance.pk and self.initial.get('pruning_months'):
+            self.initial['pruning_months'] = [str(m) for m in self.initial['pruning_months']]
+
+    def clean_pruning_months(self):
+        return [int(m) for m in self.cleaned_data.get('pruning_months', [])]
+
+
+class PruningLogForm(forms.ModelForm):
+    class Meta:
+        model = PruningLog
+        fields = ['pruned_on', 'notes']
+        widgets = {
+            'pruned_on': forms.DateInput(attrs={'type': 'date', 'class': 'form-control'}),
+            'notes': forms.Textarea(attrs={'rows': 2, 'class': 'form-control'}),
+        }