1
0
Fork 0
mirror of https://github.com/EDeev/alertbot.git synced 2026-10-07 20:49:56 +03:00

AlertBot: мониторинг парка серверов через Prometheus/Alertmanager

This commit is contained in:
Egor Deev 2026-09-29 15:04:35 +03:00
commit 1a77cf8620
9 changed files with 657 additions and 0 deletions

17
.env.example Normal file
View file

@ -0,0 +1,17 @@
# Telegram bot token from @BotFather
BOT_TOKEN=123456:your-telegram-bot-token
# Comma-separated Telegram user ids allowed to use the bot
ALLOWED_IDS=111111111,222222222
# Prometheus/Alertmanager endpoints (behind your own token-auth reverse-proxy path)
AM_ALERTS_URL=https://your-domain/ambot/api/v2/alerts
PROM_QUERY_URL=https://your-domain/prombot/api/v1/query
AM_TOKEN=change-me
PROM_TOKEN=change-me
# Node names as used in the "server" label on your Prometheus targets
SERVERS_ORDER=srv1,srv2,srv3
ALERT_POLL_SECONDS=45
ALERT_HTTP_TIMEOUT=15

5
.gitignore vendored Normal file
View file

@ -0,0 +1,5 @@
.env
*.db
__pycache__/
*.pyc
bot.log

21
LICENSE Normal file
View file

@ -0,0 +1,21 @@
MIT License
Copyright (c) 2026 Egor Deev
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.

84
README.md Normal file
View file

@ -0,0 +1,84 @@
# AlertBot
Telegram-бот для мониторинга парка серверов поверх Prometheus + Alertmanager. Работает как
push-уведомитель («что-то сломалось / починилось») и как справочная панель по команде
(`/servers`, `/services`, `/certs`).
## Возможности
- **Алерты в реальном времени** — фоновая задача раз в N секунд опрашивает Alertmanager,
дедуплицирует по fingerprint и сразу шлёт «Проблема · …» / «В норме · …» подписанным
пользователям. Никаких собственных правил — подхватывает всё, что уже настроено в
Alertmanager, по лейблам, а не по именам целей.
- **`/servers`** — CPU, RAM, диск, swap, load, аптайм и (если есть hwmon-датчики)
температура по каждому узлу, с ⚠ при превышении порогов.
- **`/services`** — какие цели Prometheus сейчас `up`, какие нет.
- **`/certs`** — сколько дней осталось у каждого TLS-сертификата, с предупреждением при <14 дней.
- **`/alerts`** — список активных алертов сейчас; `/alerts on|off` — подписка/отписка.
- Доступ — по вайтлисту Telegram user id, все остальные тихо игнорируются.
## Требования
- Python 3.10+
- Свой Prometheus + Alertmanager с уже настроенными правилами алертов.
- Alertmanager и Prometheus должны быть доступны боту по HTTP — либо напрямую (если бот
крутится на той же машине), либо через reverse-proxy с токен-параметром в URL (так это
сделано в этом проекте — см. `AM_ALERTS_URL`/`PROM_QUERY_URL`/`AM_TOKEN`/`PROM_TOKEN`
в `.env.example`), чтобы не открывать сами Prometheus/Alertmanager в интернет без авторизации.
## Установка
```bash
pip install -r requirements.txt
cp .env.example .env
# заполнить .env своими значениями
python bot.py
```
## Переменные окружения
Все — в `.env.example`:
- `BOT_TOKEN` — токен бота от @BotFather.
- `ALLOWED_IDS` — Telegram user id через запятую, кому разрешено пользоваться ботом.
- `AM_ALERTS_URL` / `PROM_QUERY_URL` / `AM_TOKEN` / `PROM_TOKEN` — адреса и токены до
Alertmanager API (`/api/v2/alerts`) и Prometheus API (`/api/v1/query`).
- `SERVERS_ORDER` — имена узлов через запятую, ровно как они указаны в лейбле `server`
у твоих метрик (`node_exporter` и т.п.) — определяет порядок и состав вывода `/servers`.
- `ALERT_POLL_SECONDS` / `ALERT_HTTP_TIMEOUT` — период опроса и таймаут запросов.
## Структура
```
init.py # конфиг из переменных окружения, инициализация Bot/Dispatcher
sql.py # SQLite: подписчики на алерты, дедуп по fingerprint
handlers.py # команды бота + фоновый опрос Alertmanager
bot.py # точка входа
```
Хранилище — один файл SQLite (`notifications.db`, путь по умолчанию задаётся при
инициализации `DatabaseManager`), без внешних зависимостей вроде Redis/Postgres.
## Продакшен
Юнит `systemd` — самый простой способ держать бота в фоне постоянно:
```ini
[Unit]
Description=AlertBot
After=network.target
[Service]
Type=simple
WorkingDirectory=/opt/alertbot
ExecStart=/opt/alertbot/venv/bin/python3 bot.py
Restart=always
RestartSec=10
[Install]
WantedBy=multi-user.target
```
## Лицензия
MIT

