Update app.py

app.py

@@ -1766,106 +1766,31 @@ def add_data_ver_cur():
 
 
 
-# Шаблон сопоставления для кураторов
-mapping_template_cur = {
-    'name': 'name',
-    'phone': 'phone',
-    'email': 'email'
-}
-
-DATABASE_NAME4 = 'data_gc.db'
-
-
-def verify_phone_number_v2(phone_number):
-    print(f"verifikation_start: {verifikation_start}")
-
-    if verifikation_start == "1":
-        full_url_ver = f"{wa_url}{wa_ak}{ws_url_ver}{wa_api_key}"
-        print(f"Full URL: {full_url_ver}")
-
-        # Создаем payload для запроса
-        payload = json.dumps({
-            "phoneNumber": phone_number
-        })
-        headers = {
-            'Content-Type': 'application/json'
-        }
-
-        # Отправляем POST-запрос
-        response = requests.post(full_url_ver, headers=headers, data=payload)
-
-        # Печатаем статус ответа
-        print(f"Response Status Code: {response.status_code}")
-
-        # Проверяем статус код ответа
-        if response.status_code == 200:
-            # Печатаем текст ответа от сервера
-            response_body = response.json()
-            print(f"Response Body: {response_body}")
-
-            # Извлекаем значение из JSON
-            exists_whatsapp = response_body.get('existsWhatsapp', 'false')
-            print(f"existsWhatsapp: {exists_whatsapp}")
-
-            # Возвращаем значение
-            return exists_whatsapp
-        else:
-            print("Error: Unable to fetch data")
-            return "Error"
-    else:
-        print("Verification not started")
-        return "false"  # Возвращаем значение, что WhatsApp не существует
-
-def add_or_update_contact(contact_data):
-    conn = sqlite3.connect(DATABASE_NAME4)
-    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')
-
-    cursor.execute("SELECT id FROM contacts WHERE email = ?", (email,))
-    contact = cursor.fetchone()
+@app.route('/gc_in', methods=['GET'])
+def add_data_gc_in():
+    global current_curator_index
 
-    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', 'ad_url', 'key_pr', 'n_con', 'canal', 'data_t'
-    ]
+    veref_on_off = request.args.get('ver', '0')  # Включает "1" и выключает "0" верификацию номера вместо verifikation_start
+    curator_on_off = request.args.get('cur', '0')  # Включает "1" и выключает "0" назначение куратора
     
-    placeholders = ", ".join([f"{field} = ?" for field in fields])
-
-    if contact:
-        update_query = f"UPDATE contacts SET {placeholders} WHERE id = ?"
-        cursor.execute(update_query, (*[contact_data.get(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.get(field, '') for field in fields))
+    template_key = request.args.get('template_key', 'avp')
+    mapping_template_cur = mapp_templates.get(template_key, mt_avp)
 
-    conn.commit()
-    conn.close()
+    user_data = {mapping_template_cur[key]: request.args.get(key, "") for key in mapping_template_cur}
 
-@app.route('/add_data_bot', methods=['GET'])
-def add_data_bot():
-    # Получаем данные пользователя из параметров запроса
-    user_data = {key: request.args.get(key, "") for key in mapping_template_cur.keys()}
+    if curator_on_off == "1":
+        user_data['curator'] = curators[current_curator_index]
 
-    # Верификация номера телефона
-    phone_verification_response = verify_phone_number_v2(user_data['phone'])
-    if phone_verification_response is not None:
-        user_data['ws_st'] = phone_verification_response
+    if veref_on_off == "1":
+        phone_verification_response = verify_phone_number(user_data.get('phone', ''))
+        if phone_verification_response is not None:
+            user_data['ws_st'] = phone_verification_response
 
     try:
-        # Добавляем пользователя в базу данных
         add_or_update_contact(user_data)
-        return jsonify({'status': 'success', 'message': 'User added'})
+        if curator_on_off == "1":
+            current_curator_index = (current_curator_index + 1) % len(curators)
+        return jsonify({'status': 'success', 'message': f'User added with curator {user_data.get("curator", "not assigned")}'})
     except Exception as e:
         logging.error(f"Error adding user: {e}")
         return jsonify({'status': 'error', 'message': str(e)}), 500
@@ -1896,7 +1821,6 @@ def add_data_bot():
 
 
 
-
 
 
 initialize_requests()