Update app.py

app.py

@@ -1488,6 +1488,8 @@ def gc_forms():
 
 
 
+
+
 # Глобальная переменная для управления верификацией   
 curators = ["Anna", "Ekaterina", "Ivan", "Maria", "Sergey", "Olga", "Alex", "Natalia", "Dmitry", "Elena"]
 
@@ -1570,8 +1572,8 @@ mapp_templates = {
 
 DATABASE_NAME3 = 'data_gc.db'
 
-def add_or_update_contact(contact_data, db_name):
-    conn = sqlite3.connect(db_name)
+def add_or_update_contact(contact_data):
+    conn = sqlite3.connect(DATABASE_NAME3)
     cursor = conn.cursor()
 
     email = contact_data.get('email')
@@ -1584,12 +1586,14 @@ def add_or_update_contact(contact_data, db_name):
     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] = ''
@@ -1608,16 +1612,15 @@ def add_or_update_contact(contact_data, db_name):
         cursor.execute(insert_query, tuple(contact_data[field] for field in fields))
 
     conn.commit()
+    replace_null_with_empty_string(conn)
     conn.close()
-
 @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')
-    db_name = request.args.get('db', 'data_gc.db')  # Получаем имя базы данных из запроса
-
+    veref_on_off = request.args.get('ver', '0')  # Включает "1" и выключает "0" верификацию номера вместо verifikation_start
+    curator_on_off = request.args.get('cur', '0')  # Включает "1" и выключает "0" назначение куратора
+    
     template_key = request.args.get('template_key', 'avp')
     mapping_template_cur = mapp_templates.get(template_key, mt_avp)
 
@@ -1644,20 +1647,20 @@ def add_data_ver_cur():
         user_data['ws_st'] = user_data.get('ws_st', '')
 
     try:
-        add_or_update_contact(user_data, db_name)
+        add_or_update_contact(user_data)
         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
 
+
 @app.route('/ver_bonus', methods=['GET'])
 def add_data_ver_bonus():
     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', 'data_gc.db')  # Получаем имя базы данных из запроса
-
+    veref_on_off = request.args.get('ver', '0')  # Включает "1" и выключает "0" верификацию номера вместо verifikation_start
+    curator_on_off = request.args.get('cur', '0')  # Включает "1" и выключает "0" назначение куратора
+    
     template_key = request.args.get('template_key', 'avp')
     mapping_template_cur = mapp_templates.get(template_key, mt_avp)
 
@@ -1683,10 +1686,11 @@ def add_data_ver_bonus():
     else:
         user_data['ws_st'] = user_data.get('ws_st', '')
 
+    # Обработка kol для pr1
     email = user_data.get('email')
     new_pr1 = user_data.get('kol')
     if email and new_pr1:
-        conn = sqlite3.connect(db_name)
+        conn = sqlite3.connect(DATABASE_NAME3)
         cursor = conn.cursor()
         cursor.execute("SELECT pr1 FROM contacts WHERE email = ?", (email,))
         result = cursor.fetchone()
@@ -1697,7 +1701,7 @@ def add_data_ver_bonus():
         conn.close()
 
     try:
-        add_or_update_contact(user_data, db_name)
+        add_or_update_contact(user_data)
         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}")