class IrisConfigStore:
def __init__(self):
self.objects = ConfigStore.objects
self.templates = ConfigStore.templates
self.specs = ConfigStore.specs
# Custom Database for easy access.
self.trunks = ObjectDatabase()
self.tunnels = ObjectDatabase()
return
def SetTunnels(self, objs):
return self.tunnels.SetAll(objs)
def GetTunnelsVxlan(self):
vxlan_tunnels = []
for tun in self.tunnels.GetAllInList():
if tun.IsVxlan(): vxlan_tunnels.append(tun)
return vxlan_tunnels
def GetTrunkingUplinks(self):
return self.trunks.GetAllInList()
def SetTrunkingUplinks(self, objs):
return self.trunks.SetAll(objs)
class EzAccessStore:
# Static variables and methods
# Batch client
batchClient = None
templates = ConfigStore.templates
specs = ConfigStore.specs
dutNode = 1
configClientDict = dict()
testbedSpec = None
# Class members
def __init__(self, node, parent=None):
self.Node = node
self.NodeObj = parent
self.objects = ConfigStore.objects
# Custom Database for easy access.
self.trunks = ObjectDatabase()
self.dhcprelayObjs = ObjectDatabase()
self.dhcpproxyObjs = ObjectDatabase()
self.tunnels = ObjectDatabase()
self.nexthops = ObjectDatabase()
self.nexthopgroups = ObjectDatabase()
self.ipsec_encrypt_objs = ObjectDatabase()
self.ipsec_decrypt_objs = ObjectDatabase()
self.device = None
self.underlay_vpc = None
self.hostport = None
self.switchport = None
self.securityprofile = None
self.upgrade = None
self.dutNode = 1
self.underlay_ips = []
return
def SetTunnels(self, objs):
if len(objs) == 0: return
if defs.TEST_TYPE == "IOTA":
self.tunnels.db.clear()
return self.tunnels.SetAll(objs)
def SetNexthops(self, objs):
if len(objs) == 0: return
if defs.TEST_TYPE == "IOTA":
self.nexthops.db.clear()
return self.nexthops.SetAll(objs)
def SetNexthopgroups(self, objs):
if len(objs) == 0: return
if defs.TEST_TYPE == "IOTA":
self.nexthopgroups.db.clear()
return self.nexthopgroups.SetAll(objs)
def SetIpsecEncryptSA(self, objs):
if len(objs) == 0: return
if defs.TEST_TYPE == "IOTA":
self.ipsec_encrypt_objs.db.clear()
return self.ipsec_encrypt_objs.SetAll(objs)
def SetIpsecDecryptSA(self, objs):
if len(objs) == 0: return
if defs.TEST_TYPE == "IOTA":
self.ipsec_decrypt_objs.db.clear()
return self.ipsec_encrypt_objs.SetAll(objs)
def SetDevice(self, obj):
self.device = obj
def GetDevice(self):
return self.device
def SetHostPort(self, port):
self.hostport = port
def GetHostPort(self):
return self.hostport
def SetSwitchPort(self, port):
self.switchport = port
def GetSwitchPort(self):
return self.switchport
def SetUnderlayVPC(self, obj):
self.underlay_vpc = obj
def GetUnderlayVPCId(self):
if self.underlay_vpc:
return self.underlay_vpc.VPCId
else:
return -1
def SetSecurityProfile(self, obj):
self.securityprofile = obj
def GetSecurityProfile(self):
return self.securityprofile
def SetUpgrade(self, obj):
self.upgrade = obj
def GetUpgrade(self):
return self.upgrade
def GetProviderIPAddr(self, count):
if self.underlay_vpc:
return self.underlay_vpc.GetProviderIPAddr(count)
else:
return None, -1
def GetSvcMapping(self, ipversion):
if self.underlay_vpc:
return self.underlay_vpc.GetSvcMapping(ipversion)
else:
return None, -1
def IsBitwMode(self):
return self.device.IsBitwMode()
def IsHostMode(self):
return self.device.IsHostMode()
def IsDeviceEncapTypeMPLS(self):
return self.device.IsEncapTypeMPLS()
def IsDeviceEncapTypeVXLAN(self):
return self.device.IsEncapTypeVXLAN()
def IsDeviceLearningEnabled(self):
return self.device.IsLearningEnabled()
def IsBridgingEnabled(self):
return self.device.IsBridgingEnabled()
def IsDeviceOverlayRoutingEnabled(self):
if self.device is None:
return False
return self.device.IsOverlayRoutingEnabled()
def GetDeviceEncapType(self):
return self.device.EncapType
def GetWorkloadTunnels(self):
tunnels = []
for tun in self.tunnels.GetAllInList():
if tun.IsWorkload(): tunnels.append(tun)
return tunnels
def GetIgwNonNatTunnels(self):
tunnels = []
for tun in self.tunnels.GetAllInList():
if tun.IsIgw() and tun.IsNat() is False:
tunnels.append(tun)
return tunnels
def GetIgwNatTunnels(self):
tunnels = []
for tun in self.tunnels.GetAllInList():
if tun.IsIgw() and tun.IsNat(): tunnels.append(tun)
return tunnels
def GetSvcTunnels(self, remote=False):
tunnels = []
for tun in self.tunnels.GetAllInList():
if tun.IsSvc() and tun.Remote is remote:
tunnels.append(tun)
return tunnels
def GetIpsecTunnels(self):
tunnels = []
for tun in self.tunnels.GetAllInList():
if tun.IsIpsec():
tunnels.append(tun)
return tunnels
def GetIpsecTunnelModeTunnels(self):
tunnels = []
for tun in self.tunnels.GetAllInList():
if tun.IsIpsecTunnelMode():
tunnels.append(tun)
return tunnels
def GetIpsecTransportModeTunnels(self):
tunnels = []
for tun in self.tunnels.GetAllInList():
if tun.IsIpsecTransportMode():
tunnels.append(tun)
return tunnels
def GetUnderlayTunnels(self, ecmp=False):
tunnels = []
for tun in self.tunnels.GetAllInList():
if ecmp is False:
if tun.IsUnderlay(): tunnels.append(tun)
elif ecmp is True:
if tun.IsUnderlayEcmp(): tunnels.append(tun)
return tunnels
def GetOverlayTunnels(self):
tunnels = []
for tun in self.tunnels.GetAllInList():
if tun.IsOverlay(): tunnels.append(tun)
return tunnels
def GetUnderlayNexthops(self, ecmp=False):
nhops = []
for nh in self.nexthops.GetAllInList():
if ecmp is False:
if nh.IsUnderlay(): nhops.append(nh)
if ecmp is True:
if nh.IsUnderlayEcmp(): nhops.append(nh)
return nhops
def GetUnderlayNhGroups(self):
nhgs = []
for nhg in self.nexthopgroups.GetAllInList():
if nhg.IsUnderlay():
nhgs.append(nhg)
return nhgs
def GetOverlayNexthops(self, ecmp=False):
nhops = []
for nh in self.nexthops.GetAllInList():
if nh.IsOverlay(): nhops.append(nh)
return nhops
def GetDualEcmpNexthops(self):
nhops = []
for nh in self.nexthops.GetAllInList():
if nh.IsOverlay() and nh.DualEcmp: nhops.append(nh)
return nhops
def GetOverlayNhGroups(self):
nhgs = []
for nhg in self.nexthopgroups.GetAllInList():
if nhg.IsOverlay(): nhgs.append(nhg)
return nhgs
def GetDualEcmpNhGroups(self):
nhgs = []
for nhg in self.nexthopgroups.GetAllInList():
if nhg.IsOverlay() and nhg.DualEcmp: nhgs.append(nhg)
return nhgs
def GetIpsecEncryptSAs(self):
encrypt_sas = []
for obj in self.ipsec_encrypt_objs.GetAllInList():
encrypt_sas.append(obj)
return encrypt_sas
def GetIpsecDecryptSAs(self):
decrypt_sas = []
for obj in self.ipsec_decrypt_objs.GetAllInList():
decrypt_sas.append(obj)
return decrypt_sas
@staticmethod
def SetTestbedSpec(tbspec):
EzAccessStore.testbedSpec = tbspec
@staticmethod
def GetTestbedSpec():
return EzAccessStore.testbedSpec
def GetDhcpRelayObjects(self):
return self.dhcprelayObjs.GetAllInList()
def SetDhcpRelayObjects(self, objs):
return self.dhcprelayObjs.SetAll(objs)
def GetDhcpProxyObjects(self):
return self.dhcpproxyObjs.GetAllInList()
def SetDhcpProxyObjects(self, objs):
return self.dhcpproxyObjs.SetAll(objs)
def GetTrunkingUplinks(self):
return self.trunks.GetAllInList()
def SetTrunkingUplinks(self, objs):
return self.trunks.SetAll(objs)
def GetNodeUuid(self, node):
if node in self.uuid_map:
node_uuid = self.uuid_map[node]
node_uuid = node_uuid.replace('.', '')
if node_uuid == '':
return None
return int(node_uuid, 16)
return None
def SetUnderlayIPs(self, underlay_ips):
self.underlay_ips = underlay_ips
def GetUnderlayIPs(self):
return self.underlay_ips
def GetLoopbackIp(self):
for ip in self.underlay_ips:
if 'Loopback' in ip.Name:
return ip.IP
return None
def GetLoopbackRemoteTEP(self):
for ip in self.underlay_ips:
if 'Loopback' in ip.Name:
return ip.RemoteTEP
return None
def GetUnderlayIp(self, intf_name):
for ip in self.underlay_ips:
if ip.Name == intf_name:
return ip.IP
return None
def GetUnderlayMaskLen(self, intf_name):
for ip in self.underlay_ips:
if ip.Name == intf_name:
return ip.MaskLen
return None
def GetUnderlayBGPRemoteASN(self, intf_name):
for ip in self.underlay_ips:
if ip.Name == intf_name:
return getattr(ip, "BGPRemoteASN", None)
return None
def GetUnderlayNexthop(self, intf_name):
for ip in self.underlay_ips:
if ip.Name == intf_name:
return getattr(ip, "Nexthop", None)
return None
@staticmethod
def SetBatchClient(obj):
EzAccessStore.batchClient = obj
@staticmethod
def GetBatchClient():
return EzAccessStore.batchClient
@staticmethod
def SetDUTNode(node):
EzAccessStore.dutNode = node
@staticmethod
def GetDUTNode():
return EzAccessStore.dutNode
@staticmethod
def SetUuidMap(uuid_map):
EzAccessStore.uuid_map = uuid_map
@staticmethod
def GetUuidMap(node):
return EzAccessStore.uuid_map
@staticmethod
def SetConfigClientDict(obj):
EzAccessStore.configClientDict = obj
return
@staticmethod
def GetConfigClient(objType):
return EzAccessStore.configClientDict.get(objType.name.lower(), None)