xos/generate/dependency_walker.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.


#!/usr/bin/env python

# NOTE is this used or replaced by `new_base/dependency_walker_new.py`?

import os
import imp
from xosconfig import Config
import inspect
import time
import traceback
import commands
import threading
import json
import pdb
from core.models import *

from xosconfig import Config
from multistructlog import create_logger

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

missing_links = {}

try:
    dep_data = open(Config.get('dependency_graph')).read()
except BaseException:
    raise Exception(
        '[XOS-Dependency-Walker] File %s not found' %
        Config.get('dependency_graph'))

dependencies = json.loads(dep_data)

inv_dependencies = {}
for k, lst in dependencies.items():
    for v in lst:
        try:
            inv_dependencies[v].append(k)
        except KeyError:
            inv_dependencies[v] = [k]


def plural(name):
    if (name.endswith('s')):
        return name + 'es'
    else:
        return name + 's'


def walk_deps(fn, object):
    model = object.__class__.__name__
    try:
        deps = dependencies[model]
    except BaseException:
        deps = []
    return __walk_deps(fn, object, deps)


def walk_inv_deps(fn, object):
    model = object.__class__.__name__
    try:
        deps = inv_dependencies[model]
    except BaseException:
        deps = []
    return __walk_deps(fn, object, deps)


def __walk_deps(fn, object, deps):
    model = object.__class__.__name__
    ret = []
    for dep in deps:
        # print "Checking dep %s"%dep
        peer = None
        link = dep.lower()
        try:
            peer = getattr(object, link)
        except AttributeError:
            link = plural(link)
            try:
                peer = getattr(object, link)
            except AttributeError:
                if model + '.' + link not in missing_links:
                    log.exception(
                        "Model missing link for dependency.",
                        model=model,
                        link=link)
                    missing_links[model + '.' + link] = True

        if (peer):
            try:
                peer_objects = peer.all()
            except AttributeError:
                peer_objects = [peer]
            except BaseException:
                peer_objects = []

            for o in peer_objects:
                # if (isinstance(o,XOSBase)):
                if (hasattr(o, 'updated')):
                    fn(o, object)
                    ret.append(o)
                # Uncomment the following line to enable recursion
                # walk_inv_deps(fn, o)
    return ret


def p(x, source):
    print x, x.__class__.__name__
    return


def main():
    # pdb.set_trace()
    s = Slice.objects.filter(name='princeton_sapan62')
    # pdb.set_trace()
    print walk_inv_deps(p, s[0])


if __name__ == '__main__':
    main()