60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
#!/usr/bin/python3
|
|
"""Download audio using youtube-dl.
|
|
|
|
Dependencies:
|
|
=============
|
|
* youtube-dl
|
|
"""
|
|
|
|
import argparse
|
|
import pathlib
|
|
import shutil
|
|
import subprocess
|
|
|
|
# =========== Constants ==========
|
|
YOUTUBE_DL_BIN = shutil.which('youtube-dl')
|
|
DEFAULT_FILENAME = f"{pathlib.Path.home()}/Music/%(title)s.%(ext)s"
|
|
|
|
# ========== Error Codes ==========
|
|
E_NOURLS = 2
|
|
|
|
# ========== Main Script ==========
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-b', '--batchfile',
|
|
help='provide the links from a text file')
|
|
parser.add_argument('-f', '--format',
|
|
type=str,
|
|
default='opus',
|
|
help='the format to use')
|
|
parser.add_argument('-n', '--filename',
|
|
type=str,
|
|
help='downloaded filename (without extension)')
|
|
parser.add_argument('urls',
|
|
nargs='*',
|
|
help='video URLs')
|
|
args = parser.parse_args()
|
|
|
|
dl_opts = [YOUTUBE_DL_BIN,
|
|
'--no-part',
|
|
'--no-continue',
|
|
'--extract-audio',
|
|
f"--audio-format={args.format}"]
|
|
|
|
# filename handling
|
|
# if -b is used, DEFAULT_FILENAME must take precedence
|
|
if args.filename is not None and args.batchfile is None:
|
|
dl_opts.append(f"--output={args.filename}")
|
|
else:
|
|
dl_opts.append(f"--output={DEFAULT_FILENAME}")
|
|
|
|
# URL handling
|
|
if args.batchfile is not None:
|
|
dl_opts.append(f"--batch-file={args.batchfile}")
|
|
elif args.urls is not None:
|
|
dl_opts.extend(args.urls)
|
|
else:
|
|
print("URLs are required")
|
|
exit(E_NOURLS)
|
|
|
|
subprocess.run(dl_opts)
|