45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
'''
|
|
I was having trouble making my Flask server perform the youtube-dl without
|
|
clogging up the other site activities. So instead I'll just have the server
|
|
export ytqueue files, which this script will download as a separate process.
|
|
|
|
Rather than maintaining a text file or database of IDs to be downloaded,
|
|
I'm fine with creating each ID as a file and letting the filesystem act
|
|
as the to-do list.
|
|
'''
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
YOUTUBE_DL = 'youtube-dlw https://www.youtube.com/watch?v={id}'
|
|
|
|
def ytqueue(only_once=False):
|
|
while True:
|
|
print(time.strftime('%H:%M:%S'), 'Looking for files.')
|
|
queue = [f for f in os.listdir() if f.endswith('.ytqueue')]
|
|
for filename in queue:
|
|
yt_id = filename.split('.')[0]
|
|
command = YOUTUBE_DL.format(id=yt_id)
|
|
exit_code = os.system(command)
|
|
if exit_code == 0:
|
|
os.remove(filename)
|
|
if only_once:
|
|
break
|
|
time.sleep(10)
|
|
|
|
def ytqueue_argparse(args):
|
|
return ytqueue(only_once=args.once)
|
|
|
|
def main(argv):
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument('--once', dest='once', action='store_true')
|
|
parser.set_defaults(func=ytqueue_argparse)
|
|
|
|
args = parser.parse_args(argv)
|
|
return args.func(args)
|
|
|
|
if __name__ == '__main__':
|
|
raise SystemExit(main(sys.argv[1:]))
|