Update app.py

app.py

@@ -72,7 +72,7 @@ if not os.path.exists(HTML_FOLDER):
 
 
 
-DATABASES = ['data_gc.db', 'data1.db', 'data2.db', 'data3.db', 'data4.db', 'data5.db']
+DATABASES = ['data_gc.db', 'site_data.db', 'ws_data.db', 'vk_data.db', 'tg_data.db', 'gk_data.db']
 
 
 
@@ -283,7 +283,7 @@ mapp_templates = {
     'quest': tl_quest
 }
 
-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}
@@ -341,8 +341,8 @@ def add_or_update_contact(contact_data, db_name):
 
 
 
-@app.route('/wr_user_ver_cur', methods=['GET'])
-def add_user_ver_cur():
+@app.route('/add_user_home', methods=['GET'])
+def add_user_home():
     global current_curator_index
 
     veref_on_off = request.args.get('ver', '0')
@@ -381,7 +381,45 @@ def add_user_ver_cur():
         logging.error(f"Error adding user: {e}")
         return jsonify({'status': 'error', 'message': str(e)}), 500
 
+@app.route('/add_user_mess', methods=['GET'])
+def add_user_mess():
+    global current_curator_index
+
+    veref_on_off = request.args.get('ver', '0')
+    curator_on_off = request.args.get('cur', '0')
+    db_name = request.args.get('db', 'site_data.db')  # Получаем имя базы данных из запроса
+
+    template_key = request.args.get('player', 'site')
+    mapping_template_cur = mapp_templates.get(template_key, mt_site)
+
+    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', '')
 
+    try:
+        add_or_update_contact(user_data, db_name)
+        return jsonify({'status': 'success', 'message': f'User added {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