smartmirror.py

from time import sleep

try:
    import tokens
except ImportError:
    print("Couldn't import 'token' file. Create it first.")
    import sys
    sys.exit(0)

import json
from user import *
from util import *
from weather import Weather
from speech import write
from urllib.parse import quote
from PyQt5.QtCore import QUrl, QThread, pyqtSignal

import random
import requests

weather_api_token = tokens.WEATHER_API_TOKEN
wit_token = tokens.WIT_ACCESS_TOKEN


class Bot(QThread):
    URL_WIT = 'https://api.wit.ai/message?v=20170319&q={}'
    URL_USER_HOME = "http://localhost/home/{}/{}"
    URL_USER_MANAGEMENT = "http://localhost/user_management"
    URL_FORECAST = "http://localhost/forecast/{}/{}"
    URL_RESPONSE = "http://localhost/response"
    URL_LOCATION = "http://localhost/location/{}/{}"
    URL_SPELL = "http://localhost/spell/{}"
    URL_SLEEP = "http://localhost/sleep"
    URL_OPTIONS ="http://localhost/options"

    page_changed = pyqtSignal(QUrl)

    def __init__(self, mode=write):
        super().__init__()
        self.current_user = None
        self.weather = Weather(weather_api_token)
        cities = get_data_from_file("cities.json")
        self.cities = cities['cities']
        self.speech = mode
        self.active_url = 'http://localhost/home/'
        self.message = False

    def run(self):
        """
        Bot running
        """

        data = get_data_from_file("last_use.json")
        if data is not None:
            if 'name' in data:
                self.login(data['name'], "I'm listening...")
                print("Welcome back, " + self.current_user.name + "!")
        else:
            f = open("last_use.json", "w+")
            f.close()
            self.pyqt_change_url(self.URL_USER_MANAGEMENT, "I'm listening...")
            self.user_management()
        while True:
            sleep(2)
            if "http://localhost/location/" in self.active_url:
                sleep(13)
            if self.message:
                self.pyqt_change_url(self.active_url, "I'm listening...")
                self.message = False
            try:
                command = self.speech()
            except Exception as e:
                print(e)
                continue

            self.action(command)

    def action(self, command):
        if self.logged_in():
            name = self.current_user.name
        else:
            name = 'None'

        r = None
        try:
            r = requests.get(self.URL_WIT.format(command), 
                headers={"Authorization": wit_token})
        except Exception as e:
            # did not get the request from wit.ai
            print("REQUEST FAILED")
            self.pyqt_change_url(self.URL_USER_HOME.format(0, name), 'Request to wit.ai did not work.')
            return

        if r is not None:
            print(r.text)
            json_resp = json.loads(r.text)
            if 'error' in json_resp:
                print('Authentication with wit.ai failed.')
                self.pyqt_change_url(self.URL_USER_HOME.format(0, name))
                return
            intent = self.get_intent(json_resp)
            location = self.get_location(json_resp)
            date_time = self.get_datetime(json_resp)

            if intent is not None:
                if intent == 'weather':
                    if location:
                        self.pyqt_change_url(self.URL_FORECAST.format(location, date_time))
                    else:
                        while not location:
                            self.pyqt_change_url(self.active_url,
                                                 "For which city would you like to have a weather forecast?")
                            location = self.speech()
                            if location in self.cities:
                                self.pyqt_change_url(self.URL_FORECAST.format(location, date_time))
                                break
                            else:
                                self.pyqt_change_url(self.active_url, 'I do not know of this place, sorry! Try again!')
                                location = None
                                sleep(2)

                elif intent == "direction":
                    t_type = None
                    try:
                        location = self.get_location(json_resp)
                    except Exception as e:
                        return
                    self.pyqt_change_url(self.active_url,
                                         '1. Directions to ' + location + '\n2. Try new destination with spell mode')
                    while 1:
                        c = self.speech()
                        if c in ["one", "directions"]:
                            break
                        elif c in ["two", "new destination", "spell mode", "spell"]:
                            location = self.spell_word()
                            break
                    while not t_type:
                        self.pyqt_change_url(self.active_url,
                                             'Means of transportation: 1.Public Transport 2.'
                                             'By foot 3.By car 4.Bicycle')
                        command = self.speech()
                        if command in ["one", "public", "public transportation"]:
                            t_type = "r"
                        elif command in ["two", "walking", "by foot", "foot", "walk"]:
                            t_type = "w"
                        elif command in ["three", "drive", "car", "driving", "by car"]:
                            t_type = "c"
                        elif command in ["four", "bicycle", "riding bicycle", "ride", "bicycling", "cycling"]:
                            t_type = "b"
                        else:
                            self.pyqt_change_url(self.active_url, 'I did not understand you. Please try again!')
                            sleep(2)

                    self.pyqt_change_url(self.URL_LOCATION.format(location, t_type))

                elif intent == 'hometown':
                    self.set_user_location()

                elif intent == 'logout':
                    self.logout()

                elif intent == 'login':
                    self.login()

                elif intent == 'greeting':
                    self.pyqt_change_url(self.active_url, random.choice(['Hello yourself!',
                                                                         'Hi there!', 'Hi!', 'Hola!', 'Greetings!']))
                elif intent == 'options':
                    self.use_options()

                elif intent == 'user_creation':
                        self.create_new_user()

                elif intent == 'user_deletion':
                    self.delete_user()

                elif intent == 'user_show':
                    self.show_users()

                elif intent == 'mirror':
                    self.pyqt_change_url(self.URL_RESPONSE,"See you later.")
                    sleep(1)
                    self.pyqt_change_url(self.URL_SLEEP)
                    self.update_file()
                    while True:
                        try:
                            command = self.speech()
                            r = None
                            try:
                                r = requests.get(self.URL_WIT.format(command),
                                    headers={"Authorization": wit_token})
                            except Exception as e:
                                # did not get the request from wit.ai
                                print("REQUEST FAILED")
                                self.pyqt_change_url(self.URL_USER_HOME.format(0, name), 'Request to wit.ai did not work.')
                                return

                            if r is not None:
                                print(r.text)
                                json_resp = json.loads(r.text)
                                if 'error' in json_resp:
                                    print('Authentication with wit.ai failed.')
                                    self.pyqt_change_url(self.URL_USER_HOME.format(0, name))
                                    return
                                intent = self.get_intent(json_resp)
                                if intent == 'wake up':
                                    return self.run()
                        except Exception as e:
                            pass
                elif intent == 'shutdown':
                    exit()
                elif intent == 'fairytale':
                    self.pyqt_change_url(self.active_url, "Laaaame...")
                elif intent == 'insult':
                    action = random.choice(['Pardon me?', 'Excuse me?', 'Be careful!', 'Come again?', 'Shame on you!',
                                            "You got 30 seconds to think about what you just did!"])

                    if len(action) > 15:
                        for i in range(1, 30):
                            self.pyqt_change_url(self.active_url, str(i) + " " + action)
                            sleep(1)

                    self.pyqt_change_url(self.URL_USER_HOME.format(0, name), action)
            else:
                self.pyqt_change_url(self.active_url, 'I do not understand the intent of your statement.')
        
    def set_user_location(self):
        """
        TODO: if old_hometown == new_hometown -> print something and exit
        """
        if self.logged_in():
            if self.current_user.hometown is not None:
                print("Your current hometown is '" + self.current_user.hometown + "'")
            self.pyqt_change_url(self.URL_USER_HOME.format(0, self.current_user.name),
                                 "1.[NORMAL MODE]\n2.[SPELL MODE]")
            print('Tell me which town to set as your hometown.')

            while 1:
                command = self.speech()
                if command in ["one", "normal", "normal mode"]:
                    self.pyqt_change_url(self.URL_RESPONSE,
                                         "[NORMAL MODE] Tell me which town to set as your hometown.")
                    command = self.speech()
                    break
                elif command in ["two", "spell mode", "spell"]:
                    command = self.spell_word()
                    break

            if command in self.cities:
                self.current_user.hometown = command
                data = get_data_from_file("users.json")
                for user in data['users']:
                    if user["name"] == self.current_user.name:
                        user['hometown'] = self.current_user.hometown
                dump_data_to_file(data, "users.json")
                self.pyqt_change_url(self.URL_USER_HOME.format(0, self.current_user.name),
                                     'Hometown successfully set to ' + self.current_user.hometown + '!')
                print("Set your hometown to '" + self.current_user.hometown + "'.")
            else:
                self.pyqt_change_url(self.URL_USER_HOME.format(0, self.current_user.name), "I may not know that city.")
                print("I may not know that city.")
        else:
            self.pyqt_change_url(self.URL_USER_HOME.format(0, 'None'), "Log in first before setting a hometown.")

    def use_options(self):
        if self.logged_in():
            self.pyqt_change_url(self.URL_OPTIONS)
        else:
            self.user_management()

    def hobby_selection(self):
        print("Do you want to:\n1.View your hobbies?\n2.Add hobbies?\n3.Remove hobbies?\n4.Go back?")

        command = self.speech()

        print("You said:" + command)

        if any(command in w for w in ["one", "view my hobbies", "view hobbies", "view your hobbies", "view"]):
            self.current_user.show_hobbies()
        elif any(command in w for w in ["two", "add", "add hobbies", "add your hobbies"]):
            self.current_user.add_hobbies()
        elif any(command in w for w in ["three", "remove", "remove hobbies", "remove your hobbies"]):
            self.current_user.remove_hobbies()
        elif any(command in w for w in ["4", "back", "go back"]):
            return

    def user_management(self):
        if(self.logged_in()):
            self.pyqt_change_url(self.URL_USER_HOME.format(0, self.current_user.name),
                                 "You are already logged in as " + self.current_user.name + ".")
            return
        else:
            self.pyqt_change_url(self.URL_USER_MANAGEMENT)

    """
    creates a new user in "./users.json"

    """
    def create_new_user(self, new_user=''):
        if self.current_user is not None:
            print("You are already logged in as " + self.current_user.name + ". Log out first.")
            self.pyqt_change_url(self.URL_USER_HOME.format(0, self.current_user.name), "You are already logged in as " + self.current_user.name + ". Log out first.")

        elif not new_user == '':
            self.current_user = User()
            self.current_user.name = new_user
            self.current_user.hometown = ''

        else:
            correct = False
            while not correct:
                self.pyqt_change_url(self.URL_RESPONSE, "New user creation. Choose: 1. [NORMAL MODE]\n2. [SPELL MODE]")
                print("New User creation. Please spell your name for me. To cancel say 'abort' or 'cancel'.")
                while 1:
                    command = self.speech()
                    if command in ["one", "normal", "normal mode"]:
                        self.pyqt_change_url(self.URL_RESPONSE, "[NORMAL MODE] New user creation. Please tell me your name.")
                        command = self.speech()
                        break
                    elif command in ["two", "spell mode", "spell"]:
                        command = self.spell_word()
                        break
                print("You said '" + command + "'. Is that correct?")
                self.pyqt_change_url(self.URL_RESPONSE, "You said '" + command + "'. Is that correct?")
                if any(command in w for w in CANCEL_LIST):
                    return self.pyqt_change_url(self.USER_MANAGEMENT, "I'm listening...")
                approval_command = self.speech()
                if any(approval_command in w for w in APPROVAL_LIST):
                    print("You said " + approval_command + ".")
                    correct = True
            # check if 'users.json'-file alrdy exists
            if not os.path.isfile("users.json"):
                f = open("users.json", "w+")
                user_data = {"users": []}
                dump_data_to_file(user_data, "users.json")
                f.close()
            # check if name exists in users.json
            else:
                user_data = get_data_from_file("users.json")
                user_list = user_data['users']
                if any(user['name'] == command for user in user_list):
                    print("User already exists... - Choose another name, please.")
                    self.pyqt_change_url(self.URL_RESPONSE, "User already exists... - Choose another name, please.")
                    return self.create_new_user()

            dump_data = get_data_from_file("users.json")

            dump_data["users"].append({"name": command, "hometown": None, "hobbies": None, "logged_in": False})

            dump_data_to_file(dump_data, "users.json")

            print("User " + command + " successfully created.")
            self.pyqt_change_url(self.URL_RESPONSE, "User " + command + " successfully created.")
            sleep(3)
            self.login(command)

    def delete_user(self):
        user_list = get_user_list()
        if user_list is None or len(user_list) == 0:
            print("There are no users yet.")
            self.pyqt_change_url(self.URL_RESPONSE, "There are no users yet.")
            sleep(3)
            self.pyqt_change_url(self.URL_USER_MANAGEMENT)
        else:
            in_deletion = True
            while in_deletion:
                print("Which user do you want to remove?")
                self.show_users("Which user do you want to remove?")
                command = self.speech()
                print("You said " + command)

                command_list = []
                user_match = {}

                count = 1
                for i in user_list:
                    if number_conversion(count) is not None:
                        command_list.append(number_conversion(count))
                        key = number_conversion(count)
                    else:
                        command_list.append(str(count))
                        key = str(count)
                    command_list.append(i)
                    user_match[key] = i
                    count += 1

                print("numbers:", command_list)
                print("NAMES:", user_match)
                if any(command in w for w in command_list):
                    if command in user_match.keys():
                        user_to_delete = user_match[command]
                    else:
                        user_to_delete = command

                    user_data = get_data_from_file("users.json")
                    user_list = user_data['users']

                    item_index = next(index for (index, user) in enumerate(user_list) if user['name'] == user_to_delete)
                    
                    user_name = user_list[item_index]['name']
                    print('Are you sure you want to delete ' + user_name + '?')
                    self.pyqt_change_url(self.URL_RESPONSE, 'Are you sure you want to delete ' + user_name + '?')
                    approval_command = self.speech()

                    if any(approval_command in w for w in APPROVAL_LIST):
                        del user_list[item_index]

                        if len(user_list) == 0:
                            os.remove('users.json')
                        else:
                            user_data['users'] = user_list
                            dump_data_to_file(user_data, "users.json")

                        print('User ' + user_name + ' deleted!')
                        msg = 'User ' + user_name + ' deleted!'
                        self.pyqt_change_url(self.URL_USER_MANAGEMENT, msg)
                        in_deletion = False
                    else:
                        msg = 'Did not delete the user'
                        self.pyqt_change_url(self.URL_USER_MANAGEMENT, msg)
                        in_deletion = False

    def login(self, user_name='', message=None):
        if self.current_user is not None:
            print("You are already logged in as " + self.current_user.name + ". Log out first.")
            self.pyqt_change_url(self.URL_USER_HOME.format(0, self.current_user.name), "You are already logged in as "
                                 + self.current_user.name + ". Log out first.")
        else:
            user_list = get_user_list()
            if user_list is None or len(user_list) == 0:
                print("There are no Users to chose from. - Redirecting to User Creation")
                self.pyqt_change_url(self.URL_RESPONSE, "There are no Users to chose from. - "
                                                        "Redirecting to User Creation")
                sleep(3)
                self.create_new_user()
            else:
                if not user_name == '':
                    user_data = get_data_from_file("users.json")
                    for user in user_data['users']:
                        if user['name'] == user_name:
                            self.current_user = User()
                            self.current_user.name = user['name']
                            self.current_user.hometown = user['hometown']
                            self.current_user.hobbies = user['hobbies']
                            user['logged_in'] = True

                            dump_data_to_file(user_data, 'users.json')
                else:
                    while(not(self.logged_in())):
                        self.show_users("Select your Profile from the list of users by saying the Name or Number:")
                        command = self.speech()

                        command_list = []
                        user_match = {}

                        count = 1
                        for i in user_list:
                            if number_conversion(count) is not None:
                                command_list.append(number_conversion(count))
                                key = number_conversion(count)
                            else:
                                command_list.append(str(count))
                                key = str(count)
                            command_list.append(i)
                            user_match[key] = i
                            count += 1
                            
                        if any(command in w for w in command_list):
                            self.current_user = User()
                            if command in user_match.keys():
                                self.current_user.name = user_match[command]
                            else:
                                self.current_user.name = command

                            data = get_data_from_file("users.json")

                            for user in data['users']:
                                if user['name'] == self.current_user.name:
                                    self.current_user.hometown = user['hometown']
                                    self.current_user.hobbies = user['hobbies']
                                    user['logged_in'] = True

                            dump_data = {"name": self.current_user.name}

                            dump_data_to_file(dump_data, "last_use.json")
                            dump_data_to_file(data, 'users.json')
                            print("Hello " + self.current_user.name + "!\nYou are logged in now!")

                if message is not None:
                    self.pyqt_change_url(self.URL_USER_HOME.format(0, self.current_user.name), message)
                else:
                    self.pyqt_change_url(self.URL_USER_HOME.format(0, self.current_user.name))

    def logout(self):
        if(self.logged_in()):
            data = get_data_from_file('users.json')
            for user in data['users']:
                if user['logged_in']:
                    user['logged_in'] = False
            dump_data_to_file(data, 'users.json')
            self.current_user = None
            
            self.user_management()
        else:
            print("You are not logged in.")

    def logged_in(self):
        if self.current_user is None:
            return False
        else:
            return True

    def who_am_i(self):
        if (self.logged_in()):
            print("You are " + self.current_user.name + ".")
        else:
            print("How would I know such a thing? You are not logged in...")

    def show_users(self, msg=None):
        print(get_user_list())
        if msg is None:
            self.pyqt_change_url("http://localhost/show_users")
        else:
            self.pyqt_change_url("http://localhost/show_users", msg)

    def update_file(self):
        if self.logged_in():
            data = {"name": self.current_user.name}
            dump_data_to_file(data, "last_use.json")
        else:
            if(os.path.isfile("last_use.json")):
                os.remove("last_use.json")

    @staticmethod
    def get_intent(json_wit):
        intent = None
        if 'entities' in json_wit and 'intent' in json_wit['entities']:
            entities = json_wit['entities']
            intent = entities['intent'][0]['value']
        return intent

    def get_location(self, json_wit):
        location = None
        if 'entities' in json_wit and 'location' in json_wit['entities']:
            entities = json_wit['entities']
            location = entities['location'][0]['value']
        return location

    def get_datetime(self, json_wit):
        date_time = None
        if 'entities' in json_wit and 'datetime' in json_wit['entities']:
            entities = json_wit['entities']
            date_time = entities['datetime'][0]['value']
        return date_time

    def pyqt_change_url(self, url, message=None):
        if message is None:
            self.page_changed.emit(QUrl(url))
        else:
            self.page_changed.emit(QUrl(url + '?msg=' + quote(message)))
            self.message = True
        self.active_url = url

    def spell_word(self):
        letter = {"one": "a", "two": "b", "three": "c", "4": "d", "5": "e", "6": "f", "7": "g", "8": "h",
                  "9": "i", "10": "j", "11": "k", "12": "l", "13": "m", "14": "n", "15": "o", "16": "p",
                  "17": "q", "18": "r", "19": "s", "20": "t", "21": "u", "22": "v", "23": "w", "24": "x",
                  "25": "y", "26": "z"}
        stop = None
        result = ""
        while not stop:
            self.pyqt_change_url(self.URL_SPELL.format(result)),
            l = self.speech()

            if l in ["done", "stop", "enter"]:
                stop = True
                return result
            elif l in letter.keys():
                result += letter[l]
            else:
                self.pyqt_change_url(self.active_url, "I did not understand you, please try again!")
                sleep(1)