Update app.py

app.py

@@ -1619,16 +1619,13 @@ def gc_forms():
 
 # Список кураторов
 curators = ["Anna", "Ekaterina", "Ivan", "Maria", "Sergey", "Olga", "Alex", "Natalia", "Dmitry", "Elena"]
-
-# Переменная для отслеживания текущего куратора
 current_curator_index = 0
 
-# Шаблоны сопоставления
 mt_avp = {
     'name': 'name',
     'phone': 'phone',
     'email': 'email',
-    'pr5': 'cur'
+    'cur': 'pr5'
 }
 
 mt_bhelp = {
@@ -1652,7 +1649,6 @@ mt_tl = {
     'cur': 'curator'
 }
 
-# Словарь для сопоставления ключей с шаблонами
 mapp_templates = {
     'avp': mt_avp,
     'bhelp': mt_bhelp,
@@ -1662,47 +1658,25 @@ mapp_templates = {
 
 DATABASE_NAME3 = 'data_gc.db'
 
-def verify_phone_number(phone_number):
-    print(f"verifikation_start: {verifikation_start}")
 
+def verify_phone_number(phone_number):
     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'
-        }
+        
+        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
+            return response_body.get('existsWhatsapp', 'false')
         else:
-            print("Error: Unable to fetch data")
             return "Error"
     else:
-        print("Verification not started")
-        return "false"  # Возвращаем значение, что WhatsApp не существует
+        return "false"
 
-def add_or_update_contact(contact_data, mapping_template_cur):
+def add_or_update_contact(contact_data):
     conn = sqlite3.connect(DATABASE_NAME3)
     cursor = conn.cursor()
 
@@ -1711,18 +1685,11 @@ def add_or_update_contact(contact_data, mapping_template_cur):
         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')
 
-    # Обработка параметра 'cur' в зависимости от шаблона
-    if 'cur' in contact_data:
-        cur_key = next((value for key, value in mapping_template_cur.items() if value == 'cur'), None)
-        if cur_key:
-            contact_data[cur_key] = contact_data.pop('cur')
-
     cursor.execute("SELECT id FROM contacts WHERE email = ?", (email,))
     contact = cursor.fetchone()
 
@@ -1731,7 +1698,7 @@ def add_or_update_contact(contact_data, mapping_template_cur):
         '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'
     ]
-    
+
     placeholders = ", ".join([f"{field} = ?" for field in fields])
 
     if contact:
@@ -1748,28 +1715,20 @@ def add_or_update_contact(contact_data, mapping_template_cur):
 def add_data_ver_cur():
     global current_curator_index
 
-    # Получаем ключ для выбора шаблона из параметров запроса
     template_key = request.args.get('template_key', 'avp')
-
-    # Выбираем соответствующий шаблон
     mapping_template_cur = mapp_templates.get(template_key, mt_avp)
 
-    # Получаем данные пользователя из параметров запроса
-    user_data = {key: request.args.get(key, "") for key in mapping_template_cur.keys()}
-
-    # Добавляем параметр 'cur' в user_data
-    user_data['cur'] = request.args.get('cur', "")
+    user_data = {mapping_template_cur[key]: request.args.get(key, "") for key in mapping_template_cur}
+    user_data['curator'] = curators[current_curator_index]
 
-    # Верификация номера телефона
-    phone_verification_response = verify_phone_number(user_data['phone'])
+    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, mapping_template_cur)
+        add_or_update_contact(user_data)
         current_curator_index = (current_curator_index + 1) % len(curators)
-        return jsonify({'status': 'success', 'message': f'User added with curator {user_data["cur"]}'})
+        return jsonify({'status': 'success', 'message': f'User added with curator {user_data["curator"]}'})
     except Exception as e:
         logging.error(f"Error adding user: {e}")
         return jsonify({'status': 'error', 'message': str(e)}), 500