64 lines
2.5 KiB
Python
64 lines
2.5 KiB
Python
import json
|
|
import traceback
|
|
|
|
import requests
|
|
from django.http import JsonResponse
|
|
from django.shortcuts import get_object_or_404
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
|
|
from cookbook.helper.ingredient_parser import IngredientParser
|
|
from cookbook.helper.permission_helper import group_required
|
|
from cookbook.models import ShoppingListEntry, TelegramBot
|
|
|
|
|
|
@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']):
|
|
request.space = tb.space # TODO this is likely a bad idea. Verify and test
|
|
request.user = tb.created_by
|
|
ingredient_parser = IngredientParser(request, False)
|
|
amount, unit, food, note = ingredient_parser.parse(data['message']['text'])
|
|
f = ingredient_parser.get_food(food)
|
|
u = ingredient_parser.get_unit(unit)
|
|
|
|
ShoppingListEntry.objects.create(food=f, unit=u, amount=amount, created_by=request.user, space=request.space)
|
|
|
|
return JsonResponse({'data': data['message']['text']})
|
|
except Exception:
|
|
traceback.print_exc()
|
|
|
|
return JsonResponse({})
|