def main(argv: Optional[List[str]] = None) -> int: if argv is None: argv = sys.argv[1:] parser = argparse.ArgumentParser() parser.add_argument("query_file", type=str) parser.add_argument("--graph_names", type=str, default=["alti"], nargs="+") parser.add_argument("--max_age_min", type=int, default=1440) parser.add_argument("--raw", default=False, action="store_true") args_ns = parser.parse_args(argv) with open(args_ns.query_file, "r") as query_fp: query = query_fp.read() endpoint = discover_neptune_endpoint() client = AltimeterNeptuneClient(max_age_min=args_ns.max_age_min, neptune_endpoint=endpoint) if args_ns.raw: raw_results = client.run_raw_query(query=query) print(raw_results.to_csv(), end="") else: results = client.run_query(graph_names=args_ns.graph_names, query=query) print(results.to_csv(), end="") return 0
def run_query(job: schemas.Job, config: QueryConfig) -> QueryResult: """Run a query and return a QueryResult object""" endpoint = NeptuneEndpoint(host=config.neptune_host, port=config.neptune_port, region=config.neptune_region) neptune_client = AltimeterNeptuneClient(max_age_min=int( job.max_graph_age_sec / 60.0), neptune_endpoint=endpoint) if job.raw_query: all_graph_metadatas = neptune_client.get_all_graph_metadatas() graph_uris_load_times: Dict[str, int] = { graph_metadata.uri: graph_metadata.end_time for graph_metadata in all_graph_metadatas } query_result_set = neptune_client.run_raw_query(query=job.query) query_result = QueryResult(graph_uris_load_times, query_result_set) else: query_result = neptune_client.run_query( graph_names=set(job.graph_spec.graph_names), query=job.query, ) return query_result
def main(argv: Optional[List[str]] = None) -> int: if argv is None: argv = sys.argv[1:] parser = argparse.ArgumentParser() parser.add_argument("query_file", type=str) parser.add_argument("--graph_names", type=str, default=["alti"], nargs="+") parser.add_argument("--max_age_min", type=int, default=1440) parser.add_argument("--raw", default=False, action="store_true") parser.add_argument("--neptune_endpoint", help="Neptune endpoint specified as host:port:region") args_ns = parser.parse_args(argv) with open(args_ns.query_file, "r") as query_fp: query = query_fp.read() if args_ns.neptune_endpoint is not None: try: host, port_str, region = args_ns.neptune_endpoint.split(":") port: int = int(port_str) except ValueError: print( f"neptune_endpoint should be a string formatted as host:port:region" ) return 1 endpoint = NeptuneEndpoint(host=host, port=port, region=region) else: endpoint = discover_neptune_endpoint() client = AltimeterNeptuneClient(max_age_min=args_ns.max_age_min, neptune_endpoint=endpoint) if args_ns.raw: raw_results = client.run_raw_query(query=query) print(raw_results.to_csv(), end="") else: results = client.run_query(graph_names=args_ns.graph_names, query=query) print(results.to_csv(), end="") return 0