fix wg.py bug

Toperlock · Oct 30, 2023 · 61a92cb · 61a92cb
1 parent c0aa5dd
commit 61a92cb
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/parsers/wg.py b/parsers/wg.py
@@ -16,15 +16,16 @@ def parse(data):
         'peer_public_key': netquery.get('publicKey')
     }
     if netquery.get('reserved'):
-        _reserved = netquery.get('reserved').split(",")
-        node['reserved'] = [int(_reserved[0]),int(_reserved[1]),int(_reserved[2])]
+        reserved_value = netquery.get('reserved')
+        node['reserved'] = [int(val) for val in reserved_value.split(",")] if ',' in reserved_value else reserved_value
     ip_value = netquery.get('ip')
     if ',' in ip_value:
-        ipv4_value = ip_value.split(",", 1)[0]+"/32"
-        ipv6_value = ip_value.split(",", 1)[1]+"/128"
-        node['local_address'] = [ipv4_value] + [ipv6_value]
+        ipv4_value, ipv6_value = ip_value.split(",", 1)
+        ipv4_value = ipv4_value + "/32" if '/' not in ipv4_value else ipv4_value
+        ipv6_value = ipv6_value + "/128" if '/' not in ipv6_value else ipv6_value
+        node['local_address'] = [ipv4_value, ipv6_value]
     else:
-        ipv4_value = ip_value+"/32"
+        ipv4_value = ip_value + "/32" if '/' not in ip_value else ip_value
         node['local_address'] = [ipv4_value]
     if netquery.get('presharedKey'):
         node['pre_shared_key'] = netquery['presharedKey']