Update app.py

app.py

@@ -1,6 +1,7 @@
 import asyncio
 import json
 import os
+import urllib.parse
 from datetime import datetime
 from aiogram import Bot, Dispatcher, types, F
 from aiogram.filters import Command
@@ -126,7 +127,7 @@ async def show_products(message: types.Message):
         await message.answer("Нет доступных товаров.")
         return
     for product in data['products']:
-        photo_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/photos/{product['photo']}" if 'photo' in product else None
+        photo_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/photos/{product['photo']}" if product.get('photo') else None
         caption = f"🏷 {product['name']} - {product['price']} руб.\nОписание: {product['description']}\n/id: {product['id']}"
         if photo_url:
             try:
@@ -140,7 +141,7 @@ async def show_products(message: types.Message):
 @dp.message(F.text == "🛒 Корзина")
 async def show_cart(message: types.Message):
     user_id = message.from_user.id
-    cart = next((o for o in data['orders'] if o['user_id'] == user_id and not o['completed']), None)
+    cart = next((o for o in data['orders'] if o['user_id'] == user_id and not o.get('completed')), None)
     if not cart or not cart['items']:
         await message.answer("Ваша корзина пуста.")
         return
@@ -161,9 +162,9 @@ async def add_to_cart(callback_query: types.CallbackQuery):
     product = next((p for p in data['products'] if p['id'] == product_id), None)
     if product:
         user_id = callback_query.from_user.id
-        cart = next((o for o in data['orders'] if o['user_id'] == user_id and not o['completed']), None)
+        cart = next((o for o in data['orders'] if o['user_id'] == user_id and not o.get('completed')), None)
         if not cart:
-            cart = {'user_id': user_id, 'items': [], 'completed': False, 'date': datetime.now().isoformat()}
+            cart = {'user_id': user_id, 'items': [], 'date': datetime.now().isoformat()}
             data['orders'].append(cart)
         cart['items'].append({'product_id': product_id, 'quantity': 1})
         save_data(data)
@@ -171,22 +172,36 @@ async def add_to_cart(callback_query: types.CallbackQuery):
     else:
         await bot.answer_callback_query(callback_query.id, "Товар не найден.")
 
+# Изменённый обработчик "Оформить заказ" – заказ сразу отправляется в WhatsApp и удаляется из корзины
 @dp.callback_query(F.data.startswith("complete_"))
 async def complete_order(callback_query: types.CallbackQuery):
     user_id = int(callback_query.data.split('_')[1])
-    cart = next((o for o in data['orders'] if o['user_id'] == user_id and not o['completed']), None)
-    if cart:
-        cart['completed'] = True
+    cart = next((o for o in data['orders'] if o['user_id'] == user_id and not o.get('completed')), None)
+    if cart and cart['items']:
+        total = 0
+        cart_text = "Привет, я хочу сделать заказ:\n"
+        for item in cart['items']:
+            product = next((p for p in data['products'] if p['id'] == item['product_id']), None)
+            if product:
+                cart_text += f"{product['name']} - {product['price']} руб. x {item['quantity']}\n"
+                total += product['price'] * item['quantity']
+        cart_text += f"\nИтого: {total} руб."
+        encoded_text = urllib.parse.quote(cart_text)
+        whatsapp_link = f"https://wa.me/996500398754?text={encoded_text}"
+        # Удаляем корзину, чтобы заказ не попадал в личный кабинет
+        data['orders'].remove(cart)
         save_data(data)
-        await bot.answer_callback_query(callback_query.id, "Заказ успешно оформлен!")
-        await bot.send_message(user_id, "Спасибо за заказ! Мы скоро свяжемся с вами.")
+        # Отправляем пользователю сообщение с ссылкой на WhatsApp
+        await bot.send_message(user_id, f"Пожалуйста, оформите заказ через WhatsApp, перейдя по ссылке:\n{whatsapp_link}")
+        await bot.answer_callback_query(callback_query.id)
     else:
         await bot.answer_callback_query(callback_query.id, "Корзина пуста или заказ уже оформлен.")
 
 @dp.message(F.text == "📦 Заказы")
 async def show_orders(message: types.Message):
     user_id = message.from_user.id
-    user_orders = [o for o in data['orders'] if o['user_id'] == user_id and o['completed']]
+    # Поскольку оформленные заказы больше не сохраняются, здесь отображаются только те, которые остались в базе (если вдруг)
+    user_orders = [o for o in data['orders'] if o.get('completed')]
     if not user_orders:
         await message.answer("У вас нет оформленных заказо��.")
         return