def main(args=None) -> None:
"""
The command line interface entry point.
:param args: the command line arguments for sqllineage command
"""
parser = argparse.ArgumentParser(prog="sqllineage",
description="SQL Lineage Parser.")
parser.add_argument("-e",
metavar="<quoted-query-string>",
help="SQL from command line")
parser.add_argument("-f", metavar="<filename>", help="SQL from files")
parser.add_argument(
"-v",
"--verbose",
help="increase output verbosity, show statement level lineage result",
action="store_true",
)
parser.add_argument(
"-g",
"--graph-visualization",
help="show graph visualization of the lineage within a webserver",
action="store_true",
)
parser.add_argument(
"-p",
help="the port visualization webserver will be listening on",
type=int,
default=DEFAULT_PORT,
metavar="<port_number>{0..65536}",
)
args = parser.parse_args(args)
if args.e and args.f:
logging.warning(
"Both -e and -f options are specified. -e option will be ignored")
if args.f or args.e:
sql = extract_sql_from_args(args)
runner = LineageRunner(
sql,
verbose=args.verbose,
draw_options={
"p": args.p,
"f": args.f if args.f else None
},
)
if args.graph_visualization:
runner.draw()
else:
print(runner)
elif args.graph_visualization:
return draw_lineage_graph(
**{
"p":
args.p,
"f":
os.path.join(os.path.dirname(__file__),
"data/tpcds/query01.sql"),
})
else:
parser.print_help()
def draw(self) -> None:
"""
to draw the lineage directed graph
"""
draw_options = self._draw_options
if draw_options.get("f") is None:
draw_options.pop("f", None)
draw_options["e"] = self._sql
return draw_lineage_graph(**draw_options)
def test_no_pygraphviz():
with pytest.raises(ImportError):
draw_lineage_graph(DiGraph())
def test_no_matplotlib():
with pytest.raises(ImportError):
draw_lineage_graph(DiGraph())
def draw(self) -> None:
"""
to draw the lineage directed graph with matplotlib.
"""
return draw_lineage_graph(self._combined_lineage_result.lineage_graph)