Update app.py

app.py

@@ -805,26 +805,25 @@ def update_or_insert_user(db_name, user_data, mapping_template):
             phone = phone[1:]
         transformed_data['phone'] = phone
 
+    web_st_value = 0
     if user:
-        web_st_value = user[0]
-        if web_st_value is None or web_st_value == "":
-            web_st_value = 0
-        else:
-            web_st_value = int(web_st_value) + 1
-
-        update_query = "UPDATE contacts SET "
-        update_values = []
-        for column, value in transformed_data.items():
-            update_query += f"{column} = ?, "
-            update_values.append(value)
-        update_query += "web_st = ? WHERE email = ?"
-        update_values.extend([web_st_value, email])
-        cursor.execute(update_query, update_values)
-    else:
+        web_st_value = int(user[0]) if user[0] else 0
+        web_st_value += 1
+
+    update_query = "UPDATE contacts SET "
+    update_values = []
+    for column, value in transformed_data.items():
+        update_query += f"{column} = ?, "
+        update_values.append(value)
+    update_query += "web_st = ? WHERE email = ?"
+    update_values.extend([web_st_value, email])
+    cursor.execute(update_query, update_values)
+
+    if cursor.rowcount == 0:
         columns = ', '.join(transformed_data.keys()) + ", web_st"
         placeholders = ', '.join('?' for _ in transformed_data) + ", ?"
         insert_query = f"INSERT INTO contacts ({columns}) VALUES ({placeholders})"
-        insert_values = list(transformed_data.values()) + [0]
+        insert_values = list(transformed_data.values()) + [web_st_value]
         cursor.execute(insert_query, insert_values)
 
     conn.commit()
@@ -881,7 +880,6 @@ def send_get_request():
 
 
 
-
 @app.route('/biz_v', methods=['GET'])
 def biz_v():
     api_sys_control = request.args.get('api_sys')