def main():
arguments = docopt(__doc__)
host = [arguments['--host']] if arguments["--host"] else ["localhost"]
session = Cluster(host).connect()
graph = arguments['<keyspace>']
if graph and graph not in session.cluster.metadata.keyspaces:
print "Graph {}{}{} not found".format(Fore.RED, graph, Style.RESET_ALL)
sys.exit(1)
session.default_graph_options.graph_name = graph
print Fore.GREEN + "Connected to {}/{}".format(host[0], graph) + Style.RESET_ALL
accum = None
eof = None
print "Gremlin REPL, use heredocs for multiline ex:<<EOF, help for help\n"
while True:
graph = session.default_graph_options.graph_name
prompt = "gremlin [{}/{}]> ".format(host[0], graph) if eof is None else "gremlin (cont)> "
input = raw_input(prompt)
output_time = False
start_time = time.time()
if input.startswith("<<"):
# heredoc
print "Multiline mode activated"
eof = input[2:]
accum = []
continue
if eof and input == eof:
eof = None
input = "\n".join(accum)
print input
elif eof:
accum.append(input)
continue
if input == "quit" or input == "exit":
break
if input == "%schema":
continue
if input == "help":
print_help()
continue
if input == "docs":
webbrowser.open("http://docs.datastax.com/en/datastax_enterprise/5.0/datastax_enterprise/graph/graphTOC.html#graphTOC__missing-elem-id--graphTOC")
continue
total_time = None
try:
try:
parsed = parse_line(input)
result = parsed.execute(session)
print_result_set(result)
continue
except ParseError as e:
pass
except Exception as e:
print e
continue
stmt = SimpleGraphStatement(input)
if input.startswith("a"):
print Fore.GREEN + "Spark Graph Traversal Enabled, this may take a while..." + Style.RESET_ALL
stmt.options.graph_source = "a"
stmt.options.graph_alias = "a"
start = time.time()
result = session.execute_graph(stmt)
total_time = time.time() - start
except Exception as e:
print e
continue
if isinstance(result, ResultSet):
print_result_set(result)
if total_time:
print Fore.RED + "Query Time: {}s".format(round(total_time, 3)) + Style.RESET_ALL
else:
try:
print "Unknown result", type(result), result
except Exception as e:
print e
def main():
arguments = docopt(__doc__)
host = [arguments['--host']] if arguments["--host"] else ["localhost"]
session = Cluster(host).connect()
graph = arguments['<keyspace>']
if graph and graph not in session.cluster.metadata.keyspaces:
print "Graph {}{}{} not found".format(Fore.RED, graph, Style.RESET_ALL)
sys.exit(1)
session.default_graph_options.graph_name = graph
print Fore.GREEN + "Connected to {}/{}".format(host[0],
graph) + Style.RESET_ALL
accum = None
eof = None
print "Gremlin REPL, use heredocs for multiline ex:<<EOF, help for help\n"
while True:
graph = session.default_graph_options.graph_name
prompt = "gremlin [{}/{}]> ".format(
host[0], graph) if eof is None else "gremlin (cont)> "
input = raw_input(prompt)
output_time = False
start_time = time.time()
if input.startswith("<<"):
# heredoc
print "Multiline mode activated"
eof = input[2:]
accum = []
continue
if eof and input == eof:
eof = None
input = "\n".join(accum)
print input
elif eof:
accum.append(input)
continue
if input == "quit" or input == "exit":
break
if input == "%schema":
continue
if input == "help":
print_help()
continue
if input == "docs":
webbrowser.open(
"http://docs.datastax.com/en/datastax_enterprise/5.0/datastax_enterprise/graph/graphTOC.html#graphTOC__missing-elem-id--graphTOC"
)
continue
total_time = None
try:
try:
parsed = parse_line(input)
result = parsed.execute(session)
print_result_set(result)
continue
except ParseError as e:
pass
except Exception as e:
print e
continue
stmt = SimpleGraphStatement(input)
if input.startswith("a"):
print Fore.GREEN + "Spark Graph Traversal Enabled, this may take a while..." + Style.RESET_ALL
stmt.options.graph_source = "a"
stmt.options.graph_alias = "a"
start = time.time()
result = session.execute_graph(stmt)
total_time = time.time() - start
except Exception as e:
print e
continue
if isinstance(result, ResultSet):
print_result_set(result)
if total_time:
print Fore.RED + "Query Time: {}s".format(round(
total_time, 3)) + Style.RESET_ALL
else:
try:
print "Unknown result", type(result), result
except Exception as e:
print e
