blob: 730b23d5af649c9390368e5687c735c76dbd7f45 [file] [log] [blame]
#
# Copyright 2016-present Ciena Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import os
import json
##load the olt config
class OltConfig:
def __init__(self, olt_conf_file = ''):
if not olt_conf_file:
self.olt_conf_file = os.getenv('OLT_CONFIG')
else:
self.olt_conf_file = olt_conf_file
try:
self.olt_handle = open(self.olt_conf_file, 'r')
self.olt_conf = json.load(self.olt_handle)
self.olt_conf['olt'] = True
except:
self.olt_handle = None
self.olt_conf = {}
self.olt_conf['olt'] = False
def on_olt(self):
return self.olt_conf['olt'] is True
def olt_port_map(self):
if self.on_olt() and self.olt_conf.has_key('port_map'):
port_map = {}
if self.olt_conf['port_map'].has_key('ports'):
port_map['ports'] = self.olt_conf['port_map']['ports']
num_ports = len(port_map['ports'])
else:
port_map['ports'] = []
num_ports = int(self.olt_conf['port_map']['num_ports'])
for port in xrange(0, num_ports*2, 2):
port_map['ports'].append('veth{}'.format(port))
##also add dhcprelay ports. We add as many relay ports as subscriber ports
relay_ports = num_ports
port_map['relay_ports'] = []
for port in xrange(relay_ports*2, relay_ports*4, 2):
port_map['relay_ports'].append('veth{}'.format(port))
port_num = 1
port_map['uplink'] = int(self.olt_conf['uplink'])
port_map['wan'] = None
if self.olt_conf.has_key('wan'):
port_map['wan'] = self.olt_conf['wan']
port_list = []
##build the port map and inverse port map
for port in port_map['ports']:
port_map[port_num] = port
port_map[port] = port_num
if port_num != port_map['uplink']:
##create tx,rx map
port_list.append( (port_map['uplink'], port_num ) )
port_num += 1
##build the port and inverse map for relay ports
for port in port_map['relay_ports']:
port_map[port_num] = port
port_map[port] = port_num
port_num += 1
port_map['start_vlan'] = 0
if self.olt_conf['port_map'].has_key('host'):
port_map['host'] = self.olt_conf['port_map']['host']
else:
port_map['host'] = 'ovsbr0'
if self.olt_conf['port_map'].has_key('start_vlan'):
port_map['start_vlan'] = int(self.olt_conf['port_map']['start_vlan'])
return port_map, port_list
else:
return None, None
def olt_device_data(self):
if self.on_olt():
accessDeviceDict = {}
accessDeviceDict['uplink'] = str(self.olt_conf['uplink'])
accessDeviceDict['vlan'] = str(self.olt_conf['vlan'])
return accessDeviceDict
return None