From a585d159a5acf03d0d75fc39d74ff5ee24e28b8a Mon Sep 17 00:00:00 2001 From: Botond Hende Date: Wed, 16 Oct 2024 17:32:13 +0200 Subject: name based intent handling --- __init__.py | 0 __main__.py | 47 +++++++++++++++++++++++++++++++++++++++++++++ modules/intents/__init__.py | 4 ++++ modules/intents/process.py | 12 ++++++++++++ 4 files changed, 63 insertions(+) create mode 100644 __init__.py create mode 100644 __main__.py create mode 100644 modules/intents/__init__.py create mode 100644 modules/intents/process.py diff --git a/__init__.py b/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/__main__.py b/__main__.py new file mode 100644 index 0000000..79d6a23 --- /dev/null +++ b/__main__.py @@ -0,0 +1,47 @@ +import sys +import yaml + +from pathlib import Path +from .modules.hassil.recognize import recognize +from .modules.hassil.util import merge_dict +from .modules.hassil.intents import Intents, TextSlotList + +from .modules.intents import * + + +def main(): + input_dict = {"intents": {}} + + yaml_path = Path(sys.argv[1]) + yaml_file_paths = yaml_path.glob("*.yaml") + + slot_lists = {} + + for yaml_file_path in yaml_file_paths: + with open(yaml_file_path, "r", encoding="utf-8") as yaml_file: + merge_dict(input_dict, yaml.safe_load(yaml_file)) + + intents = Intents.from_dict(input_dict) + + try: + for line in sys.stdin: + line = line.strip() + if not line: + continue + + result = recognize(line, intents, slot_lists=slot_lists) + if result is not None: + result_dict = { + "intent": result.intent.name, + **{e.name: e.value for e in result.entities_list}, + } + print(result_dict) + handler = getattr(globals()[result_dict["domain"]], result_dict["intent"]) + handler(result_dict) + else: + print("") + except KeyboardInterrupt: + pass + +if __name__ == '__main__': + main() \ No newline at end of file diff --git a/modules/intents/__init__.py b/modules/intents/__init__.py new file mode 100644 index 0000000..1cb91d4 --- /dev/null +++ b/modules/intents/__init__.py @@ -0,0 +1,4 @@ +from os.path import dirname, basename, isfile, join +import glob +modules = glob.glob(join(dirname(__file__), "*.py")) +__all__ = [ basename(f)[:-3] for f in modules if isfile(f) and not f.endswith('__init__.py')] \ No newline at end of file diff --git a/modules/intents/process.py b/modules/intents/process.py new file mode 100644 index 0000000..4b3fdab --- /dev/null +++ b/modules/intents/process.py @@ -0,0 +1,12 @@ +import os +from typing import Dict + + +def switch_to_workspace(data: Dict): + os.popen(f"swaymsg workspace {data["workspace"]}") + +def HesExecuteProcess(data: Dict): + if "workspace" in data: + switch_to_workspace(data) + + os.popen(data["process"].strip()) -- cgit v1.2.3-70-g09d2