From f20f3287b983bcfe961bc0a82b186a120ecc8b1f Mon Sep 17 00:00:00 2001
From: Ian Day <ian@ianday.me>
Date: Sun, 28 Jul 2024 15:51:40 -0400
Subject: [PATCH] fix post plant

---
 plant_tracker/tracker/api/schemas.py    | 1 +
 plant_tracker/tracker/api/view_plant.py | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/plant_tracker/tracker/api/schemas.py b/plant_tracker/tracker/api/schemas.py
index 353ee9b..7912160 100644
--- a/plant_tracker/tracker/api/schemas.py
+++ b/plant_tracker/tracker/api/schemas.py
@@ -179,6 +179,7 @@ class Meta:
             "graveyard",
             "death_date",
             "notes",
+            "area"
         ]
         fields_optional = "__all__"  
     purchase_date: EmptyStrToDefault[date] = None
diff --git a/plant_tracker/tracker/api/view_plant.py b/plant_tracker/tracker/api/view_plant.py
index 0583c0f..4696d45 100644
--- a/plant_tracker/tracker/api/view_plant.py
+++ b/plant_tracker/tracker/api/view_plant.py
@@ -14,7 +14,7 @@
 from django.http import HttpRequest
 from ninja import File, Form, Router
 from django.contrib.auth import get_user_model
-from tracker.models import Location, Plant
+from tracker.models import Area, Location, Plant
 from ninja_jwt.authentication import JWTAuth
 from ninja_extra import status
 from ninja_extra.exceptions import APIException
@@ -92,6 +92,9 @@ def post_plant(
             new_user = get_user_model().objects.get(id=value)
             if user != new_user:
                 plant.user = new_user
+        elif attr == "area":
+            new_area = get_object_or_404(Area, id=value)
+            plant.area = new_area
         else:
             setattr(plant, attr, value)