83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
import asyncio
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.templating import Jinja2Templates
|
|
|
|
from schemas.zabbix_client import ZabbixClient, ZabbixConfig
|
|
from settings import settings
|
|
from utils.background_tasks import check_servers
|
|
from utils.hostgroups import get_hostgroups
|
|
|
|
app = FastAPI()
|
|
zabbix_client: ZabbixClient = None
|
|
|
|
# Static Files
|
|
app.mount("/images", StaticFiles(directory="images"), name="images")
|
|
app.mount("/css", StaticFiles(directory="css"), name="css")
|
|
app.mount("/js", StaticFiles(directory="js"), name="js")
|
|
|
|
templates = Jinja2Templates(directory="templates")
|
|
|
|
|
|
# Cors middleware
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
|
|
async def get_zabbix_client() -> ZabbixClient:
|
|
global zabbix_client
|
|
config = ZabbixConfig(
|
|
api_url=settings.ZABBIX_API_ENDPOINT,
|
|
user=settings.ZABBIX_API_LOGIN,
|
|
password=settings.ZABBIX_API_PASSWORD,
|
|
)
|
|
if zabbix_client and zabbix_client.is_logged_in():
|
|
return zabbix_client
|
|
else:
|
|
client = ZabbixClient(config)
|
|
await client.login()
|
|
return client
|
|
|
|
|
|
async def check_teams_in_hostgroups():
|
|
global zabbix_client
|
|
zabbix_client = await get_zabbix_client()
|
|
hostgroups = await get_hostgroups(zabbix_client)
|
|
all_team_members = [
|
|
member for members in settings.TEAMS.values() for member in members
|
|
]
|
|
for member in all_team_members:
|
|
if member not in hostgroups:
|
|
raise Exception(f"The hostgroup '{member}' does not exist in Zabbix")
|
|
|
|
|
|
@app.on_event("startup")
|
|
async def on_startup():
|
|
global background_task
|
|
|
|
await check_teams_in_hostgroups()
|
|
background_task = asyncio.create_task((check_servers()))
|
|
|
|
|
|
@app.on_event("shutdown")
|
|
async def on_shutdown():
|
|
if background_task:
|
|
background_task.cancel()
|
|
await background_task
|
|
|
|
|
|
@app.get("/healthcheck")
|
|
async def health_check():
|
|
return {"status": "online"}
|
|
|
|
|
|
from routes.alerts import *
|
|
from routes.notes import *
|