Update app.py

app.py

@@ -806,26 +806,30 @@ def update_or_insert_user(db_name, user_data, mapping_template):
     logging.debug(f"Processing user with email: {email}")
     
     # Проверка существования пользователя в базе данных по email
-    cursor.execute("SELECT b_fin FROM contacts WHERE email = ?", (email,))
+    cursor.execute("SELECT ws_st, b_fin FROM contacts WHERE email = ?", (email,))
     user = cursor.fetchone()
     logging.debug(f"User found: {user}")
 
-    # Вынесение увеличения значения b_fin в отдельный блок
-    b_fin_value = 1  # Инициализация значения b_fin
+    # Вынесение увеличения значений ws_st и b_fin в отдельные блоки
+    ws_st_value = 1  # Инициализация значения ws_st
+    b_fin_value = 0  # Инициализация значения b_fin
     if user:
-        # Проверка текущего значения b_fin и его инкрементация
-        current_b_fin = user[0] if user[0] is not None and user[0] != "" else 0
+        # Проверка текущих значений ws_st и b_fin и их инкрементация
+        current_ws_st = user[0] if user[0] is not None and user[0] != "" else 0
+        current_b_fin = user[1] if user[1] is not None and user[1] != "" else 0
+        
+        ws_st_value = int(current_ws_st) + 1
         if user_data.get('finished') == "True":
             b_fin_value = int(current_b_fin) + 1
         else:
             b_fin_value = current_b_fin
-        logging.debug(f"Calculated b_fin_value: {b_fin_value}")
+        logging.debug(f"Calculated ws_st_value: {ws_st_value}, b_fin_value: {b_fin_value}")
 
-        # Обновление значения b_fin
-        cursor.execute("UPDATE contacts SET b_fin = ? WHERE email = ?", (b_fin_value, email))
+        # Обновление значений ws_st и b_fin
+        cursor.execute("UPDATE contacts SET ws_st = ?, b_fin = ? WHERE email = ?", (ws_st_value, b_fin_value, email))
         conn.commit()
         conn.close()
-        logging.debug(f"User {email} b_fin updated to {b_fin_value}")
+        logging.debug(f"User {email} ws_st updated to {ws_st_value}, b_fin updated to {b_fin_value}")
     else:
         conn.close()
         logging.debug(f"User {email} not found, proceeding with insert")
@@ -870,7 +874,8 @@ 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}")
 
-    # Добавление значения b_fin в данные для вставки
+    # Добавление значений ws_st и b_fin в данные для вставки
+    transformed_data['ws_st'] = ws_st_value
     transformed_data['b_fin'] = b_fin_value
 
     # Обновление данных пользователя в базе данных