Merge pull request #258 from bthrx/yt-stdin

modified yt to also accept urls via stdin
danielmiessler · Mar 16, 2024 · 371f16f · 371f16f
2 parents 059a737 + 164fe20
commit 371f16f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/installer/client/cli/yt.py b/installer/client/cli/yt.py
@@ -7,6 +7,7 @@
 import json
 import isodate
 import argparse
+import sys
 
 
 def get_video_id(url):
@@ -82,4 +83,8 @@ def main():
     parser.add_argument('--transcript', action='store_true',
                         help='Output only the transcript')
     args = parser.parse_args()
+
+    if args.url is None:
+        args.url = sys.stdin.readline().strip()
+
     main_function(args.url, args)