Update app.py

app.py

@@ -1705,15 +1705,38 @@ def add_or_update_contact(contact_data):
 
 @app.route('/add_data_ver', methods=['GET'])
 def add_data_ver():
+    # Define all possible fields
     contact_data = {
-        'name': request.args.get('name'),
-        'phone': request.args.get('phone'),
-        'email': request.args.get('email')
+        'name': request.args.get('name', ''),
+        'phone': request.args.get('phone', ''),
+        'email': request.args.get('email', ''),
+        'ws_st': '',  # Initialize ws_st and other optional fields
+        'fin_prog': '',
+        'b_city': '',
+        'b_fin': '',
+        'b_ban': '',
+        'b_ign': '',
+        'b_baners': '',
+        'b_butt': '',
+        'b_mess': '',
+        'shop_st': '',
+        'curator': '',
+        'pr1': '',
+        'pr2': '',
+        'pr3': '',
+        'pr4': '',
+        'pr5': '',
+        'ad_url': '',
+        'key_pr': '',
+        'n_con': '',
+        'canal': '',
+        'data_t': ''
     }
 
+    # Verify phone number and update contact_data
     phone_verification_response = verify_phone_number(contact_data['phone'])
     if phone_verification_response is not None:
-        contact_data['ws_st'] = phone_verification_response  # Сохраняем значение в поле ws_st
+        contact_data['ws_st'] = phone_verification_response  # Update ws_st with verification response
 
     try:
         add_or_update_contact(contact_data)
@@ -1722,6 +1745,37 @@ def add_data_ver():
         logging.error(f"Error adding/updating contact: {e}")
         return jsonify({'status': 'error', 'message': str(e)}), 500
 
+def add_or_update_contact(contact_data):
+    conn = sqlite3.connect(DATABASE_NAME)
+    cursor = conn.cursor()
+
+    email = contact_data.get('email')
+    if not email:
+        logging.error(f"Missing email in contact data: {contact_data}")
+        return
+
+    cursor.execute("SELECT id FROM contacts WHERE email = ?", (email,))
+    contact = cursor.fetchone()
+
+    # List all fields for updating or inserting
+    fields = ["name", "phone", "email", "vk_id", "chat_id", "ws_st", "fin_prog", "b_city", "b_fin", "b_ban", 
+              "b_ign", "b_baners", "b_butt", "b_mess", "shop_st", "curator", "pr1", "pr2", "pr3", "pr4", 
+              "pr5", "ad_url", "key_pr", "n_con", "canal", "data_t"]
+    
+    # Only include fields with non-empty values
+    fields_to_update = [field for field in fields if contact_data.get(field, '') != '']
+    placeholders = ", ".join([f"{field} = ?" for field in fields_to_update])
+
+    if contact:
+        update_query = f"UPDATE contacts SET {placeholders} WHERE id = ?"
+        cursor.execute(update_query, (*[contact_data[field] for field in fields_to_update], contact[0]))
+    else:
+        insert_query = f"INSERT INTO contacts ({', '.join(fields_to_update)}) VALUES ({', '.join(['?' for _ in fields_to_update])})"
+        cursor.execute(insert_query, tuple(contact_data[field] for field in fields_to_update))
+
+    conn.commit()
+    conn.close()
+