fix(zone): add way to get data from db, fix migration

ollz272 · May 1, 2023 · 3df9dcb · 3df9dcb
1 parent 86f4897
commit 3df9dcb
Show file tree

Hide file tree

Showing 6 changed files with 21 additions and 6 deletions.
diff --git a/.idea/garden-server.iml b/.idea/garden-server.iml
diff --git a/.idea/modules.xml b/.idea/modules.xml
diff --git a/apps/zones/migrations/0002_make_plant_zones.py b/apps/zones/migrations/0002_make_plant_zones.py
@@ -7,10 +7,10 @@ def set_plant_zone(apps, schema_editor):
     # Set the partners' region as the needs region as a default.
     for plant in Plant.objects.all():
         if plant.indoor:
-            zone = Zone.objects.get_or_create(user=plant.user, name="indoor")
+            zone, _ = Zone.objects.get_or_create(user=plant.user, name="indoor")
 
         else:
-            zone = Zone.objects.get_or_create(user=plant.user, name="outdoor")
+            zone, _ = Zone.objects.get_or_create(user=plant.user, name="outdoor")
 
         plant.zone = zone
         plant.save()

diff --git a/poetry.lock b/poetry.lock
diff --git a/project/settings/base.py b/project/settings/base.py
@@ -47,6 +47,7 @@
     "crispy_bootstrap5",
     "django_extensions",
     "colorfield",
+    "maskpostgresdata",
 ]
 
 PROJECT_APPS = ["plants", "api", "accounts", "weather", "alerts", "zones"]

diff --git a/pyproject.toml b/pyproject.toml
@@ -27,6 +27,7 @@ psycopg = "^3.1.8"
 whitenoise = "^6.4.0"
 crispy-bootstrap5 = "^0.7"
 pytz = "^2023.3"
+django-maskpostgresdata = "^0.1.16"
 
 
 [tool.poetry.group.local.dependencies]