Compare commits
No commits in common. "main" and "dev-2.0.1" have entirely different histories.
|
@ -28,8 +28,3 @@
|
|||
10. Адрес для подключения к redis
|
||||
11. Порт для подключения к redis
|
||||
12. Пароль для подключения к redis (если нужен)
|
||||
|
||||
|
||||
UPD
|
||||
Добавлена проверка доступности апи заббикса и алерт в телегу, если не доступен.
|
||||
Добавлено изменнеие алерта на произвольный текст, если сообщение нельзя удалить из-за срока давности.
|
||||
|
|
|
@ -14,7 +14,6 @@ CFG__ZABBIX__ALERT_TEXT_DOWN=Zabbix service DOWN
|
|||
CFG__TGBOT__TOKEN=string
|
||||
CFG__TGBOT__CHAT_ID=00000000
|
||||
CFG__TGBOT__TREAD_ID=0
|
||||
CFG__TGBOT__DELETE_MSG=string
|
||||
|
||||
CFG__REDIS__HOST=localhost
|
||||
CFG__REDIS__PORT=6379
|
||||
|
|
|
@ -42,7 +42,6 @@ class TelegramBotConfig(BaseModel):
|
|||
token: str
|
||||
chat_id: int
|
||||
tread_id: int
|
||||
delete_msg: str
|
||||
|
||||
|
||||
class ZabbixConfig(BaseModel):
|
||||
|
|
|
@ -33,7 +33,6 @@ services:
|
|||
- CFG__TGBOT__TOKEN=string
|
||||
- CFG__TGBOT__CHAT_ID=000000
|
||||
- CFG__TGBOT__TREAD_ID=0
|
||||
- CFG__TGBOT__DELETE_MSG=Deleted
|
||||
|
||||
- CFG__REDIS__HOST=redis
|
||||
- CFG__REDIS__PORT=6379
|
||||
|
|
12
main.py
12
main.py
|
@ -8,13 +8,7 @@ from redis_db import (
|
|||
del_value,
|
||||
)
|
||||
import asyncio
|
||||
from telegram import (
|
||||
del_message,
|
||||
send_message,
|
||||
start_bot,
|
||||
send_zabbix_api_alert,
|
||||
edit_message,
|
||||
)
|
||||
from telegram import del_message, send_message, start_bot, send_zabbix_api_alert
|
||||
|
||||
|
||||
async def dashboard():
|
||||
|
@ -62,10 +56,6 @@ async def dashboard():
|
|||
msg_id = await get_value(closed_alert)
|
||||
if msg_id:
|
||||
resp = await del_message(int(msg_id))
|
||||
if resp["status"] == 200:
|
||||
await del_value(closed_alert)
|
||||
if resp["status"] == 400:
|
||||
resp = await edit_message(int(msg_id), conf.tgbot.delete_msg)
|
||||
if resp["status"] == 200:
|
||||
await del_value(closed_alert)
|
||||
if resp["status"] == 400:
|
||||
|
|
|
@ -1,6 +1,5 @@
|
|||
from .message import (
|
||||
send_message,
|
||||
edit_message,
|
||||
del_message,
|
||||
send_zabbix_api_alert,
|
||||
)
|
||||
|
@ -8,7 +7,6 @@ from .bot import start_bot
|
|||
|
||||
__all__ = [
|
||||
"send_message",
|
||||
"edit_message",
|
||||
"del_message",
|
||||
"start_bot",
|
||||
"send_zabbix_api_alert",
|
||||
|
|
|
@ -45,35 +45,6 @@ async def send_message(message: str, event_id: int) -> dict:
|
|||
return {"status": e}
|
||||
|
||||
|
||||
async def edit_message(message_id: int, message: str) -> dict:
|
||||
url = f"https://api.telegram.org/bot{conf.tgbot.token}/editMessageText"
|
||||
async with aiohttp.ClientSession() as session:
|
||||
try:
|
||||
async with session.post(
|
||||
url,
|
||||
json={
|
||||
"chat_id": conf.tgbot.chat_id,
|
||||
"message_id": message_id,
|
||||
"text": message,
|
||||
},
|
||||
) as response:
|
||||
if response.status == 200:
|
||||
log.info(f"Message ID {message_id} edited")
|
||||
return {
|
||||
"status": response.status,
|
||||
}
|
||||
else:
|
||||
log.warning(
|
||||
f"Message ID {message_id} NOT edit. Response status: {response.status}"
|
||||
)
|
||||
return {
|
||||
"status": response.status,
|
||||
}
|
||||
except Exception as e:
|
||||
log.warning(f"Exception: {e}")
|
||||
return {"status": e}
|
||||
|
||||
|
||||
async def del_message(
|
||||
message_id: int,
|
||||
) -> dict:
|
||||
|
|
|
@ -10,8 +10,8 @@ from datetime import datetime, timedelta
|
|||
|
||||
|
||||
def get_active_problems() -> dict:
|
||||
try:
|
||||
api = ZabbixAPI(url=conf.zabbix.url, token=conf.zabbix.token)
|
||||
try:
|
||||
problems = api.problem.get(
|
||||
output=[
|
||||
"eventid",
|
||||
|
@ -28,7 +28,7 @@ def get_active_problems() -> dict:
|
|||
event_ids.append(problem["eventid"])
|
||||
|
||||
events = api.event.get(
|
||||
selectHosts=["name"],
|
||||
selectHosts=["host"],
|
||||
eventids=event_ids,
|
||||
output=[
|
||||
"eventid",
|
||||
|
@ -38,7 +38,7 @@ def get_active_problems() -> dict:
|
|||
)
|
||||
events_dict = {"event_ids": []}
|
||||
for event in events:
|
||||
event["host"] = event.pop("hosts", None)[0]["name"]
|
||||
event["host"] = event.pop("hosts", None)[0]["host"]
|
||||
events_dict[event["eventid"]] = event
|
||||
events_dict["event_ids"].append(event["eventid"])
|
||||
return events_dict
|
||||
|
|
Loading…
Reference in New Issue