pilot
/
restoran_max_bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Получение контента

Browse Source
main
pilot 3 months ago
parent 24dd7f0af4
commit 928ad79d9e
8 changed files with 192 additions and 18 deletions
Show Stats Download Patch File Download Diff File

71
restoran_max_bot/max_bot/max_api.py
Unescape View File

@ -19,3 +19,74 @@ def maxbot_send_text_message(chat_id: str, max_token: str, message: str):
rt = {'success': True, 'error': '', 'data': response}
return rt
def maxbot_send_img_message(chat_id: str, max_token: str, message: str, img: str):
url = f"https://platform-api.max.ru/messages?chat_id={chat_id}"
payload = json.dumps({
"text": message,
"attachments": [
{"type": "image",
"payload": {"url": img}
}
]
})
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
def maxbot_send_menu_button(chat_id: str, max_token: str, message: str, buton: list):
url = f"https://platform-api.max.ru/messages?chat_id={chat_id}"
payload = json.dumps({
"text": message,
"attachments": [
{"type": "inline_keyboard",
"payload": {
"buttons": [
[
{
"type": "link",
"text": "Откройте сайт",
"url": "https://example.com"
},
{
"type": "link",
"text": "Закройте сайт",
"url": "https://example.com"
}
],
[
{
"type": "link",
"text": "Откройте сайт",
"url": "https://example.com"
},
{
"type": "link",
"text": " 🍺 Закройте сайт",
"url": "https://example.com"
}
]
]
}
}
]
})
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

34
restoran_max_bot/max_bot/models.py
Unescape View File

@ -554,3 +554,37 @@ class MessageTemplate(models.Model):
class Meta:
db_table = 'vi_mess_template'
# Информация продукции
class Product(models.Model):
objects = None
client = models.ForeignKey(Client, on_delete=models.PROTECT)
up = models.IntegerField()
title = models.CharField(max_length=200, null=True, help_text='Наименование продукции')
descr = models.CharField(max_length=200, null=True, help_text='Описание продукции')
img = models.CharField(max_length=200, null=True, help_text='Картинка продукции')
price = models.FloatField(help_text='Стоимотсь продукции')
sku = models.CharField(max_length=100, null=True, help_text='идентификационный код товарной позиции у партнера')
status = models.IntegerField()
class Meta:
db_table = 'lk_product'
# Информация о категории товара
class ProductCategory(models.Model):
objects = None
client = models.ForeignKey(Client, on_delete=models.PROTECT, verbose_name='id client')
emojii = models.CharField(max_length=20, null=True, help_text='Картинка товара')
title = models.CharField(max_length=250, null=True, help_text='Наименование категории')
slug = models.CharField(max_length=250, null=True, help_text='Наименование категории')
descr = models.CharField(max_length=200, null=True, help_text='Описание')
src = models.IntegerField(
help_text='для кого сделана категория 01-1 telegram, 10-2-qrmenu, 11-3 qrmenu and telegram')
status = models.BooleanField(default=0, help_text='отображать или нет категорию')
ord = models.IntegerField(help_text='порядок отображения')
dt = models.DateTimeField(help_text='дата создания')
class Meta:
db_table = 'lk_product_category'

2
restoran_max_bot/max_bot/urls.py
Unescape View File

@ -4,5 +4,5 @@ from django.urls import path
from max_bot.views import api_start_max_v1
urlpatterns = [
path('api/v1/<str:token>', api_start_max_v1),
path('api/v1', api_start_max_v1),
]

51
restoran_max_bot/max_bot/views.py
Unescape View File