51
bot.py Normal file
View file

@ -0,0 +1,51 @@
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

333
handlers.py Normal file
View file

@ -0,0 +1,333 @@
import asyncio
import html
import logging
import os
import time
from typing import Dict, List
import aiohttp
from aiogram import BaseMiddleware, Router
from aiogram.filters import Command, CommandObject
from aiogram.types import Message
from init import (ALERT_HTTP_TIMEOUT, ALERT_POLL_SECONDS, ALLOWED_IDS, AM_ALERTS_URL,
AM_TOKEN, PROM_QUERY_URL, PROM_TOKEN, bot)
from sql import DatabaseManager
router = Router()
db = DatabaseManager()
log = logging.getLogger(__name__)
# Node names as used in Prometheus' "server" label — match your own scrape config.
SERVERS_ORDER = [s for s in os.environ.get("SERVERS_ORDER", "srv1,srv2,srv3").split(",") if s]
class AccessMiddleware(BaseMiddleware):
async def __call__(self, handler, event, data):
u = getattr(event, "from_user", None) or data.get("event_from_user")
if u is None or u.id in ALLOWED_IDS:
return await handler(event, data)
return None # silently ignore everyone else
router.message.outer_middleware(AccessMiddleware())
# ======================================================================
# small helpers
# ======================================================================
def fmt_uptime(sec: float) -> str:
sec = int(max(0, sec))
d, rem = divmod(sec, 86400)
h = rem // 3600
return f"{d}д {h}ч" if d else f"{h}ч"
def settings_block(row) -> str:
_, _u, alerts_on = row
firing = len(db.list_firing_fingerprints())
out = f"Алерты: {'включены' if alerts_on else 'выключены'}"
if firing:
out += f"\nСейчас активных алертов: <b>{firing}</b>"
return out
# ======================================================================
# commands
# ======================================================================
@router.message(Command("start"))
async def cmd_start(msg: Message):
uid = msg.from_user.id
db.add_user(uid, msg.from_user.username or msg.from_user.first_name or str(uid))
row = db.get_user_info(uid)
await msg.answer(
"<b>Мониторинг инфраструктуры</b>\n\n"
"• сразу сообщает о проблемах и о возврате в норму (алерты);\n"
"• по запросу отдаёт состояние серверов, сервисов и сертификатов.\n\n"
+ settings_block(row) + "\n\n"
"Команды — в меню слева от поля ввода, или /help."
)
@router.message(Command("help"))
async def cmd_help(msg: Message):
await msg.answer(
"<b>Команды</b>\n\n"
"<b>Сводки</b>\n"
"/servers — CPU, RAM, диск, аптайм по узлам\n"
"/services — что работает и что нет\n"
"/certs — сколько дней осталось у TLS-сертификатов\n"
"/alerts — активные алерты сейчас\n\n"
"<b>Настройки</b>\n"
"/status — что включено\n"
"/alerts on | off — подписка на алерты"
)
@router.message(Command("status"))
async def cmd_status(msg: Message):
row = db.get_user_info(msg.from_user.id)
if not row:
await msg.answer("Нажми /start.")
return
await msg.answer("<b>Настройки</b>\n\n" + settings_block(row))
@router.message(Command("alerts"))
async def cmd_alerts(msg: Message, command: CommandObject):
uid = msg.from_user.id
if not db.get_user_info(uid):
db.add_user(uid, msg.from_user.username or str(uid))
arg = (command.args or "").strip().lower()
if arg in ("on", "вкл"):
db.set_alerts(uid, True)
await msg.answer("Алерты включены.")
return
if arg in ("off", "выкл"):
db.set_alerts(uid, False)
await msg.answer("Алерты выключены.")
return
await msg.answer(await render_active_alerts())
@router.message(Command("servers"))
async def cmd_servers(msg: Message):
await msg.answer(await render_servers())
@router.message(Command("services"))
async def cmd_services(msg: Message):
await msg.answer(await render_services())
@router.message(Command("certs"))
async def cmd_certs(msg: Message):
await msg.answer(await render_certs())
# ======================================================================
# Prometheus
# ======================================================================
async def _promq(session: aiohttp.ClientSession, query: str) -> List[dict]:
async with session.get(PROM_QUERY_URL, params={"query": query, "token": PROM_TOKEN},
timeout=aiohttp.ClientTimeout(total=ALERT_HTTP_TIMEOUT)) as r:
r.raise_for_status()
j = await r.json()
if j.get("status") != "success":
raise RuntimeError(j.get("error", "prometheus error"))
return j["data"]["result"]
def _by_server(result: List[dict]) -> Dict[str, float]:
out = {}
for s in result:
srv = s["metric"].get("server")
if srv:
try:
out[srv] = float(s["value"][1])
except (ValueError, TypeError):
pass
return out
async def render_servers() -> str:
q = {
"cpu": '100 - (avg by (server) (rate(node_cpu_seconds_total{mode="idle"}[2m])) * 100)',
"ram": '(1 - node_memory_MemAvailable_bytes / node_memory_MemTotal_bytes) * 100',
"disk": '(1 - avg by (server)(node_filesystem_avail_bytes{mountpoint="/",fstype!~"tmpfs|overlay|squashfs"}) '
'/ avg by (server)(node_filesystem_size_bytes{mountpoint="/",fstype!~"tmpfs|overlay|squashfs"})) * 100',
"load": 'node_load1',
"up": 'up{job=~"node_.*"}',
"boot": 'node_boot_time_seconds',
"swap": '(node_memory_SwapTotal_bytes > bool 0) * (1 - node_memory_SwapFree_bytes / (node_memory_SwapTotal_bytes > 0)) * 100',
# real hwmon sensors only (chip=~"pci.*") — excludes bogus acpitz/thermal_zone
# readings some laptops-as-servers report as a stuck fake value
"temp": 'max by (server) (node_hwmon_temp_celsius{chip=~"pci.*"})',
}
try:
async with aiohttp.ClientSession() as s:
res = {k: _by_server(await _promq(s, v)) for k, v in q.items()}
except Exception as e:
return f"Не удалось получить метрики: {e}"
now = time.time()
out = ["<b>Серверы</b>"]
for srv in SERVERS_ORDER:
if srv not in res["up"]:
out.append(f"\n<b>{srv.upper()}</b> — нет данных")
continue
alive = res["up"].get(srv, 0) >= 1
cpu, ram = res["cpu"].get(srv), res["ram"].get(srv)
disk, swap = res["disk"].get(srv), res["swap"].get(srv, 0.0)
load = res["load"].get(srv)
temp = res["temp"].get(srv)
up = fmt_uptime(now - res["boot"][srv]) if srv in res["boot"] else "?"
def g(x):
return f"{x:.0f}%" if isinstance(x, (int, float)) else "?"
warn = []
if isinstance(disk, float) and disk >= 85:
warn.append("диск")
if isinstance(ram, float) and ram >= 90:
warn.append("память")
if isinstance(swap, float) and swap >= 60:
warn.append("swap")
if isinstance(temp, float) and temp >= 85:
warn.append("температура")
head = f"\n<b>{srv.upper()}</b>"
if not alive:
head += " · НЕ ОТВЕЧАЕТ"
elif warn:
head += " · ⚠ " + ", ".join(warn)
out.append(head)
line = f"CPU {g(cpu)} · RAM {g(ram)} · диск {g(disk)} · swap {g(swap)}"
if isinstance(temp, float):
line += f" · temp {temp:.0f}°C"
out.append(line)
out.append(f"load {load:.2f} · аптайм {up}" if isinstance(load, float)
else f"аптайм {up}")
return "\n".join(out)
async def render_services() -> str:
try:
async with aiohttp.ClientSession() as s:
res = await _promq(s, "up")
except Exception as e:
return f"Не удалось получить статус сервисов: {e}"
total = len(res)
down = [(m["metric"].get("job", "?"), m["metric"].get("instance", "?"))
for m in res if m["value"][1] != "1"]
if not down:
return f"<b>Сервисы</b>\n\nВсе {total} в норме."
lines = [f"<b>Сервисы</b>\n\n{total - len(down)}/{total} в норме. Не отвечают:"]
for job, inst in sorted(down):
lines.append(f"• {html.escape(job)} — {html.escape(inst)}")
return "\n".join(lines)
async def render_certs() -> str:
try:
async with aiohttp.ClientSession() as s:
res = await _promq(s, "(probe_ssl_earliest_cert_expiry - time()) / 86400")
except Exception as e:
return f"Не удалось получить данные по сертификатам: {e}"
rows = []
for m in res:
try:
days = float(m["value"][1])
except (ValueError, TypeError):
continue
host = m["metric"].get("instance", "?").replace("https://", "").split("/")[0]
rows.append((days, host))
if not rows:
return "Данных по сертификатам нет."
rows.sort()
soon = [f"{h} — {d:.0f} дн" for d, h in rows if d < 14]
body = "\n".join(f"{d:>4.0f} {h}" for d, h in rows)
head = "<b>TLS-сертификаты</b>"
if soon:
head += "\n⚠ скоро истекают: " + "; ".join(soon)
return head + f"\n<pre>{body}</pre>"
# ======================================================================
# Alertmanager
# ======================================================================
def _sev(labels): return labels.get("severity", "").lower()
def _where(labels): return labels.get("server") or labels.get("instance") or labels.get("job") or ""
def _fmt_firing(a: dict) -> str:
lb, an = a.get("labels", {}), a.get("annotations", {})
name = html.escape(lb.get("alertname", "alert"))
sub = " · ".join(x for x in (_sev(lb), html.escape(_where(lb))) if x)
summ = html.escape(an.get("summary") or an.get("description") or "")
text = f"<b>Проблема · {name}</b>"
if sub:
text += f"\n{sub}"
if summ:
text += f"\n{summ}"
return text
def _fmt_resolved(name: str) -> str:
return f"<b>В норме · {html.escape(name)}</b>"
async def _fetch_alerts(session: aiohttp.ClientSession) -> List[dict]:
params = {"token": AM_TOKEN, "active": "true", "silenced": "false", "inhibited": "false"}
async with session.get(AM_ALERTS_URL, params=params,
timeout=aiohttp.ClientTimeout(total=ALERT_HTTP_TIMEOUT)) as r:
r.raise_for_status()
return await r.json()
async def _broadcast(text: str):
for uid in db.get_alert_users():
try:
await bot.send_message(uid, text)
except Exception as e:
log.error("alert to %s failed: %s", uid, e)
async def render_active_alerts() -> str:
try:
async with aiohttp.ClientSession() as s:
data = await _fetch_alerts(s)
except Exception as e:
return f"Не удалось получить алерты: {e}"
firing = [a for a in data if a.get("status", {}).get("state") == "active"]
if not firing:
return "Активных алертов нет."
return f"<b>Активные алерты: {len(firing)}</b>\n\n" + "\n\n".join(_fmt_firing(a) for a in firing)
async def alert_poller():
log.info("alert poller started (every %ss)", ALERT_POLL_SECONDS)
async with aiohttp.ClientSession() as session:
while True:
try:
data = await _fetch_alerts(session)
current = {}
for a in data:
if a.get("status", {}).get("state") != "active":
continue
fp = a.get("fingerprint")
if not fp:
continue
current[fp] = a
if db.get_alert_status(fp) != "firing":
await _broadcast(_fmt_firing(a))
db.upsert_alert(fp, "firing", a.get("labels", {}).get("alertname", "alert"))
for fp, name in db.list_firing_fingerprints():
if fp not in current:
await _broadcast(_fmt_resolved(name))
db.upsert_alert(fp, "resolved", name)
db.purge_old_resolved()
except asyncio.CancelledError:
break
except Exception as e:
log.warning("alert poll failed: %s", e)
await asyncio.sleep(ALERT_POLL_SECONDS)

