77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from typing import Self
|
|
from Abstracts.ControllerAbstract import ControllerAbstract
|
|
from Models.ResponseModel import ResponseModel
|
|
from Models.RequestModel import RequestModel
|
|
from Models.IterationModel import (
|
|
CPUModel, MemoryModel, NetUseModel, InterfaceModel, DiskModel,
|
|
CandleTimeModel, IterationModel
|
|
)
|
|
from Dispatchers.AgentsDispatcher import AgentsDispatcher
|
|
from Utils.Utils import Utils
|
|
|
|
class AgentsController(ControllerAbstract):
|
|
|
|
def debian(self:Self, request:RequestModel, response:ResponseModel) -> None:
|
|
|
|
key:str = request.get("key")
|
|
hostnames:list[str]
|
|
domain:str|None
|
|
interfaces_data:list[list[int, str, bool, str, int]]
|
|
disks:list[list[str, int, int, str|None]]
|
|
iterations:int
|
|
candle_times:list[int, int]
|
|
cpu:list[float, float, float, float, float]
|
|
memory:list[int, int, int, int, int, float]
|
|
net_use:list[list[list[str, int, int, int, int, int, int]]]
|
|
received:list[NetUseModel] = []
|
|
transmitted:list[NetUseModel] = []
|
|
|
|
(
|
|
hostnames, domain, interfaces_data, disks,
|
|
iterations, candle_times, cpu, memory, net_use
|
|
) = Utils.json_decode(request.body)
|
|
|
|
print(Utils.json_decode(request.body))
|
|
|
|
received = {net[0] : NetUseModel(*net[1:]) for i, net in enumerate(net_use[0])}
|
|
transmitted = {net[0] : NetUseModel(*net[1:]) for i, net in enumerate(net_use[1])}
|
|
|
|
self.nucelar_monitor.dispatchers.get(AgentsDispatcher, "sql_server").save(IterationModel(
|
|
key, hostnames, domain,
|
|
[InterfaceModel(
|
|
i, name, is_ipv6, ip, mask, received[name], transmitted[name]
|
|
) for i, name, is_ipv6, ip, mask in interfaces_data],
|
|
[DiskModel(*disk) for disk in disks],
|
|
iterations,
|
|
CandleTimeModel(*candle_times),
|
|
CPUModel(*cpu),
|
|
MemoryModel(*memory)
|
|
))
|
|
|
|
def windows(self:Self, request:RequestModel, response:ResponseModel) -> None:
|
|
|
|
key:str = request.get("key")
|
|
hostnames:list[str]
|
|
domain:str|None
|
|
interfaces:list[str]
|
|
disks:list[list[str, int, int]]
|
|
iterations:int
|
|
candle_times:list[int, int]
|
|
cpu:list[float, float, float, float, float]
|
|
memory:list[int, int, int, int, int, float]
|
|
net_use:list[list[list[str, int, int, int, int, int, int]]] = []
|
|
|
|
print(Utils.json_decode(request.body))
|
|
|
|
hostnames, domain, interfaces, disks, candle_times, iterations, cpu, memory = Utils.json_decode(request.body)
|
|
|
|
print([hostnames, domain, interfaces, disks, iterations, candle_times, cpu, memory, net_use])
|
|
|
|
def test(self:Self, request:RequestModel, response:ResponseModel) -> None:
|
|
response.set({
|
|
"message": "Test successful",
|
|
"key" : request.get("key")
|
|
}, "application/json") |