diff --git a/sqllineage/cli.py b/sqllineage/cli.py index 20a580e9..89ca6daa 100644 --- a/sqllineage/cli.py +++ b/sqllineage/cli.py @@ -3,7 +3,14 @@ import logging.config import warnings -from sqllineage import DEFAULT_DIALECT, DEFAULT_HOST, DEFAULT_LOGGING, DEFAULT_PORT +from sqllineage import ( + DEFAULT_DIALECT, + DEFAULT_HOST, + DEFAULT_LOGGING, + DEFAULT_PORT, + NAME as MAIN_NAME, + VERSION as MAIN_VERSION, +) from sqllineage.drawing import draw_lineage_graph from sqllineage.runner import LineageRunner from sqllineage.utils.constant import LineageLevel @@ -23,6 +30,9 @@ def main(args=None) -> None: parser = argparse.ArgumentParser( prog="sqllineage", description="SQL Lineage Parser." ) + parser.add_argument( + "--version", action="version", version="%s %s" % (MAIN_NAME, MAIN_VERSION) + ) parser.add_argument( "-e", metavar="", help="SQL from command line" )