Как сделать 1 отклик на 2 команды?

Question

[UMFUCHI]

Вот весь код:

spoiler

import os
import pyautogui
from gtts import gTTS
import random
import time
from time import sleep
import datetime
import playsound
import speech_recognition as sr
import pyautogui as pg




def listen_command():

    r = sr.Recognizer()
    with sr.Microphone() as source:
        print("Скажите вашу команду: ")
        audio = r.listen(source)


    try:
        our_speech = r.recognize_google(audio, language="ru")
        print("Вы сказали: " + our_speech)
        return our_speech
    except sr.UnknownValueError:
        return "ошибка"
    except sr.RequestError:
        return "ошибка"

class VoiceCommandList:
    def __init__(self):
        self.actions = list()

    def on(self, condition):
        if isinstance(condition, str):
            condition = condition.lower()
            predicate = lambda text: condition in text
        elif callable(condition):
            predicate = condition
        else:
            raise TypeError('Condition must be either string or function!')


        def decorator(command_func):
            self.actions.append((predicate, command_func))
            return command_func

        return decorator

    def run_command(self, text):
        text = text.lower()
        for predicate, command in self.actions:
            if predicate(text):
                try:
                    response = command(text)
                    if response is None:
                        response = "Команда выполнена"
                except Exception as err:
                    response = "Ошибка при выполнении команды"
                    print(err)
                if response:
                    say_message(response)
                break
        else:
            say_message("Неизвестная команда")
vcl = VoiceCommandList()



@vcl.on('привет')
def hello(text):
  return "Здравствуйте, сэр!"



def say_message(message):
    voice = gTTS(message, lang="ru")
    file_voice_name = "_audio_" + str(time.time()) + "_" + str(random.randint(0, 100000)) + ".mp3"
    voice.save(file_voice_name)
    playsound.playsound(file_voice_name)
    print("Голосовой ассистент: " + message)


if __name__ == '__main__':
    while True:
        command = listen_command()
        vcl.run_command(command)

Как сделать чтобы к примеру

@vcl.on('привет' or 'Здравствуй' or 'Доброе утро')
def hello(text):
  return "Здравствуйте, сэр!"

Откликался не только на "Привет", а и на остальные?

Comments

[Алан Гибизов]

Рекомендую прятать длинную простыню кода под спойлер.

[Алан Гибизов]

Также рекомендую убрать из примера всё лишнее и добавить своё (пусть неработающее) решение. Тогда будет предмет для разговора.

[UMFUCHI]

Алан Гибизов, Теперь всё верно?

[Алан Гибизов]

UMFUCHI, дружище… «верно» тут не совсем правильное слово. Но так гораздо лучше.

Answer 1

[Андрей Марченко]

Думаю лучше будет использовать списки чем or

В классе VCL нужно создать функцию on которая будет возвращать декоратор, который в свою очередь будет сохранять информацию в список аля [{func => rules}], Дальше просто ищешь нужный рул, и если он есть - вызываешь функцию к которой он относится.

В таком случае конструкция будет похожа на
@vcl.on(['привет', 'здравствуй, 'доброе утро'])

Можешь глянуть как это реализовано в VKBottle

Comments

[UMFUCHI]

А можешь пожалуйста показать как это должно выглядеть в коде? У меня почему-то гитхаб не грузит

[Андрей Марченко]

UMFUCHI, к сожалению сейчас нет времени. Почти любой фреймворк для разработки ботов имеет такую возможность, изучай

Answer 2

[soremix]

Так же как и с одним словом, по аналогии

def on(self, condition):
    if isinstance(condition, list):
        conditions = [el.lower() for el in condition]
        predicate = lambda text: any(condition in text for condition in conditions)
    elif isinstance(condition, str):
        condition = condition.lower()
        predicate = lambda text: condition in text

Comments

[UMFUCHI]

То-есть должно быть вот так?

class VoiceCommandList:
    def __init__(self):
        self.actions = list()

    def on(self, condition):
        if isinstance(condition, list):
            conditions = [el.lower() for el in condition]
            predicate = lambda text: any(condition in text for condition in conditions)
        elif isinstance(condition, str):
            condition = condition.lower()
            predicate = lambda text: condition in text

        if isinstance(condition, str):
            condition = condition.lower()
            predicate = lambda text: condition in text
        elif callable(condition):
            predicate = condition
        else:
            raise TypeError('Condition must be either string or function!')


        def decorator(command_func):
            self.actions.append((predicate, command_func))
            return command_func

        return decorator

    def run_command(self, text):
        text = text.lower()
        for predicate, command in self.actions:
            if predicate(text):
                try:
                    response = command(text)
                    if response is None:
                        response = "Команда выполнена"
                except Exception as err:
                    response = "Ошибка при выполнении команды"
                    print(err)
                if response:
                    say_message(response)
                break
        else:
            say_message("Неизвестная команда")
vcl = VoiceCommandList()

@vcl.on(lambda text: ('Новая страница' in text) or ('Открой новую страницу' in text))
def new_webpage(text):
    new_page()
    return "Страница открыта"

[soremix]

UMFUCHI, нет. в декоратор передаете либо список, либо строку, либо функцию

[UMFUCHI]

soremix, Можешь показать как должен выглядеть декоратор? Просто пока не сильно разобрался в нём