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))
