__init__.py

# -*- coding: utf-8 -*-
import urllib.error
import urllib.parse
import urllib.request
from os.path import dirname
from subprocess import DEVNULL, STDOUT, check_call
from adapt.intent import IntentBuilder
from bs4 import BeautifulSoup

from mycroft.skills.core import MycroftSkill

try:
    from mycroft.device import device as d_hw
except ImportError:
    d_hw = 'desktop'


class AVmusicSkill(MycroftSkill):
    def __init__(self):
        super(AVmusicSkill, self).__init__(name="AVmusicSkill")
        self.process = None

    def initialize(self):
        playnow_intent = IntentBuilder("playnow_intent"). \
            require("AgreementKeyword").build()
        self.register_intent(playnow_intent, self.handle_playnow_intent)

        # For future implementation:
        # pause_intent = IntentBuilder("pause_intent"). \
        #    require("PauseKeyword").build()
        # self.register_intent(pause_intent, self.handle_pause_intent)
        # self.disable_intent('pause_intent')

        not_now_intent = IntentBuilder("not_now_intent"). \
            require("DeclineKeyword").build()
        self.register_intent(not_now_intent, self.handle_not_now_intent)

        AVmusic = IntentBuilder("AVmusic_intent"). \
            require("AVmusicKeyword").build()
        self.register_intent(AVmusic, self.AVmusic)

        self.load_data_files(dirname(__file__))

        self.disable_intent('playnow_intent')
        self.disable_intent('not_now_intent')

    @staticmethod
    def search(text):
        query = urllib.parse.quote(text)
        url = "https://www.youtube.com/results?search_query=" + query
        response = urllib.request.urlopen(url)
        html = response.read()
        soup = BeautifulSoup(html, "html.parser")
        for vid in soup.findAll(attrs={'class': 'yt-uix-tile-link'}):
            if not vid['href'].startswith("https://googleads.g.doubleclick.net/‌​") \
                    and not vid['href'].startswith("/user") and not vid['href'].startswith("/channel"):
                return "http://www.youtube.com/" + vid['href']

    def AVmusic(self, message):
        self.stop()
        self.speak('Would you like me to play it now?', True)

        utterance = message.data.get('utterance').lower()
        utterance = utterance.replace(
            message.data.get('AVmusicKeyword'), '')
        self.vid = self.search(utterance + "playlist")

        self.enable_intent('playnow_intent')
        self.enable_intent('not_now_intent')

    def handle_playnow_intent(self, message):
        try:
            if d_hw == 'pi':
                self.process = check_call(["mpv", "--vid=no", self.vid], stdout=DEVNULL, stderr=STDOUT)
            else:
                self.process = check_call(["mpv", self.vid], stdout=DEVNULL, stderr=STDOUT)
            # self.enable_intent('pause_intent')
            self.speak_dialog('SayStop')

        except TypeError:
            self.speak_dialog('TryAgain')
            self.stop()

        self.disable_intent('not_now_intent')
        self.disable_intent('playnow_intent')

    def handle_not_now_intent(self, message):
        self.speak_dialog('ChangeMind')
        self.disable_intent('not_now_intent')

    # def handle_pause_intent(self, message):
    #    self.process.Popen.communicate("P")

    def stop(self):
        if self.process:
            self.process.terminate()
        pass


def create_skill():
    return AVmusicSkill()