from operator import itemgetter # we can store test data in this module like users users = [ {"name": "invalid_user", "email": "[email protected]", "password": "qwert1235"}, {"name": "valid_user", "email": "[email protected]", "password": "ValidPassword"}, {"name": "Staff2", "email": "[email protected]", "password": "qwert1235"}, {"name": "Admin0", "email": "[email protected]", "password": "qwert1234"}, {"name": "Admin1", "email": "[email protected]", "password": "qwert1234"}, {"name": "Admin2", "email": "[email protected]", "password": "qwert1234"}, ] def get_user(name): try: return (user for user in users if user["name"] == name).next() except: print "\n User %s is not defined, enter a valid user.\n" %name