Update app.py

app.py

@@ -791,6 +791,7 @@ def update_or_insert_user(db_name, user_data, mapping_template):
     email = user_data.get('email')
     logging.debug(f"Processing user with email: {email}")
     
+    # Проверяем, существует ли пользователь с заданным email
     cursor.execute("SELECT web_st FROM contacts WHERE email = ?", (email,))
     user = cursor.fetchone()
     logging.debug(f"User found: {user}")
@@ -823,11 +824,12 @@ def update_or_insert_user(db_name, user_data, mapping_template):
         transformed_data['phone'] = phone
     logging.debug(f"Transformed data after phone processing: {transformed_data}")
 
-    # Увеличиваем значение web_st на единицу
-    update_query = "UPDATE contacts SET web_st = web_st + 1 WHERE email = ?"
-    cursor.execute(update_query, (email,))
-
-    if cursor.rowcount == 0:
+    # Если пользователь существует, увеличиваем значение web_st на единицу
+    if user:
+        update_query = "UPDATE contacts SET web_st = web_st + 1 WHERE email = ?"
+        cursor.execute(update_query, (email,))
+    else:
+        # Если пользователь не существует, вставляем новую запись с web_st равным 1
         columns = ', '.join(transformed_data.keys()) + ", web_st"
         placeholders = ', '.join('?' for _ in transformed_data) + ", ?"
         insert_query = f"INSERT INTO contacts ({columns}) VALUES ({placeholders})"