| Server IP : 192.158.238.246 / Your IP : 3.147.103.209 Web Server : LiteSpeed System : Linux uniform.iwebfusion.net 4.18.0-553.27.1.lve.1.el8.x86_64 #1 SMP Wed Nov 20 15:58:00 UTC 2024 x86_64 User : jenniferflocom ( 1321) PHP Version : 8.1.32 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : OFF | Pkexec : OFF Directory : /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/ |
Upload File : |
"""JSON encoders to help with sending messages to server."""
import json
from ipaddress import IPv4Address, IPv4Network, IPv6Address, IPv6Network
from playhouse.shortcuts import model_to_dict
from defence360agent.model import Model
def ip_net_to_string(net) -> str:
"""
IPv4Network('192.168.1.1/32') -> '192.168.1.1'
IPv4Network('192.168.1.0/24') -> '192.168.1.0/24'
"""
if not int(net.hostmask):
return str(net.network_address)
return str(net)
class IPEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, (IPv4Network, IPv6Network)):
return ip_net_to_string(obj)
if isinstance(obj, (IPv4Address, IPv6Address)):
return str(obj)
return json.JSONEncoder.default(self, obj)
class ServerJSONEncoder(IPEncoder):
def default(self, obj):
if isinstance(obj, Model):
return model_to_dict(obj)
return super().default(obj)