Update app.py

app.py

@@ -193,72 +193,192 @@ def clean_phone_number_ss(phone_number):
 
 
 
-# Добавляем пользователя с сайта без VK_ID
-@app.route('/wr_user', methods=['GET'])
-def new_wr_user():
-    try:
-        logging.debug("Starting new_wr_user")
-        api_sys_control = request.args.get('api_sys')
+# Добавляем пользователя 
+
+mt_avp = {
+    'name': 'name',
+    'phone': 'phone',
+    'email': 'email',
+    'vk_id': 'vk_id',
+    'utm_campaign': 'utm_campaign',
+    'utm_content': 'utm_content',
+    'utm_medium': 'utm_medium',
+    'utm_source': 'utm_source',
+    'utm_term': 'utm_term',
+    'gcpc': 'gcpc'
+}
 
-        if api_sys_control != api_key_sys:
-            logging.warning("Unauthorized access attempt")
-            return json.dumps({"error": "Unauthorized access"}), 403
+mt_bhelp = {
+    'name': 'name',
+    'phone': 'phone',
+    'email': 'email',
+    'vk_id': 'vk_id',
+    'utm_campaign': 'utm_campaign',
+    'utm_content': 'utm_content',
+    'utm_medium': 'utm_medium',
+    'utm_source': 'utm_source',
+    'utm_term': 'utm_term',
+    'gcpc': 'gcpc'
+}
 
-        name = request.args.get('name', '')
-        email = request.args.get('email', '')
-        phone = request.args.get('phone', '').lstrip('+')
+mt_gc = {
+    'name': 'name',
+    'phone': 'phone',
+    'email': 'email',
+    'vk_id': 'vk_id',
+    'utm_campaign': 'utm_campaign',
+    'utm_content': 'utm_content',
+    'utm_medium': 'utm_medium',
+    'utm_source': 'utm_source',
+    'utm_term': 'utm_term',
+    'gcpc': 'gcpc'
+}
 
-        source = request.args.get('utm_source', '')
-        medium = request.args.get('utm_medium', '')
-        campaign = request.args.get('utm_campaign', '')
-        term = request.args.get('utm_term', '')
-        content = request.args.get('utm_content', '')
-        gcpc = request.args.get('gcpc', '')
-        
+mt_tl = {
+    'name': 'name',
+    'phone': 'phone',
+    'email': 'email',
+    'kol': 'pr1',    
+    'pr2': 'pr2',    
+    'gen_pass': 'pr5',
+    'utm_campaign': 'utm_campaign',
+    'utm_content': 'utm_content',
+    'utm_medium': 'utm_medium',
+    'utm_source': 'utm_source',
+    'utm_term': 'utm_term',
+    'gcpc': 'gcpc'
+}
 
-        if not email or not phone:
-            logging.error("Email and phone are required")
-            return json.dumps({"error": "Email and phone are required"}), 400
+tl_quest = {
+    'name': 'name',
+    'phone': 'phone',
+    'email': 'email',   
+    'pr2': 'pr2',
+    'utm_campaign': 'utm_campaign',
+    'utm_content': 'utm_content',
+    'utm_medium': 'utm_medium',
+    'utm_source': 'utm_source',
+    'utm_term': 'utm_term',
+    'gcpc': 'gcpc'
+}
 
-        phone = clean_phone_number_ss(phone)
+mapp_templates = {
+    'avp': mt_avp,
+    'bhelp': mt_bhelp,
+    'gc': mt_gc,
+    'tilda': mt_tl,
+    'quest': tl_quest
+}
 
-        conn = sqlite3.connect(DATABASE_NEW)
-        cursor = conn.cursor()
+DATABASE_NAME3 = 'data_gc.db'
+def verify_phone_number(phone_number):
+    full_url_ver = f"{wa_url}{wa_ak}{ws_url_ver}{wa_api_key}"
+    payload = {"phoneNumber": phone_number}
+    headers = {'Content-Type': 'application/json'}
+    response = requests.post(full_url_ver, headers=headers, json=payload)
+    if response.status_code == 200:
+        response_body = response.json()
+        return response_body.get('existsWhatsapp', 'false')
+    else:
+        return "false"
 
-        cursor.execute("SELECT * FROM contacts WHERE email = ? OR phone = ?", (email, phone))
-        result = cursor.fetchone()
+def generate_password(length=8):
+    letters_and_digits = string.ascii_letters + string.digits
+    return ''.join(random.choice(letters_and_digits) for i in range(length))
 
-        if result:
-            logging.warning("User already exists")
-            return json.dumps({"error": "User already exists"}), 409
+def add_or_update_contact(contact_data, db_name):
+    conn = sqlite3.connect(db_name)
+    cursor = conn.cursor()
 
