ft:update_occupants -- increases occupants count

BrianMwevi · Jun 16, 2022 · 36da373 · 36da373
1 parent 3e5c6f3
commit 36da373
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/neighborhoods/models.py b/neighborhoods/models.py
@@ -32,6 +32,10 @@ def update_neighborhood(cls, obj):
         neighborhood.save()
         return neighborhood
 
+    def update_occupants(self, occupants):
+        self.occupants += occupants
+        self.save()
+        return self
 
     def __str__(self):
         return self.location