Create user API fix: return code for existing user

app/api/web_api.py

@@ -82,9 +82,11 @@ def api_add_user():
 
     user = User(user_data["name"], user_data["key"], user_data.get("slack_id"))
 
-    user.save()
-
-    return jsonify({"message": "User added successfully"})
+    number_of_new_user_added = user.save()
+    if number_of_new_user_added == 1:
+        return jsonify({"message": "User added successfully"}), 201
+    if number_of_new_user_added == 0:
+        return jsonify({"message": "User already exists"}), 303
 
 
 @web_api.route("/api/users/<user_key>", methods=["DELETE"])

app/models/user.py

@@ -19,14 +19,17 @@ def get_by_key(cls, key):
             return None
 
     def save(self):
+        """
+        Returns number of new user added
+        """
         # Connect to the database
         conn = sqlite3.connect(app.config["DATABASE_URI"])
         cursor = conn.cursor()
 
         # Check if user already exists
         cursor.execute("SELECT * FROM users WHERE key = ?", (self.key,))
         existing_user = cursor.fetchone()
-
+        number_of_new_user = 0
         if existing_user:
             # Update existing user data
             cursor.execute(
@@ -39,10 +42,12 @@ def save(self):
                 "INSERT INTO users (name, key, slack_id) VALUES (?, ?, ?)",
                 (self.name, self.key, self.slack_id),
             )
+            number_of_new_user = 1
             conn.commit()
 
         # Close the connection
         conn.close()
+        return number_of_new_user
 
     def delete(self):
         # Connect to the database