count = 0 while 1: if count==0 or count==1: choice = raw_input("\n\nChoose:\n0. Exit\n1. Add a user\n") elif count>=2: choice = raw_input("\n\nChoose:\n0. Exit\n1. Add a user\n2. Add a friend link\n3. Check connectivity\n4. Check degree of connection\n") count = count+1 if choice=='0': break elif choice=='1': username=raw_input("Enter a username:\t") b = createUser(username,g,m) if b.makeNode(): print "User created" else: print "User already exists" elif choice=='2': user1,user2 = raw_input("Enter the usernames of each user seperated by a tab:\t").split('\t') c = createFriends(user1,user2,g,m) if c.makeLinks(): print "Friendship link established" else: print "Either user doesn't exist or both names given are the same" elif choice=='3': print "Enter names seperated by Enter key. Press 0 to stop\n\n" l=[] while 1:
from createGraph import createGraph,createUser, createFriends from queryGraph import checkConnectivity, getDegree print "\n\nCreating the schema:\n" a = createGraph() g = a.createGraph() m = a.createMapping() print "\n\nTesting user creation:\n" b = createUser("Immanuel Kant",g,m) if b.makeNode(): print "Node added successfully" else: print "TEST CASE FAILED" b = createUser("Elon Musk",g,m) if b.makeNode(): print "Node added successfully" else: print "TEST CASE FAILED" b = createUser("Voltaire",g,m) if b.makeNode(): print "Node added successfully" else: print "TEST CASE FAILED"