healthchecks_healthchecks/hc/api/management/commands/settelegramwebhook.py
Pēteris Caune 22268c1484
Move absolute URL construction to hc.lib.urls.absolute_reverse()
absolute_reverse() works the same as django.urls.reverse()
except it generates absolute URLs (starting with http[s]://)
2024-12-03 17:24:27 +02:00

32 lines
932 B
Python

from __future__ import annotations
from typing import Any
from django.conf import settings
from django.core.management.base import BaseCommand
from hc.lib import curl
from hc.lib.urls import absolute_reverse
SETWEBHOOK_TMPL = "https://api.telegram.org/bot%s/setWebhook"
class Command(BaseCommand):
help = "Set up telegram bot's webhook address"
def handle(self, **options: Any) -> str:
if settings.TELEGRAM_TOKEN is None:
return "Abort: settings.TELEGRAM_TOKEN is not set"
form = {
"url": absolute_reverse("hc-telegram-webhook"),
"allowed_updates": ["message", "channel_post"],
}
url = SETWEBHOOK_TMPL % settings.TELEGRAM_TOKEN
r = curl.post(url, json=form)
if r.status_code != 200:
return "Fail: status=%d, %s" % (r.status_code, r.content.decode())
return "Done, Telegram's webhook set to: %s" % form["url"]