def addEdge()

in scenario-notebooks/UserSecurityMetadata/GraphVis.py [0:0]

11 lines of code
9 McCabe index (conditional complexity)


    def addEdge(self, fromNode, toNode, label):
        if fromNode.group != "User" and fromNode.group != "Group" and fromNode.group != "AzureSubscription" and fromNode.group != "ServicePrincipal":
            raise Exception("Error: Unsupported node type - " +
                            jsonpickle.encode(fromNode, unpicklable=False, make_refs=False))
        if toNode.group != "User" and toNode.group != "Group" and toNode.group != "AzureSubscription" and toNode.group != "ServicePrincipal":
            raise Exception("Error: Unsupported node type - " +
                            jsonpickle.encode(toNode, unpicklable=False, make_refs=False))
        self.nodes.add(fromNode)
        self.nodes.add(toNode)
        edge = Edge(fromNode.id, toNode.id, label)
        self.edges.add(edge)