Внедрение проекта

restoran_max_bot/max_bot/max_api.py

@@ -264,7 +264,7 @@ def maxbot_send_feedback_button(client: Client, chat_id: str, max_token: str, me
     button = []
     cnt = 0
     for dt in range(1, 6):
-        payload_data = {"payload_type": "feedback", "data": f"feedback{dt}"}
+        payload_data = {"payload_type": "feedback", "data": f"feedback_{dt}"}
         button_data = {"type": "callback", "text": dt, "payload": json.dumps(payload_data)}
         button.append(button_data)
         cnt = cnt + 1
@@ -292,3 +292,44 @@ def maxbot_send_feedback_button(client: Client, chat_id: str, max_token: str, me
     rt = {'success': True, 'error': '', 'data': response}
 
     return rt
+
+
+def maxbot_send_booking_count_people_button(client: Client, chat_id: str, max_token: str, message: str):
+    url = f"https://platform-api.max.ru/messages?chat_id={chat_id}"
+    buttons = []
+    button = []
+    for dt in range(1, 6):
+        payload_data = {"payload_type": "booking_count_people", "data": f"booking_count_people_{dt}"}
+        button_data = {"type": "callback", "text": dt, "payload": json.dumps(payload_data)}
+        button.append(button_data)
+
+    buttons.append(button)
+
+    button = []
+    payload_data = {"payload_type": "booking_count_people", "data": f"booking_count_people_many"}
+    button_data = {"type": "callback", "text": 'Больше 5', "payload": json.dumps(payload_data)}
+    button.append(button_data)
+
+    buttons.append(button)
+
+    payload = json.dumps({
+        "text": message,
+        "attachments": [
+            {"type": "inline_keyboard",
+             "payload": {
+                 "buttons": buttons
+             }
+             }
+        ]
+
+    })
+
+    headers = {
+        'Authorization': f'{max_token}',
+        'Content-Type': 'application/json'
+    }
+
+    response = requests.request("POST", url, headers=headers, data=payload)
+    rt = {'success': True, 'error': '', 'data': response}
+
+    return rt

restoran_max_bot/max_bot/urls.py

@@ -2,7 +2,7 @@ from django.urls import path
 from max_bot.views import api_start_max_v1, iiko_webhook, iiko_send_message
 
 urlpatterns = [
-    path('api/v1', api_start_max_v1),
+    path('api/v1/<str:token>', api_start_max_v1),
     path('iiko/webhook/<str:token>', iiko_webhook),
     path('iiko/send-message/<str:token>', iiko_send_message),
 ]

restoran_max_bot/max_bot/views.py

@@ -61,7 +61,7 @@ def api_decorator(func):
 
 @csrf_exempt
 @api_decorator
-def api_start_max_v1(request, **kwargs):
+def api_start_max_v1(request, token, **kwargs):
     data = json.loads(request.body.decode('utf-8'))
     client = kwargs['client']
     settings_max = kwargs['settings_max']

restoran_max_bot/restoran_max_bot/bot_message.py

@@ -75,7 +75,6 @@ def get_attachments(message: dict):
 
 
 def bot_message(client: Client, message: dict, settings: dict, message_type):
-
     chat_id = message['message']['recipient']['chat_id']
     attachments = get_attachments(message=message)
 
@@ -94,8 +93,8 @@ def bot_message(client: Client, message: dict, settings: dict, message_type):
         data = common_get_data(client=client, key=str(message['message']['body']['text']).lower())
 
     if message_type == 'message_callback':
-        key = json.loads(message['callback']['payload'])
+        payload = json.loads(message['callback']['payload'])
-        data = common_get_data(client=client, key=str(key['data']).lower())
+        data = common_get_data(client=client, key=str(payload['data']).lower())
     if data:
         for dt in data:
             if dt.title == '##bonus##':
@@ -107,12 +106,17 @@ def bot_message(client: Client, message: dict, settings: dict, message_type):
 
                 bonus = get_bonus(client=client, phone=contact.phone, default_bonus=float(dt.price), contact=contact)
                 maxbot_send_text_message(chat_id=chat_id, max_token=settings['token'],
-                                         message=f"Ваши бонусы: {bonus} руб")
+                                         message=f"💰 Ваш бонусный баланс: {bonus} руб")
 
             if dt.title == '##feedback##':
                 maxbot_send_feedback_button(client=client, message='👇', chat_id=chat_id, max_token=settings['token'])
                 return True
 
+            if dt.title == '##booking##':
+                maxbot_send_booking_count_people_button(client=client, message='👇',
+                                                        chat_id=chat_id, max_token=settings['token'])
+                return True
+
             if dt.img:
                 url = f"https://cdn.telefon-ip.ru/{dt.img}?thumb=600"
                 maxbot_send_img_message(chat_id=chat_id, max_token=settings['token'], img=url, message=dt.descr)
@@ -126,6 +130,9 @@ def bot_message(client: Client, message: dict, settings: dict, message_type):
         # отправка меню
         maxbot_send_menu_button(chat_id=chat_id, max_token=settings['token'], message='👇 Выбери раздел', client=client)
     else:
+        if payload['payload_type'] == 'booking_count_people':
+            print('запрос даты')
+
         # отправка меню
         maxbot_send_menu_button(chat_id=chat_id, max_token=settings['token'], message='👇 Выбери раздел', client=client)
 

restoran_max_bot/restoran_max_bot/bot_stopped.py

@@ -3,8 +3,6 @@ from max_bot.models import Client
 
 
 def bot_stopped(client: Client, message: dict, settings: dict):
-    chat_id = message['message']['recipient']['chat_id']
+    chat_id = message['chat_id']
-    chat_type = message['message']['recipient']['chat_type']
+    # maxbot_send_text_message(chat_id=chat_id, max_token=settings['token'], message=text)
-    text = str(message['message']['body']['text']).lower()
-    maxbot_send_text_message(chat_id=chat_id, max_token=settings['token'], message=text)
     return True