From f89b38ebf3e24ba951ee8b249d73bbf48685928a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Miko=C5=82aj=20Gradowski?= <mikolaj.gradowski@hotmail.com>
Date: Tue, 3 Jan 2023 19:54:18 +0100
Subject: [PATCH] +support for parsing stdin in CLI (#880)

---
 sqlglot/__main__.py | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/sqlglot/__main__.py b/sqlglot/__main__.py
index 42a54bccf5..f9613b236d 100644
--- a/sqlglot/__main__.py
+++ b/sqlglot/__main__.py
@@ -1,9 +1,15 @@
 import argparse
+import sys
 
 import sqlglot
 
 parser = argparse.ArgumentParser(description="Transpile SQL")
-parser.add_argument("sql", metavar="sql", type=str, help="SQL string to transpile")
+parser.add_argument(
+    "sql",
+    metavar="sql",
+    type=str,
+    help="SQL statement(s) to transpile, or - to parse stdin.",
+)
 parser.add_argument(
     "--read",
     dest="read",
@@ -48,14 +54,20 @@
 args = parser.parse_args()
 error_level = sqlglot.ErrorLevel[args.error_level.upper()]
 
+sql = sys.stdin.read() if args.sql == "-" else args.sql
+
 if args.parse:
     sqls = [
         repr(expression)
-        for expression in sqlglot.parse(args.sql, read=args.read, error_level=error_level)
+        for expression in sqlglot.parse(
+            sql,
+            read=args.read,
+            error_level=error_level,
+        )
     ]
 else:
     sqls = sqlglot.transpile(
-        args.sql,
+        sql,
         read=args.read,
         write=args.write,
         identify=args.identify,