xos/synchronizer/pull_steps/pull_onos_devices.py

# Copyright 2017-present Open Networking Foundation
#
# 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.

from xossynchronizer.pull_steps.pullstep import PullStep

from xosconfig import Config
from multistructlog import create_logger

import requests
from requests import ConnectionError
from requests.auth import HTTPBasicAuth
from requests.models import InvalidURL

import os, sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from helpers import Helpers

log = create_logger(Config().get('logging'))

class ONOSDevicePullStep(PullStep):
    def __init__(self, model_accessor):
        super(ONOSDevicePullStep, self).__init__(model_accessor=model_accessor)

    def get_onos_fabric_service(self, fabric):
        fabric_service = self.model_accessor.Service.objects.get(id=fabric.id)
        onos_fabric_service = fabric_service.provider_services[0].leaf_model
        return onos_fabric_service

    def pull_records(self):
        log.debug("[ONOS device pull step] pulling devices from ONOS")

        fabric_service=self.model_accessor.FabricService.objects.all()
        for fabric in fabric_service:
            onos = self.get_onos_fabric_service(fabric)
            log.info("onos_fabric: %s" % onos)

            url = 'http://%s:%s/onos/v1/devices/' % (onos.rest_hostname, onos.rest_port)
            r = requests.get(url, auth=HTTPBasicAuth(onos.rest_username, onos.rest_password))
            if r.status_code != 200:
                log.error(r.text)
                raise Exception("Failed to get onos devices")
            else:
                try:
                    log.info("Get devices response", json=r.json())
                except Exception:
                    log.info("Get devices exception response", text=r.text)

            for device in r.json()["devices"]:
                if device["type"] != "SWITCH":
                    continue

                xos_devices = self.model_accessor.Switch.objects.filter(ofId = device["id"])
                if not xos_devices:
                    continue

                xos_device = xos_devices[0]
                changed = False

                managementAddress = device.get("annotations", {}).get("managementAddress")
                if (xos_device.managementAddress != managementAddress):
                    log.info("Setting managementAddress on switch %s to %s" % (xos_device.id, managementAddress))
                    xos_device.managementAddress = managementAddress
                    changed = True

                if changed:
                    xos_device.save_changed_fields()