ytubejb.py

#!/usr/bin/env python


import subprocess
import os
import sys
import re
import argparse
import configparser
from config import CONFIG, AUDIO, write_config_if_not_exists


APP_NAME = AUDIO

DEFAULT = {'number to play': 5}


def get_args(args):
    parser = argparse.ArgumentParser(description='YouTube as a command line jukebox')
    parser.add_argument('source', type=str, help='file or url to play')
    parser.add_argument('-n', '--number-to-play', type=int, help='number of streams to play', metavar='')
    repeater = parser.add_mutually_exclusive_group()
    repeater.add_argument('-r', '--repeat', action='store_true', help='play the first audio on repeat')
    repeater.add_argument('-ra', '--repeat-all', action='store_true', help='play all on repeat')
    return parser.parse_args(args)
 

class PlayAudio:

    def __init__(self, source, source_type, mpv_command, number_to_play):
        self.source = source
        self.source_type = source_type
        self.mpv_command = mpv_command
        self.number_to_play = number_to_play
        self.play_audio()

    def play_url(self):
        subprocess.run([*self.mpv_command, self.source])

    def play_search_result(self):
        search = f'ytsearch{self.number_to_play}:{self.source}'
        search_command = ['youtube-dl',
                          '--format',
                          'bestaudio',
                          '--get-url',
                          search]
        search_results = subprocess.Popen(search_command,
                                          stdout=subprocess.PIPE)
        output, _ = search_results.communicate()
        to_play = [audio for audio in output.split(b'\n') if audio]
        subprocess.run([*self.mpv_command, *to_play])

    def play_audio(self):
        play = {'url': self.play_url,
                'search': self.play_search_result}
        play[self.source_type]()


def main(argv):
    write_config_if_not_exists()
    config = configparser.ConfigParser()
    config.read(CONFIG)
    app_config = config[APP_NAME]

    args = get_args(argv)

    source = args.source

    mpv_command = ['mpv', '--no-video']

    if args.repeat:
        mpv_command.append('--loop')
    elif args.repeat_all:
        mpv_command.append('--loop-playlist')

    if re.match(r'^http', source):
        source_type = 'url'
    else:
        source_type = 'search'

    if args.number_to_play:
        number_to_play = args.number_to_play
    else:
        number_to_play = app_config.get('number to play', DEFAULT['number to play'])

    PlayAudio(source, source_type, mpv_command, number_to_play)


if __name__ == '__main__':
    main(sys.argv[1:])