From b209029640e6f2485b6f9e030956ea9cfcd90b9e Mon Sep 17 00:00:00 2001
From: Milan Gruner <milangruner@gmail.com>
Date: Tue, 19 Mar 2024 18:23:40 +0100
Subject: [PATCH] Fix existing population entries not being overwritten

---
 .../app/api/v0/city/[city]/population/route.ts    | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/app/src/app/api/v0/city/[city]/population/route.ts b/app/src/app/api/v0/city/[city]/population/route.ts
index eda446ae6..a72c0bbfb 100644
--- a/app/src/app/api/v0/city/[city]/population/route.ts
+++ b/app/src/app/api/v0/city/[city]/population/route.ts
@@ -21,7 +21,10 @@ export const POST = apiHandler(async (req, { session, params }) => {
       year: body.cityPopulationYear,
     },
   });
-  if (!cityPopulation) {
+  if (cityPopulation) {
+    cityPopulation.population = body.cityPopulation;
+    await cityPopulation.save();
+  } else {
     await db.models.Population.create({
       cityId,
       population: body.cityPopulation,
@@ -35,7 +38,10 @@ export const POST = apiHandler(async (req, { session, params }) => {
       year: body.regionPopulationYear,
     },
   });
-  if (!regionPopulation) {
+  if (regionPopulation) {
+    regionPopulation.regionPopulation = body.regionPopulation;
+    await regionPopulation.save();
+  } else {
     await db.models.Population.create({
       cityId,
       regionPopulation: body.regionPopulation,
@@ -49,7 +55,10 @@ export const POST = apiHandler(async (req, { session, params }) => {
       year: body.countryPopulationYear,
     },
   });
-  if (!countryPopulation) {
+  if (countryPopulation) {
+    countryPopulation.countryPopulation = body.countryPopulation;
+    await countryPopulation.save();
+  } else {
     countryPopulation = await db.models.Population.create({
       cityId,
       countryPopulation: body.countryPopulation,