streampy.py livestreamer wrapper

streampy

README

URXVT integration

Change the url-select script for urxvt like below

~/.urxvt/ext$ diff url-select urxvt-perls/url-select
137,140c137
<     } elsif ($char eq 's' || $char eq 'd') {
<         $self->exec_async( "streampy.py", ${$self->{found}[$self->{n}]}[4], "&" );
<         deactivate($self) unless $char eq 's';
<     } elsif ($char eq 'y') {
---
> 	} elsif ($char eq 'y') {


i3 integration

add the following lines to your i3 config

assign [class="Vlc"] 9
for_window [title="^VLC$"] move workspace 8

streampy.py

#!/usr/bin/python3
from __future__ import division
from subprocess import Popen, PIPE
import pyperclip as pc
import sys
import urllib.parse
import urllib.request
import re

#html redirect following taken from torskbot
#https://github.com/sebth/torskbot

class FinalURLHTTPRedirectHandler(urllib.request.HTTPRedirectHandler):

    def __init__(self, *args, **kwargs):
        self.final_url = None
        super().__init__(*args, **kwargs)

    def redirect_request(self, req, fp, code, msg, hdrs, newurl):
        self.final_url = newurl
        return super().redirect_request(req, fp, code, msg, hdrs, newurl)

def quote_nonascii_path(path):
    return re.sub(
            b'[\x80-\xff]',
            lambda match: '%{:x}'.format(ord(match.group())).encode('ascii'),
            path.encode()).decode('ascii')

def urlquote(url):
    parts = urllib.parse.urlsplit(url)
    return urllib.parse.urlunsplit(
            (parts[0], parts[1].encode('idna').decode('ascii'),
                quote_nonascii_path(parts[2])) + parts[3:])

def final_url(url):
    if "www" not in url:
        return url
    url = quote_nonascii_path(url)
    rh = FinalURLHTTPRedirectHandler()
    opener = urllib.request.build_opener(rh)
    opener.open(url)
    final_url = rh.final_url
    return rh.final_url if rh.final_url else url

#end html redirect
#

def read_url_from_clipboard():
    clipboard = pc.paste()
    if isinstance(clipboard, bytes):
        clipboard = clipboard.encode('UTF-8','ignore')
    return clipboard

def is_url(text):
    return '.' in text

def conditional_print(arguments, verbose=False):
    if verbose:
        print(arguments)

def main():
    arguments = sys.argv
    url = None
    verbose = False
    for arg in sys.argv[1:]:
        if is_url(arg):
            url = arg
        elif arg[0] == 'v':
            verbose=True

    if not url:
        url = read_url_from_clipboard()

    try:
        url = final_url(url)
        play_url(url, verbose)
    except Exception as e:
        print(e)
        pass

def play_url(url, verbose=False):
    livestreamer_args = ['best']
    youtubeviewer_args = ['best', '--no-interactive']

    #if isinstance(url, unicode):
    #    url = url.encode('UTF-8','ignore')
    if isinstance(url, bytes):
        url = url.decode('UTF-8')

    conditional_print('verbose output on', verbose)
    if 'twitch.tv' in url:
        conditional_print('twitch stream', verbose)

        args = ['livestreamer', url] + livestreamer_args
        if verbose:
            print(args, len(args))
        p = Popen(args, stdout=PIPE, stderr=PIPE)
        outmsg, errmsg = p.communicate()

    elif 'youtube.com' not in url:
        conditional_print('not youtube url - possibly shortened', verbose)
        url = final_url(url)

    if 'youtube.com' in url:
        conditional_print('youtube url', verbose)
        args = ['livestreamer', url] + livestreamer_args
        if verbose:
            print(args, len(args))

        p = Popen(args, stdout=PIPE, stderr=PIPE)
        outmsg, errmsg = p.communicate()

        if verbose:
            print("process output\n", outmsg)
            print("error?", errmsg)
            print("msg type", type(outmsg))

        outmsg_string = outmsg.decode(encoding='UTF-8')
        if 'youtube-dl' in outmsg_string:
            if verbose:
                print("process output\n", outmsg)
                print("TRYING youtube-viewer")

            #args[0] = 'youtube-viewer'
            #args.append('--no-interactive')
            args = ['youtube-viewer', url] + youtubeviewer_args
            if verbose:
                print("args:\n", args, verbose)
            p = Popen(args, stdout=PIPE, stderr=PIPE)
            outmsg = p.communicate()

    elif "twitch.tv" not in url:
        #hope livestreamer works, output to stdout
        args = ['livestreamer', url] + livestreamer_args
        if verbose:
           print(args, len(args), verbose)
        Popen(args)
        p.communicate()



if __name__ == '__main__':
    main()
    #a couple of example links to test whether it works
    """
    url = 'http://www.twitch.tv/forgg'
    url = 'https://www.youtube.com/watch?v=QcqC4jtrR9Y'     #regular
    url = 'https://www.youtube.com/watch?v=NJ3aiM8K6D0'     #protected
    """