use ipaddress

yanet-platform · Oct 3, 2024 · 16a5443 · 16a5443
1 parent 9459d23
commit 16a5443
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/yanet-announcer.py b/yanet-announcer.py
@@ -7,6 +7,7 @@
 import logging
 import signal
 import subprocess
+import ipaddress
 import textwrap
 import time
 import typing
@@ -149,7 +150,8 @@ def bgp_remove_ipv6(prefix):
 def bgp_update(prefix_list):
     for prefix in prefix_list:
         try:
-            if ":" in prefix:
+            parsed = ipaddress.ip_network(prefix)
+            if parsed.version == 6:
                 bgp_update_ipv6(prefix)
             else:
                 bgp_update_ipv4(prefix)
@@ -160,7 +162,8 @@ def bgp_update(prefix_list):
 def bgp_remove(prefix_list):
     for prefix in prefix_list:
         try:
-            if ":" in prefix:
+            parsed = ipaddress.ip_network(prefix)
+            if parsed.version == 6:
                 bgp_remove_ipv6(prefix)
             else:
                 bgp_remove_ipv4(prefix)