Update app.py

app.py

@@ -156,42 +156,53 @@ def add_data_gc():
         name = request.args.get('name')
         phone = request.args.get('phone')
         email = request.args.get('email')
-        api_sys_control = request.args.get('api_sys')
+        vk_id = request.args.get('vk_id', '')
+        chat_id = request.args.get('chat_id', '')
+        ws_statys = request.args.get('ws_statys', '')
+        ws_stop = request.args.get('ws_stop', '')
+        web_statys = request.args.get('web_statys', 0, type=int)
+        fin_progress = request.args.get('fin_progress', 0, type=int)
+        shop_statys_full = request.args.get('shop_statys_full', '')
+        pr1 = request.args.get('pr1', '')
+        pr2 = request.args.get('pr2', '')
+        pr3 = request.args.get('pr3', '')
+        pr4 = request.args.get('pr4', '')
+        pr5 = request.args.get('pr5', '')
+        ad_url = request.args.get('ad_url', '')
 
         if not name or not phone or not email:
             return "Parameters 'name', 'phone', and 'email' are required.", 400
 
-        if api_sys_control != api_key_sys:
-            return "EUR 22", 200
-
-        # Проверка и удаление плюса в начале номера телефона, если он есть
-        if phone.startswith('+'):
-            phone = phone[1:]
-
         conn = sqlite3.connect('data_gc.db')
         cursor = conn.cursor()
         cursor.execute('SELECT * FROM contacts WHERE phone = ? OR email = ?', (phone, email))
         existing_contact = cursor.fetchone()
 
         if existing_contact:
-            return f"Contact with phone {phone} or email {email} already exists."
+            cursor.execute('''
+                UPDATE contacts SET
+                    name = ?, email = ?, vk_id = ?, chat_id = ?, ws_statys = ?, ws_stop = ?, 
+                    web_statys = ?, fin_progress = ?, shop_statys_full = ?, pr1 = ?, pr2 = ?, 
+                    pr3 = ?, pr4 = ?, pr5 = ?, ad_url = ?
+                WHERE phone = ? OR email = ?
+            ''', (name, email, vk_id, chat_id, ws_statys, ws_stop, web_statys, fin_progress, 
+                  shop_statys_full, pr1, pr2, pr3, pr4, pr5, ad_url, phone, email))
+        else:
+            cursor.execute('''
+                INSERT INTO contacts (
+                    name, phone, email, vk_id, chat_id, ws_statys, ws_stop, web_statys, fin_progress, 
+                    shop_statys_full, pr1, pr2, pr3, pr4, pr5, ad_url
+                ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+            ''', (name, phone, email, vk_id, chat_id, ws_statys, ws_stop, web_statys, fin_progress, 
+                  shop_statys_full, pr1, pr2, pr3, pr4, pr5, ad_url))
 
-        cursor.execute('''
-            INSERT INTO contacts (
-                name, phone, email, vk_id, chat_id, ws_statys, ws_stop, web_statys, fin_progress, 
-                shop_statys_full, curator, shop_statys_cur, partner, pr1, pr2, pr3, pr4, pr5, ad_url
-            )
-            VALUES (?, ?, ?, '', '', '', '', 0, 0, '', '', '', '', '', '', '', '', '', '')
-        ''', (name, phone, email))
         conn.commit()
         conn.close()
 
-        return f"Contact added: {name} - {phone} - {email}", 200
+        return f"Contact updated/added: {name} - {phone} - {email}", 200
     except Exception as e:
-        print(f"Error adding contact: {e}")
+        print(f"Error adding/updating contact: {e}")
         return "Internal Server Error", 500
-
-
         
 
 
@@ -201,29 +212,62 @@ def add_contact():
         name = request.args.get('name')
         phone = request.args.get('phone')
         email = request.args.get('email')
-        api_sys_control = request.args.get('api_sys')
+        vk_id = request.args.get('vk_id', '')
+        chat_id = request.args.get('chat_id', '')
+        ws_statys = request.args.get('ws_statys', '')
+        ws_stop = request.args.get('ws_stop', '')
+        web_statys = request.args.get('web_statys', 0, type=int)
+        fin_progress = request.args.get('fin_progress', 0, type=int)
+        shop_statys_full = request.args.get('shop_statys_full', '')
+        pr1 = request.args.get('pr1', '')
+        pr2 = request.args.get('pr2', '')
+        pr3 = request.args.get('pr3', '')
+        pr4 = request.args.get('pr4', '')
+        pr5 = request.args.get('pr5', '')
+        ad_url = request.args.get('ad_url', '')
 
         if not name or not phone or not email:
             return "Parameters 'name', 'phone', and 'email' are required.", 400
 
-        if api_sys_control != api_key_sys:
-            return "EUR 22", 200
-
-        # Проверка и удаление плюса в начале номера телефона, если он есть
-        if phone.startswith('+'):
-            phone = phone[1:]
-
         conn = sqlite3.connect('data1.db')
         cursor = conn.cursor()
-        cursor.execute('INSERT INTO contacts (name, phone, email) VALUES (?, ?, ?)', (name, phone, email))
+        cursor.execute('SELECT * FROM contacts WHERE phone = ? OR email = ?', (phone, email))
+        existing_contact = cursor.fetchone()
+
+        if existing_contact:
+            cursor.execute('''
+                UPDATE contacts SET
+                    name = ?, email = ?, vk_id = ?, chat_id = ?, ws_statys = ?, ws_stop = ?, 
+                    web_statys = ?, fin_progress = ?, shop_statys_full = ?, pr1 = ?, pr2 = ?, 
+                    pr3 = ?, pr4 = ?, pr5 = ?, ad_url = ?
+                WHERE phone = ? OR email = ?
+            ''', (name, email, vk_id, chat_id, ws_statys, ws_stop, web_statys, fin_progress, 
+                  shop_statys_full, pr1, pr2, pr3, pr4, pr5, ad_url, phone, email))
+        else:
+            cursor.execute('''
+                INSERT INTO contacts (
+                    name, phone, email, vk_id, chat_id, ws_statys, ws_stop, web_statys, fin_progress, 
+                    shop_statys_full, pr1, pr2, pr3, pr4, pr5, ad_url
+                ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+            ''', (name, phone, email, vk_id, chat_id, ws_statys, ws_stop, web_statys, fin_progress, 
+                  shop_statys_full, pr1, pr2, pr3, pr4, pr5, ad_url))
+
         conn.commit()
         conn.close()
 
-        return f"Contact added: {name} - {phone} - {email}", 200
+        return f"Contact updated/added: {name} - {phone} - {email}", 200
     except Exception as e:
-        print(f"Error adding contact: {e}")
+        print(f"Error adding/updating contact: {e}")
         return "Internal Server Error", 500
 
+
+
+
+
+
+
+        
+
 @app.route('/data_gc')
 def show_data_gc():
     try: