Добавлен параметр конфигурации "consider", отвечающий за фильрацию маршрутов по перечисленным спискам. Мелкие исправления.

download.py

@@ -41,7 +41,7 @@ def ipv4_find(strip:str, size:int):
 # метод сбора словаря ip адресов ipv6 из текста
 def ipv6_find(strip:str, size:int):
     """
-    Метод сбора словаря ip адресов ipv4 из текста
+    Метод сбора словаря ip адресов ipv6 из текста
         возвращает словарь ip
         где:
         ключ - имя сети
@@ -53,7 +53,7 @@ def ipv6_find(strip:str, size:int):
         prefix_str = c.group(2)
         # определяем префикс
         if (prefix:=int(prefix_str) if prefix_str else 128) > size: continue
-        # проверка корректности IPv4
+        # проверка корректности IPv6
         try:
             ipv6_obj=ipaddress.IPv6Address(ip_str)
         except:
@@ -123,7 +123,7 @@ def list_ip(c_list: list = []):
 
         # пробегаем словарь выгрузки
         for c_dict in c_list:
-            # # если есть источник ссылка
+            # если есть источник ссылка
             if 'url' in list(c_dict):
                 # бежим весь список ссылок пока не код 200
                 for c_url in c_dict['url']:
@@ -170,7 +170,7 @@ def list_ip(c_list: list = []):
         if ipv6_list:
             # строим дерево
             Root = net_tree.Node(net_tree.Net(1 << 127, 0, 6))
-            # добавляем IPv4 подсети
+            # добавляем IPv6 подсети
             for ip_int, mask in sorted(ipv6_list.values(), key=lambda x: x[0]):
                 Root.insert(net_tree.Net(ip_int, mask, 6))
             # считаем статистику
@@ -188,6 +188,32 @@ def list_ip(c_list: list = []):
         print(f"Ошибка: {e}")
     return False, False
 
+# метод анализа элементов списка (аргументов)
+def right_list(ip_list: list, args_list: list=[]):
+    """
+    Метод анализа элементов списка выгрузок, возвращает актульный список выгрузок,
+        основываясь на списке аргументов, переданных вторым параметром
+    """
+    # собираем словарь аргументов
+    # элементы списка начинающиеся с "-"
+    # попадают в off, остальные в on
+    c_dict = defaultdict(list)
+    for c in args_list:
+        key = "off" if c[0] == "-" else "on"
+        value = c[1:].upper() if c[0] == "-" else c.upper()
+        c_dict[key].append(value)
+    c_dict = dict(c_dict)
+    # если словарь аргументов не пустой
+    if c_dict:
+        # пробегаем список выгрузок
+        for c in ip_list[:]:
+            # пропускаем
+            if len(c_dict) == 2 and c in c_dict["on"] and not c in c_dict["off"]: continue
+            if len(c_dict) == 1 and (("on" in c_dict and c in c_dict["on"]) or ("off" in c_dict and not c in c_dict["off"])): continue
+            # удаляем элемент из списка
+            ip_list = list(filter(lambda x: x != c, ip_list))
+    return ip_list
+
 # главная фукция
 if __name__ == "__main__":
     # словарь выгружаемых списков
@@ -230,13 +256,8 @@ if __name__ == "__main__":
     # cловари для группировки
     ipv4_dict=dict()
     ipv6_dict=dict()
-    # словарь аргументов вызова
+    # список того, что будем выгружать/обновлять
-    arg_dict = defaultdict(list)
+    download_ip_list=right_list(list(ip_list.keys()), sys.argv[1:])
-    for c in sys.argv[1:]:
-        key = "off" if c[0] == "-" else "on"
-        value = c[1:].upper() if c[0] == "-" else c.upper()
-        arg_dict[key].append(value)
-    arg_dict = dict(arg_dict)
     # создаем дерриктори. для сохранения
     outdir=os.path.join(os.path.dirname(os.path.realpath(__file__)), 'unloading')
     if not os.path.exists(outdir):
@@ -246,18 +267,15 @@ if __name__ == "__main__":
     open(ipv6_bird2_m4 := f"{outdir}/bird2_v6.m4.tmp", "w").close()
     # удаляем старые файлы группировок
     [os.remove(path) for f in os.listdir(outdir) if "__" in f and os.path.isfile(path := os.path.join(outdir, f))]
-    # обходим массив списков для выкрузки
+    # обходим массив списков для выгрузки
     for clist, value in ip_list.items():
         # имена выходых файлов
         ipv4_out_file=f"{outdir}/{clist.lower()}_v4.txt"
         ipv6_out_file=f"{outdir}/{clist.lower()}_v6.txt"
         # извлекаем community
         community=[c for c in value[0].get('community', "").split(",") if c]
-        # если передан аргумент(ы) запуска,
+        # обновляем только указанные списки
-        # значит пытаемся обновить только указанные списки
+        if clist in download_ip_list:
-        if not arg_dict or \
-            (len(arg_dict) == 2 and clist in arg_dict["on"] and not clist in arg_dict["off"]) or \
-            (len(arg_dict) == 1 and (("on" in arg_dict and clist in arg_dict["on"]) or ("off" in arg_dict and not clist in arg_dict["off"]))):
             print("")
             # вычисляем кол-во записей прошлой выгрузки
             ipv4_count_old = sum(1 for line in open(ipv4_out_file)) if os.path.isfile(ipv4_out_file) else 0
@@ -323,19 +341,37 @@ if __name__ == "__main__":
         # список комьюнити маршрутов        
         bgp_community=" ".join([f"bgp_community.add(({str(c).replace(':',',')}));" for c in sorted([c for c in value[0].get('community', "").split(",") if c])])
         if os.path.exists(ipv4_out_file):
+            # фильтер маршрутов ipv4, если список consider существует в конфигурации
+            ip_addresses_filter=list()
+            if (consider:=value[0].get('consider', [])):
+                for c in right_list(list(ip_list.keys()), consider):
+                    if os.path.exists(f_open:=f"{outdir}/{c.lower()}_v4.txt"):
+                        with open(f_open, "r") as file:
+                            ip_addresses_filter.extend([line.strip().split()[1]+"+" for line in file])
             with open(ipv4_bird2_m4, "a") as file:
-                file.write(f"protocol static static_{clist.lower()}_v4 {{\n\tipv4 {{ import filter {{ {bgp_community} accept; }}; }};\n\tinclude \"{ipv4_out_file}\";\n}}\n")
+                bgp_filter=f"if net !~ [{','.join(ip_addresses_filter)}] then reject; " if ip_addresses_filter else ''
+                file.write(f"protocol static static_{clist.lower()}_v4 {{\n\tipv4 {{ import filter {{ {bgp_filter}{bgp_community} accept; }}; }};\n\tinclude \"{ipv4_out_file}\";\n}}\n")
         if os.path.exists(ipv6_out_file):
+            # фильтер маршрутов ipv6, если список consider существует в конфигурации
+            ip_addresses_filter=list()
+            if (consider:=value[0].get('consider', [])):
+                for c in right_list(list(ip_list.keys()), consider):
+                    if os.path.exists(f_open:=f"{outdir}/{c.lower()}_v6.txt"):
+                        with open(f_open, "r") as file:
+                            ip_addresses_filter.extend([line.strip().split()[1]+"+" for line in file])
             with open(ipv6_bird2_m4, "a") as file:
-                file.write(f"protocol static static_{clist.lower()}_v6 {{\n\tipv4 {{ import filter {{ {bgp_community} accept; }}; }};\n\tinclude \"{ipv6_out_file}\";\n}}\n")
+                bgp_filter=f"if net !~ [{','.join(ip_addresses_filter)}] then reject; " if ip_addresses_filter else ''
+                file.write(f"protocol static static_{clist.lower()}_v6 {{\n\tipv6 {{ import filter {{ {bgp_filter}{bgp_community} accept; }}; }};\n\tinclude \"{ipv6_out_file}\";\n}}\n")
 
     # проверяем, что временный файл конфигурации ipv4 не пустой, сохраняем в постоянный
     if os.path.exists(ipv4_bird2_m4) and os.path.getsize(ipv4_bird2_m4) != 0:
         os.replace(ipv4_bird2_m4, ipv4_bird2_m4.removesuffix(".tmp"))
-        os.system("systemctl reload bird.service")
+        os.system("systemctl reload bird.service >/dev/null 2>&1 || \
-        print(f"Новый файл конфигурации ipv4 применён")
+                systemctl restart bird.service >/dev/null 2>&1 && \
+                echo 'Новый файл конфигурации ipv6 применён' || echo '\\e[31mBird2 error...\\e[0m'")
     # проверяем, что временный файл конфигурации ipv6 не пустой, сохраняем в постоянный
     if os.path.exists(ipv6_bird2_m4) and os.path.getsize(ipv6_bird2_m4) != 0:
         os.replace(ipv6_bird2_m4, ipv6_bird2_m4.removesuffix(".tmp"))
-        os.system("systemctl reload bird.service")
+        os.system("systemctl reload bird.service >/dev/null 2>&1 || \
-        print(f"Новый файл конфигурации ipv6 применён")
+                systemctl restart bird.service >/dev/null 2>&1 && \
+                echo 'Новый файл конфигурации ipv6 применён' || echo '\\e[31mBird2 error...\\e[0m'")

list

@@ -5,6 +5,7 @@
             'ipv4': True, # не обязательный аргумент, по умолчанию True (24)
             'ipv6': True, # не обязательный аргумент, по умолчанию False (64)
             'compress': False, # не обязательный аргумент, по умолчанию True
+            'consider': ['CHINA', 'JAPAN'], # только маршруты, которые входят в перечисленный список (работает кроме: -CHINA)
             'community': '65432:LOCATION,65432:200' # не обязательный аргумент, по умолчанию пусто
         },
         # Большая часть RU сегмента