@ -1,14 +1,28 @@
import json
from django.http import JsonResponse
from max_bot.max_api import maxbot_send_text_message
from django.views.decorators.csrf import csrf_exempt
from max_bot.models import Partner, Integration, Client
from restoran_max_bot.utils import is_json
from restoran_max_bot.bot_started import bot_started
from restoran_max_bot.bot_stopped import bot_stopped
from restoran_max_bot.bot_message import bot_message
from restoran_max_bot.utils import is_json, has_key
def api_decorator(func):
def wrapper_api_decorator(*args, **kwargs):
if 'X-Max-Bot-Api-Secret' in args[0].headers:
token = args[0].headers['X-Max-Bot-Api-Secret']
token = '71232b71-dacc-4586-95d0-0f352d9d09c3'
else:
rt = {'success': False, 'error': 'Unauthorized token', 'data': {}}
return JsonResponse(rt, status=401)
if not is_json(args[0].body.decode('utf-8')):
rt = {'success': False, 'error': 'data not JSON', 'data': ''}
return JsonResponse(rt, status=200)
# получаем информацию по токену клиента и делаем проверку
client = Client.objects.filter(token=kwargs['token']).first()
client = Client.objects.filter(token=token).first()
if not client:
rt = {'success': False, 'error': 'Unauthorized token', 'data': {}}
return JsonResponse(rt, status=401)
@ -30,6 +44,7 @@ def api_decorator(func):
return JsonResponse(rt, status=404)
settings_max = json.loads(integration.setting)
settings_max['token'] = integration.token
kwargs['client'] = client
kwargs['integration'] = integration
@ -40,26 +55,28 @@ def api_decorator(func):
return wrapper_api_decorator
@csrf_exempt
@api_decorator
def api_start_max_v1(request, token, **kwargs):
def api_start_max_v1(request, **kwargs):
data = json.loads(request.body.decode('utf-8'))
print(data)
client = kwargs['client']
settings_max = kwargs['settings_max']
# получение сообщения
if not has_key(data, 'update_type'):
rt = {'success': False, 'error': 'data update_type not found', 'data': ''}
return JsonResponse(rt, status=200)
# получение сообщения из бота
if data['update_type'] == 'message_created':
print(data)
chat_id = data['message']['recipient']['chat_id']
chat_type = data['message']['recipient']['chat_type']
text = str(data['message']['body']['text']).lower()
maxbot_send_text_message(chat_id=chat_id, max_token=settings_max['token'], message=text)
# добавление бота в группу
if data['update_type'] == 'bot_added':
chat_id = data['chat_id']
maxbot_send_text_message(chat_id=chat_id, max_token=settings_max['token'], message='bot_added')
bot_message(client=client, message=data, settings=settings_max)
# start - запуск бота
if data['update_type'] == 'bot_started':
bot_started(client=client, message=data, settings=settings_max)
# выход из бота
if data['update_type'] == 'bot_stopped':
bot_stopped(client=client, message=data, settings=settings_max)
rt = {'success': True, 'error': '', 'data': ''}
return JsonResponse(rt, status=200)

21
restoran_max_bot/restoran_max_bot/bot_message.py
Unescape View File

@ -0,0 +1,21 @@
from max_bot.max_api import maxbot_send_text_message, maxbot_send_img_message, maxbot_send_menu_button
from max_bot.models import Client
from restoran_max_bot.common import common_get_data
def bot_message(client: Client, message: dict, settings: dict):
chat_id = message['message']['recipient']['chat_id']
chat_type = message['message']['recipient']['chat_type']
text = str(message['message']['body']['text']).lower()
data = common_get_data(client=client, key=text)
if data:
for dt in data:
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)
else:
maxbot_send_text_message(chat_id=chat_id, max_token=settings['token'], message=dt.descr)
# отправка меню
rs = maxbot_send_menu_button(chat_id=chat_id, max_token=settings['token'], message='test', buton=[])
print(rs['data'].text)
return True

10
restoran_max_bot/restoran_max_bot/bot_started.py
Unescape View File

@ -0,0 +1,10 @@
from max_bot.max_api import maxbot_send_text_message
from max_bot.models import Client
def bot_started(client: Client, message: dict, settings: dict):
chat_id = message['message']['recipient']['chat_id']
chat_type = message['message']['recipient']['chat_type']
text = str(message['message']['body']['text']).lower()
maxbot_send_text_message(chat_id=chat_id, max_token=settings['token'], message=text)
return True

10
restoran_max_bot/restoran_max_bot/bot_stopped.py
Unescape View File

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

11
restoran_max_bot/restoran_max_bot/common.py
Unescape View File

@ -0,0 +1,11 @@
# функция поиска контента по ключевому слову
from max_bot.models import Client, ProductCategory, Product
def common_get_data(client: Client, key: str) -> Product:
product_category = ProductCategory.objects.filter(client=client, status=1, title=key, src=1).first()
if product_category:
data = Product.objects.filter(up=product_category.pk, status=1)
else:
data = None
return data
Write Preview
Loading…
Cancel
Save