telegram_bots/bots/management/commands/run_bots.py
2019-11-12 20:13:56 +03:00

54 lines
1.9 KiB
Python

import logging
import traceback
import sentry_sdk
from django.core.cache import cache
from django.core.management import BaseCommand
from telegram import TelegramError
from telegram.error import TimedOut
from bots.models import TelegramBot
class Command(BaseCommand):
def handle(self, *args, **options):
dispatchers = []
while True:
try:
if not dispatchers or cache.get('bots_reset'):
logging.warning('Reloading dispatchers')
dispatchers = []
for bot in TelegramBot.objects.filter(active=True):
try:
dispatcher = bot.build_dispatcher()
dispatcher.last_update_id = 0
dispatchers.append(dispatcher)
except TelegramError:
pass
cache.delete('bots_reset')
for dispatcher in dispatchers:
try:
updates = dispatcher.bot.get_updates(dispatcher.last_update_id)
except TimedOut:
continue
except TelegramError as e:
sentry_sdk.capture_exception(e)
traceback.print_exc()
updates = []
for update in updates:
try:
dispatcher.process_update(update)
except KeyboardInterrupt:
return
except Exception as e:
sentry_sdk.capture_exception(e)
traceback.print_exc()
dispatcher.last_update_id = update.update_id + 1
except KeyboardInterrupt:
return
except Exception as e:
sentry_sdk.capture_exception(e)
traceback.print_exc()