summaryrefslogtreecommitdiff
path: root/modules/responses
diff options
context:
space:
mode:
Diffstat (limited to 'modules/responses')
-rw-r--r--modules/responses/__init__.py0
-rw-r--r--modules/responses/libnotify.py15
-rw-r--r--modules/responses/response_handler.py6
-rw-r--r--modules/responses/response_manager.py32
-rw-r--r--modules/responses/responses.py29
5 files changed, 82 insertions, 0 deletions
diff --git a/modules/responses/__init__.py b/modules/responses/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/modules/responses/__init__.py
diff --git a/modules/responses/libnotify.py b/modules/responses/libnotify.py
new file mode 100644
index 0000000..0bc774f
--- /dev/null
+++ b/modules/responses/libnotify.py
@@ -0,0 +1,15 @@
+import os
+
+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
new file mode 100644
index 0000000..9cc0cf0
--- /dev/null
+++ b/modules/responses/response_handler.py
@@ -0,0 +1,6 @@
+from abc import ABC, abstractmethod
+
+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/modules/responses/responses.py b/modules/responses/responses.py
new file mode 100644
index 0000000..f89a1fe
--- /dev/null
+++ b/modules/responses/responses.py
@@ -0,0 +1,29 @@
+from dataclasses import dataclass, field
+from typing import List, Dict, Any
+
+@dataclass(frozen=True)
+class Response:
+ name: str
+ sentence_texts: List[str]
+
+@dataclass(frozen=True)
+class Responses:
+ responses: Dict[str, Response]
+
+ @staticmethod
+ def from_dict(input_dict: Dict[str, Any]) -> "Responses":
+ # responses:
+ # ResponseName:
+ # data:
+ # - sentences:
+ # - "<sentence>"
+ return Responses(
+ responses={
+ response_name: Response(
+ name=response_name,
+ sentence_texts=response_dict["sentences"]
+ )
+ for response_name, response_dict in input_dict["responses"].items()
+ },
+ )
+