-        utc_now = datetime.utcnow()
-        msk_tz = pytz.timezone('Europe/Moscow')
-        msk_now = utc_now.replace(tzinfo=pytz.utc).astimezone(msk_tz)
-        data_on = msk_now.strftime('%Y-%m-%d %H:%M:%S')
+    email = contact_data.get('email')
+    if not email:
+        logging.error(f"Missing email in contact data: {contact_data}")
+        return
 
-        columns_to_insert = ['name', 'phone', 'email', 'data_on']
-        values_to_insert = [name, phone, email, data_on]
+    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')
 
-        query = f"INSERT INTO contacts ({', '.join(columns_to_insert)}) VALUES ({', '.join(['?' for _ in columns_to_insert])})"
-        cursor.execute(query, values_to_insert)
+    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', 'orders', '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'
+    ]
 
-        conn.commit()
+    for field in fields:
+        if field not in contact_data:
+            contact_data[field] = ''
 
-        replace_null_with_empty_string(conn)
+    cursor.execute("SELECT id FROM contacts WHERE email = ?", (email,))
+    contact = cursor.fetchone()
 
-        conn.close()
+    if contact:
+        update_fields = [f"{field} = ?" for field in fields if contact_data[field] != '']
+        update_values = [contact_data[field] for field in fields if contact_data[field] != '']
+        update_values.append(contact[0])
+        update_query = f"UPDATE contacts SET {', '.join(update_fields)} WHERE id = ?"
+        cursor.execute(update_query, update_values)
+    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()
 
-        return json.dumps({"message": "User added successfully"}), 201
 
-    except Exception as e:
-        logging.error(f"An error occurred: {str(e)}")
-        return json.dumps({"error": str(e)}), 500
 
+@app.route('/ver_cur', methods=['GET'])
+def add_data_ver_cur():
+    global current_curator_index
 
+    veref_on_off = request.args.get('ver', '0')
+    curator_on_off = request.args.get('cur', '0')
+    gen_pass_start = request.args.get('pass', '0')
+    db_name = request.args.get('db', 'data_gc.db')  # Получаем имя базы данных из запроса
 
+    template_key = request.args.get('template_key', 'avp')
+    mapping_template_cur = mapp_templates.get(template_key, mt_avp)
 
+    user_data = {mapping_template_cur[key]: request.args.get(key, "") for key in mapping_template_cur}
 
+    logging.debug(f"Received data: {user_data}")
+
+    if curator_on_off == "1":
+        user_data['curator'] = curators[current_curator_index]
+        current_curator_index = (current_curator_index + 1) % len(curators)
+    else:
+        user_data['curator'] = user_data.get('curator', '')
+
+    if veref_on_off == "1":
+        phone_number = user_data.get('phone', '')
+        if not phone_number:
+            logging.error("Phone number is empty")
+            return jsonify({'status': 'error', 'message': 'Phone number is empty'}), 400
+        
+        phone_verification_response = verify_phone_number(phone_number)
+        if phone_verification_response is not None:
+            user_data['ws_st'] = '1' if phone_verification_response else '0'
+    else:
+        user_data['ws_st'] = user_data.get('ws_st', '')
+
+    if gen_pass_start == "1":
+        user_data['pr5'] = generate_password()
+    else:
+        conn = sqlite3.connect(db_name)
+        cursor = conn.cursor()
+        cursor.execute("SELECT pr5 FROM contacts WHERE email = ?", (user_data['email'],))
+        result = cursor.fetchone()
+        if result and result[0]:
+            user_data['pr5'] = result[0]
+        conn.close()
+
+    try:
+        add_or_update_contact(user_data, db_name)
+        return jsonify({'status': 'success', 'message': f'User added with curator {user_data.get("curator", "not assigned")}', 'password': user_data['pr5']})
+    except Exception as e:
+        logging.error(f"Error adding user: {e}")
+        return jsonify({'status': 'error', 'message': str(e)}), 500
 
 
 
@@ -593,16 +713,7 @@ def biz_v():
 # ОБНОВЛЯЕМ CSV-файла
 DATABASE2 = 'data_gc.db'
 
-def verify_phone_number(phone_number):
-    full_url_ver = f"{wa_url}{wa_ak}{ws_url_ver}{wa_api_key}"
-    payload = {"phoneNumber": phone_number}
-    headers = {'Content-Type': 'application/json'}
-    response = requests.post(full_url_ver, headers=headers, json=payload)
-    if response.status_code == 200:
-        response_body = response.json()
-        return response_body.get('existsWhatsapp', 'false')
-    else:
-        return "false"
+
 
 def parse_csv_data(data):
     parsed_data = []