from net_flow import TrustNetwork capacities = [ 20, 7, 2, 1 ] network = TrustNetwork() network.add_edge("-", "el_seed") network.add_edge("el_seed", "test1") network.add_edge("test1", "test2") network.add_edge("test3", "test4") network.add_edge("-", "test4") network.calculate(capacities) for user in network: print "%-10s\t%s" % (user, network.is_auth(user))
# Create edge network.add_edge(from_node, to_node) # Randomly select some nodes to start with starting_nodes = [choice(network) for i in xrange(3)] for node in starting_nodes: network.add_edge('-', node) # Calculate the flow network.calculate(capacities) # Save results to output file if specified if options.output: print 'Saving results to %s' % options.output with open(options.output, 'w') as f: f.writelines( ['%s,%s\n' % (user, network.is_auth(user)) for user in network]) else: # Print results otherwise print "%s\t%s" % ('user', 'is_trusted') for user in network: print "%-10s\t%s" % (user, network.is_auth(user)) # Print stats print """ Total users %d Trusted users %d Untrusted users %d """ % (len(network), len([user for user in network if network.is_auth(user)]), len([user for user in network if not network.is_auth(user)]))
from net_flow import TrustNetwork capacities = [20, 7, 2, 1] network = TrustNetwork() network.add_edge("-", "el_seed") network.add_edge("el_seed", "test1") network.add_edge("test1", "test2") network.add_edge("test3", "test4") network.add_edge("-", "test4") network.calculate(capacities) for user in network: print "%-10s\t%s" % (user, network.is_auth(user))