Idoit-Tests 01

idoit_scaleup/cat_network.py

@@ -0,0 +1,40 @@
+from .consts import C__CATS__NET
+from .category import IDoitCategory
+from pprint import pprint
+from copy import deepcopy
+from ipaddress import IPv4Network
+from ipaddress import IPv6Network
+
+
+class IDoitNetwork(IDoitCategory):
+
+    CATEGORY = C__CATS__NET
+
+    def __init__(self, cfg):
+        super().__init__(cfg, self.CATEGORY)
+
+    def convert_field_with_name_layer2_assignments(self, data):
+        rtn = []
+        for ele in data['layer2_assignments']:
+            rtn.append(int(ele['id']))
+        return rtn
+
+    def fix_mask_and_range(self, data):
+        net_str=f"{data['address']}/{data['cidr_suffix']}"
+        if int(data['type'])==1:
+            net= IPv4Network(net_str)
+        elif int(data['type'])==1000:
+            net= IPv6Network(net_str)
+        data['netmask']= str(net.netmask)
+        data['range_from'] = str(net[0])
+        data['range_to'] = str(net[-1])
+
+    def save_category(self, objId, data):
+        cdata = deepcopy(data)
+        self.fix_mask_and_range(cdata)
+        return super().save_category(objId, cdata)
+
+    def update_category(self, objId, data):
+        cdata = deepcopy(data)
+        self.fix_mask_and_range(cdata)
+        return super().update_category(objId, cdata)