diff options
| -rw-r--r-- | __main__.py | 8 | ||||
| -rw-r--r-- | config.yaml | 1 | ||||
| -rw-r--r-- | modules/config.py | 6 | ||||
| -rw-r--r-- | modules/responses/libnotify.py | 15 | ||||
| -rw-r--r-- | modules/responses/response_handler.py | 35 | ||||
| -rw-r--r-- | modules/responses/response_manager.py | 32 | ||||
| -rw-r--r-- | resources/hestia_images/hestia.jpg | bin | 0 -> 43277 bytes |
7 files changed, 60 insertions, 37 deletions
diff --git a/__main__.py b/__main__.py index 5196760..7d6f84b 100644 --- a/__main__.py +++ b/__main__.py @@ -9,12 +9,10 @@ from .modules.hassil.recognize import recognize from .modules.hassil.util import merge_dict from .modules.hassil.intents import Intents, TextSlotList -from .modules.responses.response_handler import ResponseHandler +from .modules.responses.response_manager import ResponseManager from .modules.input_handlers.stdin_input import StdinInput from .modules.input_handlers.pipewire_record import PipeWireRecord -from .modules.responses.libnotify import respond - from .modules.intents import * @@ -44,7 +42,7 @@ def main(): } input_handler = PipeWireRecord() if config.input_mode == Config.INPUT_PW else StdinInput() - response_handler = ResponseHandler(config) + response_manager = ResponseManager(config) try: for input_text in input_handler.get_input(): @@ -56,7 +54,7 @@ def main(): } print(result_dict) - response_handler.respond(result.response, result.intent.name) + response_manager.respond(result.response, result.intent.name) handler = getattr(globals()[result_dict["domain"]], result_dict["intent"]) handler(result_dict, config) diff --git a/config.yaml b/config.yaml index cc3c816..ac3069c 100644 --- a/config.yaml +++ b/config.yaml @@ -1,5 +1,6 @@ intents_dir: sentences responses_dir: responses +resources_dir: resources applications_dir: /usr/bin input_mode: pw-record # other option: stdin diff --git a/modules/config.py b/modules/config.py index b0fc624..1a5865b 100644 --- a/modules/config.py +++ b/modules/config.py @@ -15,15 +15,21 @@ class Config: self.intents_dir = "" self.responses_dir = "" + self.resources_dir = "" self.applications_dir = "" self.lock = "" + self.hestia_images = "" + def update(self, **entries) -> None: self.__dict__.update(entries) self.intents_dir = Config.__convert_to_absolute_path(self.intents_dir) self.responses_dir = Config.__convert_to_absolute_path(self.responses_dir) + self.resources_dir = Config.__convert_to_absolute_path(self.resources_dir) + + self.hestia_images = os.path.join(self.resources_dir, "hestia_images") @staticmethod def __convert_to_absolute_path(path: str) -> str: diff --git a/modules/responses/libnotify.py b/modules/responses/libnotify.py index 2e83101..237b9ac 100644 --- a/modules/responses/libnotify.py +++ b/modules/responses/libnotify.py @@ -1,4 +1,15 @@ import os -def respond(text: str) -> None: - os.popen(f'notify-send Hestia "{text}" --app-name=hestia')
\ No newline at end of file +from .response_handler import ResponseHandler +from ..config import Config + + +class LibNotify(ResponseHandler): + def __init__(self, config: Config): + self.config = config + self.hestia_images = {} + for path in os.listdir(self.config.hestia_images): + self.hestia_images[os.path.splitext(os.path.basename(path))[0]] = os.path.join(self.config.hestia_images, path) + + def respond(self, text: str) -> None: + os.popen(f'notify-send Hestia "{text}" -i "{self.hestia_images["hestia"]}" --app-name=hestia')
\ No newline at end of file diff --git a/modules/responses/response_handler.py b/modules/responses/response_handler.py index ee0acca..9cc0cf0 100644 --- a/modules/responses/response_handler.py +++ b/modules/responses/response_handler.py @@ -1,31 +1,6 @@ -import yaml +from abc import ABC, abstractmethod -from pathlib import Path - -from ..config import Config -from ..responses.responses import Responses -from ..hassil.util import merge_dict -from .libnotify import respond as respond_libnotify - -class ResponseHandler: - def __init__(self, config: Config): - response_dict = {"responses": {}} - - response_yaml_path = Path(config.responses_dir) - response_yaml_file_paths = response_yaml_path.glob("*.yaml") - for yaml_file_path in response_yaml_file_paths: - with open(yaml_file_path, "r", encoding="utf-8") as yaml_file: - merge_dict(response_dict, yaml.safe_load(yaml_file)) - - self.responses = Responses.from_dict(response_dict) - self.respond_methods = [respond_libnotify] - - def respond(self, response: str, intent_name: str): - response_key = intent_name if response == "default" else response - if response_key not in self.responses.responses: - print(f"No response found for: {response_key}") - return - - response_text = self.responses.responses[response_key].sentence_texts[0] - for respond_method in self.respond_methods: - respond_method(response_text) +class ResponseHandler(ABC): + @abstractmethod + def respond(self, response: str): + pass
\ No newline at end of file diff --git a/modules/responses/response_manager.py b/modules/responses/response_manager.py new file mode 100644 index 0000000..34c0a41 --- /dev/null +++ b/modules/responses/response_manager.py @@ -0,0 +1,32 @@ +import yaml + +from pathlib import Path + +from ..config import Config +from ..responses.responses import Responses +from ..hassil.util import merge_dict +from .response_handler import ResponseHandler +from .libnotify import LibNotify + +class ResponseManager: + def __init__(self, config: Config): + response_dict = {"responses": {}} + + response_yaml_path = Path(config.responses_dir) + response_yaml_file_paths = response_yaml_path.glob("*.yaml") + for yaml_file_path in response_yaml_file_paths: + with open(yaml_file_path, "r", encoding="utf-8") as yaml_file: + merge_dict(response_dict, yaml.safe_load(yaml_file)) + + self.responses = Responses.from_dict(response_dict) + self.respond_handlers: list[ResponseHandler] = [LibNotify(config)] + + def respond(self, response: str, intent_name: str): + response_key = intent_name if response == "default" else response + if response_key not in self.responses.responses: + print(f"No response found for: {response_key}") + return + + response_text = self.responses.responses[response_key].sentence_texts[0] + for handler in self.respond_handlers: + handler.respond(response_text) diff --git a/resources/hestia_images/hestia.jpg b/resources/hestia_images/hestia.jpg Binary files differnew file mode 100644 index 0000000..3091af0 --- /dev/null +++ b/resources/hestia_images/hestia.jpg |
