__init__.py

"""
skill fairytalez
Copyright (C) 2018  Andreas Lorensen

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

from mycroft import MycroftSkill, intent_file_handler
from mycroft.util.parse import match_one
from mycroft.audio import wait_while_speaking
import requests
from bs4 import BeautifulSoup
import time


class Fairytalez(MycroftSkill):
    def __init__(self):
        MycroftSkill.__init__(self)

    def initialize(self):
        self.is_reading = False

    @intent_file_handler('fairytalez.intent')
    def handle_fairytalez(self, message):
        if message.data.get("tale") is None:
            response = self.get_response('fairytalez', num_retries=0)
            if response is None:
                return
        else:
            response = message.data.get("tale")
        self.speak_dialog('let_me_think', data={"story": response})
        index = self.get_index("https://fairytalez.com/fairy-tales/")
        result = match_one(response, list(index.keys()))

        if result[1] < 0.8:
            self.speak_dialog('that_would_be', data={"story": result[0]})
            response = self.ask_yesno('is_it_that')
            if response != 'yes':
                self.speak_dialog('no_story')
                return
        self.speak_dialog('i_know_that', data={"story": result[0]})
        # self.log.info(result + " " + result[0])
        self.settings['story'] = result[0]
        time.sleep(3)
        self.tell_story(index.get(result[0]), 0)

    @intent_file_handler('continue.intent')
    def handle_continue(self, message):
        if self.settings.get('story') is None:
            self.speak_dialog('no_story_to_continue')
        else:
            story = self.settings.get('story')
            self.speak_dialog('continue', data={"story": story})
            index = self.get_index("https://fairytalez.com/fairy-tales/")
            self.tell_story(index.get(story), self.settings.get('bookmark') - 1)

    def tell_story(self, url, bookmark):
        self.is_reading = True
        self.settings['bookmark'] = bookmark
        if bookmark == 0:
            title = self.get_title(url)
            author = self.get_author(url)
            self.speak_dialog('title_by_author', data={'title': title, 'author': author})
            time.sleep(1)
        lines = self.get_story(url)
        for line in lines[bookmark:]:
            self.settings['bookmark'] += 1
            time.sleep(.5)
            if self.is_reading is False:
                break
            sentenses = line.split('. ')
            for sentens in sentenses:
                if self.is_reading is False:
                    break
                else:
                    wait_while_speaking()
                    self.speak(sentens, wait=True)
        if self.is_reading is True:
            self.is_reading = False
            self.settings['bookmark'] = 0
            self.settings['story'] = None
            time.sleep(2)
            self.speak_dialog('from_fairytalez')

    def stop(self):
        self.log.info('stop is called')
        if self.is_reading is True:
            self.is_reading = False
            return True
        else:
            return False

    def get_soup(self, url):
        try:
            return BeautifulSoup(requests.get(url).text, "html.parser")
        except Exception as SockException:
            self.log.error(SockException)

    def get_story(self, url):
        soup = self.get_soup(url)
        lines = [a.text.strip() for a in soup.find(id="main").find_all("p")[1:]]
        lines = [l for l in lines if not l.startswith("{") and not l.endswith("}")]
        return lines

    def get_title(self, url):
        soup = self.get_soup(url)
        title = [a.text.strip() for a in soup.findAll("h1")][0]
        return title

    def get_author(self, url):
        soup = self.get_soup(url)
        author = [a.text.strip() for a in soup.findAll("div", {"class": "author-name"})][0]
        return str(author).split("  ")[0]

    def get_index(self, url):
        soup = self.get_soup(url)
        index = {}
        for link in soup.find(id="main").find_all('a'):
            index.update({link.text[2:]: link.get("href")})
        return index


def create_skill():
    return Fairytalez()