configurator done

This commit is contained in:
brent s. 2020-05-12 02:55:38 -04:00
parent 3f2d6be7dd
commit c596a4d341
Signed by: bts
GPG Key ID: 8C004C2F93481F6B

View File

@ -1,10 +1,163 @@
import copy
import ipaddress
import os
import re
##
import netaddr
import requests
from lxml import etree


class IP(object):
type = None
version = None
_ip = ipaddress.ip_address
_net = ipaddress.ip_network
_net_ip = netaddr.IPAddress
_net_net = netaddr.IPNetwork

def __init__(self, ip, prefix, *args, **kwargs):
self.str = ip
self.prefix = int(prefix)
self.net_ip = self._net_ip(self.str)
self.net_net = self._net_net('{0}/{1}'.format(self.str, self.prefix))

def _ext_init(self):
self.ip = self._ip(self.str)
self.net = self._net('{0}/{1}'.format(self.str, self.prefix), strict = False)
return(None)


class IP4(IP):
type = 'IPv4'
version = 4
_ip = ipaddress.IPv4Address
_net = ipaddress.IPv4Network

def __init__(self, ip, prefix, *args, **kwargs):
super().__init__(ip, prefix, *args, **kwargs)
self._ext_init()


class IP6(IP):
type = 'IPv6'
version = 6
_ip = ipaddress.IPv6Address
_net = ipaddress.IPv6Network

def __init__(self, ip, prefix, *args, **kwargs):
super().__init__(ip, prefix, *args, **kwargs)
self._ext_init()


class Allocation(object):
def __init__(self, alloc_xml):
self.xml = alloc_xml
self.prefix = None
self.ip = None
self.iface = None
self.parse()

def _iface(self):
_iface_txt = self.xml.attrib['iface']
self.iface = _iface_txt.strip()
return(None)

def _ip(self):
_ip_txt = self.xml.text.strip()
_prefix_txt = self.xml.attrib['prefix'].strip()
self.ip = IP6(_ip_txt, _prefix_txt)
self.prefix = self.ip.prefix
return(None)

def parse(self):
self._iface()
self._ip()
return(None)


class Tunnel(object):
def __init__(self, tun_xml):
self.xml = tun_xml
self.id = None
self.client = None
self.server = None
self.creds = None # This should be handled externally and map to a Cred obj
self.allocations = []
self.parse()

def _allocations(self):
for _allocation_xml in self.xml.findall('allocations'):
self.allocations.append(Allocation(_allocation_xml))
return(None)

def _client(self):
_client_xml = self.xml.find('client')
_ip_txt = _client_xml.text.strip()
_prefix_txt = _client_xml.attrib['prefix'].strip()
self.client = IP6(_ip_txt, _prefix_txt)
return(None)

def _id(self):
self.id = int(self.xml.attrib['id'].strip())
return(None)

def _server(self):
_server_xml = self.xml.find('server')
_ip_text = _server_xml.text.strip()
self.server = IP4(_ip_text, 32)
return(None)

def parse(self):
self._id()
self._client()
self._server()
self._allocations()
return(None)


class Credential(object):
def __init__(self, cred_xml):
self.xml = cred_xml
self.id = None
self.user = None
self.key = None
self.parse()

def _id(self):
_id = self.xml.attrib.get('id')
if not _id:
raise ValueError('Missing required id attribute')
self.id = _id.strip()
return(None)

def _update_key(self):
_key_xml = self.xml.find('updateKey')
if _key_xml is None:
raise ValueError('Missing required updateKey element')
_key_txt = _key_xml.text
if not _key_txt:
raise ValueError('updateKey element is empty')
self.key = _key_txt.strip()
return(None)

def _user(self):
_user_xml = self.xml.find('user')
if _user_xml is None:
raise ValueError('Missing required user element')
_user_txt = _user_xml.text
if not _user_txt:
raise ValueError('user element is empty')
self.user = _user_txt.strip()
return(None)

def parse(self):
self._id()
self._user()
self._update_key()
return(None)


class Config(object):
default_xsd = 'http://schema.xml.r00t2.io/projects/he_ipv6.xsd'

@ -22,10 +175,22 @@ class Config(object):
self.obj = None
self.tunnels = {}
self.creds = {}
self.parse_raw()
self.get_xsd()
self.populate_defaults()
self.validate()
self.parse()

def _creds(self):
creds_xml = self.xml.find('creds')
for cred_xml in creds_xml.findall('cred'):
cred = Credential(cred_xml)
self.creds[cred.id] = cred
return(None)

def _tunnels(self):
tunnels_xml = self.xml.find('tunnels')
for tun_xml in tunnels_xml.findall('tunnel'):
tun = Tunnel(tun_xml)
tun.creds = self.creds.get(tun.creds)
self.tunnels[tun.id] = tun
return(None)

def get_xsd(self):
raw_xsd = None
@ -52,6 +217,14 @@ class Config(object):
self.xsd = etree.XMLSchema(etree.XML(raw_xsd, base_url = base_url))
return(None)

def parse(self):
self.parse_raw()
self.get_xsd()
self.populate_defaults()
self.validate()
self.subparse()
return(None)

def parse_raw(self, parser = None):
if not self.raw:
with open(self.xml_path, 'rb') as fh:
@ -93,6 +266,11 @@ class Config(object):
raise ValueError('Did not know how to parse obj parameter')
return(None)

def subparse(self):
self._creds()
self._tunnels()
return(None)

def validate(self):
if not self.xsd:
self.get_xsd()