core-extra/daemon/core/configservices/defaultroute/service.py

47 lines
1.3 KiB
Python

import logging
import netaddr
from core.configservice.base import ConfigService, ConfigServiceMode
from core.emulator.session import Session
from core.nodes.base import CoreNode
from core.nodes.interface import Veth
class DefaultRoute(ConfigService):
name = "DefaultRoute"
group = "Utility"
executables = []
dependencies = []
startup = []
validate = []
shutdown = []
validation_mode = ConfigServiceMode.BLOCKING
def create_files(self):
self.create_default_route()
def create_default_route(self):
addresses = []
for netif in self.node.netifs():
if getattr(netif, "control", False):
continue
for addr in netif.addrlist:
net = netaddr.IPNetwork(addr)
if net[1] != net[-2]:
addresses.append(net[1])
data = dict(addresses=addresses)
self.render("defaultroute.sh", data)
if __name__ == "__main__":
logging.basicConfig(level=logging.INFO)
session = Session(1, mkdir=False)
node = CoreNode(session, _id=1, start=False)
netif = Veth(session, node, "eth0", "eth0", start=False)
netif.addaddr("10.0.0.1/24")
node.addnetif(netif, 0)
service = DefaultRoute(node)
service.create_files()
# data = service.render(node, "defaultroute.sh", dict(addresses=[]))
# print(data)