Update app.py

app.py

@@ -1559,7 +1559,6 @@ def add_data_ver_cur2():
     curator_on_off = request.args.get('cur', '0')  # Включает "1" и выключает "0" назначение куратора
     order = request.args.get('order', '')
     status = request.args.get('status', '')
-    del_flag = request.args.get('del', '')
     template_key = request.args.get('template_key', 'avp')
     mapping_template_cur = mapp_templates.get(template_key, mt_avp)
 
@@ -1601,6 +1600,47 @@ def add_data_ver_cur2():
         return jsonify({'status': 'error', 'message': str(e)}), 500
 
 
+def add_or_update_contact(contact_data):
+    conn = sqlite3.connect(DATABASE_NAME3)
+    cursor = conn.cursor()
+
+    email = contact_data.get('email')
+    if not email:
+        logging.error(f"Missing email in contact data: {contact_data}")
+        return
+
+    utc_now = datetime.utcnow()
+    msk_tz = pytz.timezone('Europe/Moscow')
+    msk_now = utc_now.replace(tzinfo=pytz.utc).astimezone(msk_tz)
+    contact_data['data_t'] = msk_now.strftime('%Y-%m-%d %H:%M:%S')
+
+    # Список всех возможных полей
+    fields = [
+        'name', 'phone', 'email', 'vk_id', 'chat_id', 'ws_st', 'ws_stop', 'web_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', 'gc_url', 'key_pr', 'n_con', 'canal', 'data_on', 'data_t', 'utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content', 'gcpc'
+    ]
+
+    # Устанавливаем значения по умолчанию для отсутствующих полей
+    for field in fields:
+        if field not in contact_data:
+            contact_data[field] = ''
+
+    placeholders = ", ".join([f"{field} = ?" for field in fields])
+
+    cursor.execute("SELECT id FROM contacts WHERE email = ?", (email,))
+    contact = cursor.fetchone()
+
+    if contact:
+        update_query = f"UPDATE contacts SET {placeholders} WHERE id = ?"
+        cursor.execute(update_query, (*[contact_data[field] for field in fields], contact[0]))
+    else:
+        insert_query = f"INSERT INTO contacts ({', '.join(fields)}) VALUES ({', '.join(['?' for _ in fields])})"
+        cursor.execute(insert_query, tuple(contact_data[field] for field in fields))
+
+    conn.commit()
+    conn.close()
+