Greatly improving GUI Log Messages by coloring and adding object Netb…

…ox linking

netbox_proxbox/backend/exception.py

@@ -1,5 +1,6 @@
-from netbox_proxbox.backend.logging import logger
 
+from fastapi import WebSocket
+
 class ProxboxException(Exception):
     def __init__(
         self,
@@ -18,8 +19,14 @@ def __init__(
 
         if self.python_exception:
             log_message+=f"\n > Python Exception: {self.python_exception}"
-
-
-        logger.error(log_message)
+
 
-
+async def exception_log(
+    logger,
+    message: str,
+    detail: str | None = None,
+    python_exception: str | None = None,
+    websocket: WebSocket | None = None,
+):
+    log = logger
+    await log(websocket=websocket, msg=message, level="ERROR")

netbox_proxbox/backend/logging.py

@@ -1,5 +1,6 @@
 import logging
 from logging.handlers import TimedRotatingFileHandler
+from netbox_proxbox.backend.exception import ProxboxException
 
 # ANSI escape sequences for colors
 class AnsiColorCodes:
@@ -70,4 +71,21 @@ def setup_logger():
     return logger
 
 
-logger = setup_logger()
+logger = setup_logger()
+
+from fastapi import WebSocket
+
+async def log(websocket: WebSocket, msg, level = None):
+    if websocket: 
+        await websocket.send_text(msg)
+
+    if level == "debug": logger.debug(msg)
+
+    if level == "ERROR" or level == "error":
+        logger.error(msg)
+        #raise ProxboxException(
+        #    message=msg,
+        #    python_exception=python_exception
+        #)
+
+    else: logger.info(msg)

netbox_proxbox/backend/routes/netbox/dcim/device_types.py

@@ -15,7 +15,7 @@ class DeviceType(NetboxBase):
     object_name = "Device Types"
 
     async def get_base_dict(self):
-        manufacturer = await Manufacturer(nb = self.nb).get()
+        manufacturer = await Manufacturer(nb = self.nb, websocket = self.websocket).get()
 
         return {
             "model": self.default_name,

netbox_proxbox/backend/routes/netbox/dcim/devices.py

@@ -18,10 +18,10 @@ class Device(NetboxBase):
     object_name = "Device"
 
     async def get_base_dict(self):
-        site = await Site(nb = self.nb).get()
-        role = await DeviceRole(nb = self.nb).get()
-        device_type = await DeviceType(nb = self.nb).get()
-        cluster = await Cluster(nb = self.nb).get()
+        site = await Site(nb = self.nb, websocket = self.websocket).get()
+        role = await DeviceRole(nb = self.nb, websocket = self.websocket).get()
+        device_type = await DeviceType(nb = self.nb, websocket = self.websocket).get()
+        cluster = await Cluster(nb = self.nb, websocket = self.websocket).get()
 
         return {
             "name": self.default_name,

netbox_proxbox/backend/routes/netbox/dcim/interfaces.py

@@ -19,7 +19,7 @@ class Interface(NetboxBase):
     object_name = "Interface"
 
     async def get_base_dict(self):
-        device = await Device(nb = self.nb).get()
+        device = await Device(nb = self.nb, websocket = self.websocket).get()
 
         return {
             "device": device.id,

netbox_proxbox/backend/routes/netbox/virtualization/cluster.py

@@ -15,7 +15,7 @@ class Cluster(NetboxBase):
 
 
     async def get_base_dict(self):
-        type = await ClusterType(nb = self.nb).get()
+        type = await ClusterType(nb = self.nb, websocket = self.websocket).get()
 
         return {
             "name": self.default_name,

netbox_proxbox/backend/routes/netbox/virtualization/interfaces.py

@@ -15,7 +15,7 @@ class VMInterface(NetboxBase):
 
     async def get_base_dict(self):
 
-        virtual_machine = await VirtualMachine(nb = self.nb).get()
+        virtual_machine = await VirtualMachine(nb = self.nb, websocket = self.websocket).get()
 
         return {
             "virtual_machine": virtual_machine.id,

netbox_proxbox/backend/routes/netbox/virtualization/virtual_machines.py

@@ -13,7 +13,7 @@ class VirtualMachine(NetboxBase):
 
 
     async def get_base_dict(self):
-        cluster = await Cluster(nb = self.nb).get()
+        cluster = await Cluster(nb = self.nb, websocket = self.websocket).get()
         return {
             "name": self.default_name,
             "slug": self.default_slug,

netbox_proxbox/templates/netbox_proxbox/home.html

@@ -16,8 +16,11 @@ <h1>Log Messages</h1>
         ws.onmessage = function(event) {
             var messages = document.getElementById('messages')
             var message = document.createElement('li')
-            var content = document.createTextNode(event.data)
-            message.appendChild(content)
+
+            message.innerHTML = event.data
+
+            // var content = document.createTextNode(event.data)
+            // message.appendChild(content)
             messages.appendChild(message)
         };
         function sendMessage(event) {