mirror of
https://github.com/EDeev/alertbot.git
synced 2026-10-07 20:49:56 +03:00
63c9976888
- токены Prometheus/Alertmanager передаются в заголовке Authorization, а не в адресе; - если Alertmanager не отвечает ALERT_WATCHDOG_FAILURES опросов подряд, бот сообщает «Мониторинг недоступен» и отдельно — когда он снова доступен; - один цикл опроса вынесен в process_alerts(); путь к SQLite — ALERTBOT_DB; - переводы строк приведены к LF.
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
import asyncio
|
|
import logging
|
|
import sys
|
|
|
|
from aiogram.types import BotCommand
|
|
|
|
from init import bot, dp
|
|
from handlers import router, alert_poller
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
handlers=[logging.StreamHandler(sys.stdout),
|
|
logging.FileHandler('bot.log', encoding='utf-8')],
|
|
)
|
|
log = logging.getLogger(__name__)
|
|
|
|
COMMANDS = [
|
|
BotCommand(command="servers", description="CPU / RAM / диск / аптайм по узлам"),
|
|
BotCommand(command="services", description="что работает и что нет"),
|
|
BotCommand(command="certs", description="дни до истечения TLS"),
|
|
BotCommand(command="alerts", description="активные алерты (on/off — подписка)"),
|
|
BotCommand(command="status", description="что включено"),
|
|
BotCommand(command="help", description="справка"),
|
|
]
|
|
|
|
|
|
async def main() -> None:
|
|
log.info("alertbot starting")
|
|
dp.include_router(router)
|
|
await bot.delete_webhook(drop_pending_updates=True)
|
|
try:
|
|
await bot.set_my_commands(COMMANDS)
|
|
except Exception as e:
|
|
log.warning("set_my_commands failed: %s", e)
|
|
|
|
poller = asyncio.create_task(alert_poller())
|
|
try:
|
|
await dp.start_polling(bot, allowed_updates=dp.resolve_used_update_types(),
|
|
timeout=20, relax=0.1)
|
|
finally:
|
|
poller.cancel()
|
|
await bot.session.close()
|
|
log.info("alertbot stopped")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
asyncio.run(main())
|
|
except (KeyboardInterrupt, SystemExit):
|
|
pass
|