25
init.py Normal file
View file

@ -0,0 +1,25 @@
import os
from dotenv import load_dotenv
from aiogram import Bot, Dispatcher
from aiogram.enums.parse_mode import ParseMode
from aiogram.fsm.storage.memory import MemoryStorage
from aiogram.client.bot import DefaultBotProperties
load_dotenv()
BOT_TOKEN = os.environ["BOT_TOKEN"]
# Only these Telegram user ids may use the bot. Add friends' ids here if needed.
ALLOWED_IDS = {int(x) for x in os.environ.get("ALLOWED_IDS", "").split(",") if x.strip()}
# --- Monitoring APIs (pull via token-auth reverse-proxy paths, no basic-auth) ---
AM_ALERTS_URL = os.environ["AM_ALERTS_URL"]
PROM_QUERY_URL = os.environ["PROM_QUERY_URL"]
AM_TOKEN = os.environ["AM_TOKEN"]
PROM_TOKEN = os.environ["PROM_TOKEN"]
ALERT_POLL_SECONDS = int(os.environ.get("ALERT_POLL_SECONDS", "45"))
ALERT_HTTP_TIMEOUT = int(os.environ.get("ALERT_HTTP_TIMEOUT", "15"))
bot = Bot(token=BOT_TOKEN, default=DefaultBotProperties(parse_mode=ParseMode.HTML))
dp = Dispatcher(storage=MemoryStorage())

