aif-ng/aif/network/networkmanager.py

58 lines
1.8 KiB
Python

import ipaddress
import uuid
##
from . import _common
_NM = _common.NM
class Connection(object):
def __init__(self, iface_xml):
self.xml = iface_xml
self.connection_type = None
self.provider_type = 'NetworkManager'
self.client = _NM.Client.new()
self.addrs = {'ipv4': [],
'ipv6': []}
self.resolvers = []
self.uuid = uuid.uuid4()
self._initAddrs()
self._initResolvers()
def _initAddrs(self):
for t in ('ipv4', 'ipv6'):
for a in self.xml.findall('addresses/{0}/address'.format(t)):
if a.text in ('dhcp', 'dhcp6', 'slaac'):
addr = net = None
else:
components = a.text.split('/')
if len(components) > 2:
raise ValueError('Invalid IP/CIDR format: {0}'.format(a.text))
if len(components) == 1:
addr = components[0]
if t == 'ipv4':
components.append('24')
elif t == 'ipv6':
components.append('64')
addr = ipaddress.ip_address(components[0])
net = ipaddress.ip_network('/'.join(components), strict = False)
self.addrs[t].append((addr, net))
return()
def _initResolvers(self):
for r in self.xml.findall('resolvers/resolver'):
self.resolvers.append(ipaddress.ip_address(r.text))
return()
class Ethernet(Connection):
def __init__(self, iface_xml):
super().__init__(iface_xml)
self.connection_type = 'ethernet'
class Wireless(Connection):
def __init__(self, iface_xml):
super().__init__(iface_xml)
self.connection_type = 'wireless'