65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
import json
|
|
|
|
import requests
|
|
from django.db.models import Q
|
|
from django.http import JsonResponse
|
|
from django.shortcuts import render, get_object_or_404
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
|
|
from cookbook.helper.ingredient_parser import parse, get_unit, get_food
|
|
from cookbook.helper.permission_helper import group_required
|
|
from cookbook.models import TelegramBot, ShoppingList, ShoppingListEntry, Food, Unit
|
|
|
|
|
|
@group_required('user')
|
|
def setup_bot(request, pk):
|
|
bot = get_object_or_404(TelegramBot, pk=pk, space=request.space)
|
|
|
|
hook_url = f'{request.build_absolute_uri("/")}telegram/hook/{bot.webhook_token}/'
|
|
|
|
create_response = requests.get(f'https://api.telegram.org/bot{bot.token}/setWebhook?url={hook_url}')
|
|
info_response = requests.get(f'https://api.telegram.org/bot{bot.token}/getWebhookInfo')
|
|
|
|
return JsonResponse({'hook_url': hook_url, 'create_response': json.loads(create_response.content.decode()), 'info_response': json.loads(info_response.content.decode())}, json_dumps_params={'indent': 4})
|
|
|
|
|
|
@group_required('user')
|
|
def remove_bot(request, pk):
|
|
bot = get_object_or_404(TelegramBot, pk=pk, space=request.space)
|
|
|
|
remove_response = requests.get(f'https://api.telegram.org/bot{bot.token}/deleteWebhook')
|
|
info_response = requests.get(f'https://api.telegram.org/bot{bot.token}/getWebhookInfo')
|
|
|
|
return JsonResponse({'remove_response': json.loads(remove_response.content.decode()), 'info_response': json.loads(info_response.content.decode())}, json_dumps_params={'indent': 4})
|
|
|
|
|
|
@csrf_exempt
|
|
def hook(request, token):
|
|
try:
|
|
tb = get_object_or_404(TelegramBot, webhook_token=token)
|
|
|
|
data = json.loads(request.body.decode())
|
|
|
|
if tb.chat_id == '':
|
|
tb.chat_id = data['message']['chat']['id']
|
|
tb.save()
|
|
|
|
if tb.chat_id == str(data['message']['chat']['id']):
|
|
sl = ShoppingList.objects.filter(Q(created_by=tb.created_by)).filter(finished=False, space=tb.space).order_by('-created_at').first()
|
|
if not sl:
|
|
sl = ShoppingList.objects.create(created_by=tb.created_by, space=tb.space)
|
|
|
|
amount, unit, ingredient, note = parse(data['message']['text'])
|
|
f = get_food(ingredient, tb.space)
|
|
u = get_unit(unit, tb.space)
|
|
sl.entries.add(
|
|
ShoppingListEntry.objects.create(
|
|
food=f, unit=u, amount=amount
|
|
)
|
|
)
|
|
return JsonResponse({'data': data['message']['text']})
|
|
except:
|
|
pass
|
|
|
|
return JsonResponse({})
|