3
requirements.txt Normal file
View file

@ -0,0 +1,3 @@
aiogram>=3.0
aiohttp
python-dotenv

118
sql.py Normal file
View file

@ -0,0 +1,118 @@
import sqlite3
from typing import List, Optional, Tuple
class DatabaseManager:
def __init__(self, db_path: str = "notifications.db"):
self.db_path = db_path
self.init_database()
def _conn(self):
return sqlite3.connect(self.db_path)
def init_database(self):
with self._conn() as conn:
conn.execute('''
CREATE TABLE IF NOT EXISTS users (
user_id INTEGER PRIMARY KEY,
username TEXT,
alerts_enabled BOOLEAN DEFAULT 1,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
)
''')
conn.execute('''
CREATE TABLE IF NOT EXISTS alert_seen (
fingerprint TEXT PRIMARY KEY,
status TEXT,
name TEXT,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
)
''')
conn.commit()
# ---------- users ----------
def add_user(self, user_id: int, username: str) -> bool:
try:
with self._conn() as conn:
conn.execute(
"INSERT INTO users (user_id, username) VALUES (?, ?) "
"ON CONFLICT(user_id) DO UPDATE SET username=excluded.username",
(user_id, username),
)
conn.commit()
return True
except sqlite3.Error:
return False
def set_alerts(self, user_id: int, value: bool) -> bool:
try:
with self._conn() as conn:
conn.execute("UPDATE users SET alerts_enabled = ? WHERE user_id = ?",
(1 if value else 0, user_id))
conn.commit()
return True
except sqlite3.Error:
return False
def get_user_info(self, user_id: int) -> Optional[Tuple]:
"""(user_id, username, alerts_enabled)"""
try:
with self._conn() as conn:
return conn.execute(
"SELECT user_id, username, alerts_enabled FROM users WHERE user_id = ?",
(user_id,)
).fetchone()
except sqlite3.Error:
return None
def get_alert_users(self) -> List[int]:
try:
with self._conn() as conn:
return [r[0] for r in conn.execute(
"SELECT user_id FROM users WHERE alerts_enabled = 1"
).fetchall()]
except sqlite3.Error:
return []
# ---------- alert dedup ----------
def get_alert_status(self, fingerprint: str) -> Optional[str]:
try:
with self._conn() as conn:
r = conn.execute("SELECT status FROM alert_seen WHERE fingerprint = ?",
(fingerprint,)).fetchone()
return r[0] if r else None
except sqlite3.Error:
return None
def upsert_alert(self, fingerprint: str, status: str, name: str):
try:
with self._conn() as conn:
conn.execute(
"INSERT INTO alert_seen (fingerprint, status, name, updated_at) "
"VALUES (?, ?, ?, CURRENT_TIMESTAMP) "
"ON CONFLICT(fingerprint) DO UPDATE SET status=excluded.status, "
"name=excluded.name, updated_at=CURRENT_TIMESTAMP",
(fingerprint, status, name),
)
conn.commit()
except sqlite3.Error:
pass
def list_firing_fingerprints(self) -> List[Tuple[str, str]]:
try:
with self._conn() as conn:
return conn.execute(
"SELECT fingerprint, name FROM alert_seen WHERE status = 'firing'"
).fetchall()
except sqlite3.Error:
return []
def purge_old_resolved(self, days: int = 3):
try:
with self._conn() as conn:
conn.execute(
"DELETE FROM alert_seen WHERE status = 'resolved' "
"AND updated_at < datetime('now', ?)", (f'-{days} days',))
conn.commit()
except sqlite3.Error:
pass