server_metrics.py

from argparse import ArgumentParser
from pyVmomi import vim
import socket

from data_retriever.dto import result_message, server_metrics_info, output
from data_retriever.vm_ware_connection import VMwareConnection


def server_metrics(moid: str, ip: str, user: str, password: str, port: int) -> dict:
    """
    Retrieve metrics of a server from the VCenter
    Args:
        moid (str): The Managed Object ID of the server to get metrics from
        ip (str): The ip of the VCenter to connect to
        user (str): The username of the VCenter to connect to
        password (str): The password of the VCenter to connect to
        port (int): The port to use to connect to the VCenter
    Returns:
        dict: A dictionary formatted for json dump containing the server metrics (server_metrics_info()), or an error message (result_message())
    """
    conn = VMwareConnection()
    try:
        conn.connect(ip, user, password, port=port)
        host = conn.get_host_system(moid)
        if not host:
            return result_message("Server not found", 404)

        return server_metrics_info(host)

    except vim.fault.InvalidLogin:
        return result_message("Invalid credentials", 401)
    except (vim.fault.NoCompatibleHost, vim.fault.InvalidHostState, OSError, socket.error):
        return result_message("Host is unreachable", 404)
    except vim.fault.VimFault:
        return result_message(f"Can't get metrics from Server '{moid}'", 403)
    except Exception as err:
        return result_message(str(err), 400)
    finally:
        conn.disconnect()


if __name__ == "__main__":
    parser = ArgumentParser(description="Récupérer les métriques d'un serveur")
    parser.add_argument("--moid", required=True, help="Le Managed Object ID du serveur'")
    parser.add_argument("--ip", required=True, help="Adresse IP du vCenter")
    parser.add_argument("--user", required=True, help="Nom d'utilisateur du vCenter")
    parser.add_argument("--password", required=True, help="Mot de passe de l'utilisateur du vCenter")
    parser.add_argument("--port", type=int, default=443, help="Port du vCenter (optionnel, 443 par défaut)")

    args = parser.parse_args()

    output(server_metrics(args.moid, args.ip, args.user, args.password, args.port))