From c9ad32e25c01bf5edd780ba285bc271d4a560720 Mon Sep 17 00:00:00 2001 From: Boris Litvinenko Date: Wed, 2 Oct 2024 18:46:47 +0300 Subject: [PATCH] use ipaddress --- yanet-announcer.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/yanet-announcer.py b/yanet-announcer.py index e0d59719..34fb0eea 100755 --- 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)