piptube.py

#!/usr/bin/env python


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


APP_NAME = VIDEO

DEFAULT = {
        'video size': 'medium',
        'video quality': 'high',
        'position': 'bottom right',
        'number to play': 5
        }

VIDEO_SIZE = {
        'small': '384x216',
        'medium': '640x360',
        'large': '1280x720',
        'extra large': '1920x1080'
        }

VIDEO_POSITION = {
        'top left': '2%:2%',
        'top right': '98%:2%',
        'bottom right': '98%:98%',
        'bottom left': '2%:98%'
        }

VIDEO_QUALITY = {
        'low': '(mp4)[height<=480]/best[height<=480]',
        'high': '(mp4)[height<=1080]/best[height<=1080]'
        }


def get_args(args):
    parser = argparse.ArgumentParser(description='Picture-in-picture video for mpv and YouTube')
    parser.add_argument('source', type=str, help='file or url to play')
    parser.add_argument('-n', '--number-to-play', type=int, help='number of videos to play', metavar='')
    parser.add_argument('-lq', '--low-quality', action='store_true', help='use a lower quality stream')
    parser.add_argument('-c', '--channel', action='store_true', help='stream channel')
    size = parser.add_mutually_exclusive_group()
    size.add_argument('-s', '--small', action='store_true', help='small video')
    size.add_argument('-m', '--medium', action='store_true', help='medium video')
    size.add_argument('-l', '--large', action='store_true', help='large video')
    size.add_argument('-xl', '--extra-large', action='store_true', help='extra large video')
    position = parser.add_mutually_exclusive_group()
    position.add_argument('-tl', '--top-left', action='store_true', help='initial placement top left')
    position.add_argument('-tr', '--top-right', action='store_true', help='initial placement top right')
    position.add_argument('-bl', '--bottom-left', action='store_true', help='initial placement bottom left')
    position.add_argument('-br', '--bottom-right', action='store_true', help='initial placement bottom right')
    return parser.parse_args(args)


class PlayVideo:

    MPV_BASE = ['mpv',
                '--ontop',
                '--no-border',
                '--on-all-workspaces'
                ]

    def __init__(self,
                 source,
                 source_type,
                 size,
                 position,
                 video_format,
                 number_to_play):
        self.source = source
        self.source_type = source_type
        self.size = f'--autofit={size}'
        self.position = f'--geometry={position}'
        self.video_format = f'{video_format}'
        self.number_to_play = number_to_play
        self.mpv = [*self.MPV_BASE, self.size, self.position]
        self.search_base = ['youtube-dl', '--format', self.video_format]
        self.play_video()

    def play_local(self):
        subprocess.run([*self.mpv,
                        self.source])

    def play_url(self):
        subprocess.run([*self.mpv,
                        '--ytdl-format',
                        self.video_format,
                        self.source])

    def play_search(self):
        search = f'ytsearch{self.number_to_play}:{self.source}'
        search_command = [*self.search_base, '--get-url', search]
        self.stream(search_command)

    def play_channel(self):
        channel = f'ytuser:{self.source}'
        channel_command = [*self.search_base,
                           '--max-downloads',
                           f'{self.number_to_play}',
                           '--get-url',
                           channel]
        self.stream(channel_command)

    def stream(self, command):
        results = subprocess.Popen(command, stdout=subprocess.PIPE)
        output, _ = results.communicate()
        to_play = [video for video in output.split(b'\n') if video]
        subprocess.run([*self.mpv, *to_play])

    def play_video(self):
        play = {
                'file': self.play_local,
                'url': self.play_url,
                'search': self.play_search,
                'channel': self.play_channel
                }
        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

    if os.path.isfile(source):
        source_type = 'file'
    elif re.match(r'^http', source):
        source_type = 'url'
    elif args.channel:
        source_type = 'channel'
    else:
        source_type = 'search'

    if args.small:
        size = VIDEO_SIZE['small']
    elif args.medium:
        size = VIDEO_SIZE['medium']
    elif args.large:
        size = VIDEO_SIZE['large']
    elif args.extra_large:
        size = VIDEO_SIZE['extra large']
    else:
        default_size = app_config.get('video size', DEFAULT['video size'])
        size = VIDEO_SIZE[default_size]

    if args.top_left:
        position = VIDEO_POSITION['top left']
    elif args.top_right:
        position = VIDEO_POSITION['top right']
    elif args.bottom_left:
        position = VIDEO_POSITION['bottom left']
    elif args.bottom_right:
        position = VIDEO_POSITION['bottom right']
    else:
        default_position = app_config.get('position', DEFAULT['position'])
        position = VIDEO_POSITION[default_position]

    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'])

    if args.low_quality:
        video_format = VIDEO_QUALITY['low']
    else:
        default_quality = app_config.get('video quality', DEFAULT['video quality'])
        video_format = VIDEO_QUALITY[default_quality]

    PlayVideo(source,
              source_type,
              size,
              position,
              video_format,
              number_to_play)


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