| Server IP : 66.29.132.122 / Your IP : 3.149.232.152 Web Server : LiteSpeed System : Linux business142.web-hosting.com 4.18.0-553.lve.el8.x86_64 #1 SMP Mon May 27 15:27:34 UTC 2024 x86_64 User : admazpex ( 531) PHP Version : 7.2.34 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : OFF | Pkexec : OFF Directory : /proc/self/root/proc/thread-self/root/opt/imunify360/venv/lib64/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)