refactor: modular codegen system (#734)

Co-authored-by: Mohamed Koubaa <[email protected]>
Co-authored-by: pyansys-ci-bot <[email protected]>

Author: 3 people

codegen/generate.py

@@ -0,0 +1,90 @@
+import copy
+import json
+import os
+import typing
+
+KWDM_INSTANCE = None
+MANIFEST = None
+ADDITIONAL_CARDS = None
+
+KWD_TO_ALIAS: typing.Dict[str, str] = {}
+ALIAS_TO_KWD: typing.Dict[str, str] = {}
+
+from .insertion import Insertion
+
+
+def get_card(setting: typing.Dict[str, str]):
+    source = setting["source"]
+    if source == "kwd-data":
+        data = KWDM_INSTANCE.get_keyword_data_dict(setting["keyword-name"])
+        card = data[setting["card-index"]]
+        return card
+
+    if source == "additional-cards":
+        return ADDITIONAL_CARDS[setting["card-name"]]
+
+    raise Exception()
+
+
+def add_alias(keyword: str, alias: str):
+    KWD_TO_ALIAS[keyword] = alias
+    ALIAS_TO_KWD[alias] = keyword
+
+
+def get_alias(keyword: str) -> typing.Optional[str]:
+    return KWD_TO_ALIAS.get(keyword, None)
+
+
+def get_aliased_by(keyword: str):
+    return ALIAS_TO_KWD.get(keyword, None)
+
+
+def is_aliased(keyword: str):
+    return keyword in ALIAS_TO_KWD.keys()
+
+
+def _load_manifest(filename) -> typing.Dict:
+    with open(filename) as f:
+        manifest = json.load(f)
+    return manifest
+
+
+class AdditionalCards:
+    def __init__(self, filename):
+        with open(filename) as f:
+            self._cards = json.load(f)
+
+    def __getitem__(self, name):
+        """return a copy of the additional card, since the client may mutate it."""
+        return copy.deepcopy(self._cards[name])
+
+
+class KWDM:
+    def __init__(self, filename):
+        with open(filename, encoding="utf-8") as f:
+            self._data: typing.Dict = json.load(f)
+
+    def get_keywords_list(self) -> typing.List[str]:
+        return list(self._data.keys())
+
+    def get_keyword_data_dict(self, name: str) -> typing.Dict:
+        return copy.deepcopy(self[name])
+
+    def __getitem__(self, name: str) -> typing.Dict:
+        return self._data[name]
+
+
+def load(this_folder: str, kwd_file: str, manifest: str, additional_cards: str):
+    global KWDM_INSTANCE, MANIFEST, ADDITIONAL_CARDS
+    if kwd_file == "":
+        KWDM_INSTANCE = KWDM(os.path.join(this_folder, "kwd.json"))
+    else:
+        KWDM_INSTANCE = KWDM(kwd_file)
+    if manifest == "":
+        MANIFEST = _load_manifest(this_folder / "manifest.json")
+    else:
+        MANIFEST = _load_manifest(manifest)
+    if additional_cards == "":
+        ADDITIONAL_CARDS = AdditionalCards(this_folder / "additional-cards.json")
+    else:
+        ADDITIONAL_CARDS = AdditionalCards(additional_cards)

codegen/keyword_generation/data_model/__init__.py

@@ -0,0 +1,9 @@
+from dataclasses import dataclass
+import typing
+
+
+@dataclass
+class Insertion:
+    target_index: int = None
+    target_class: str = None
+    card: typing.Dict = None

codegen/keyword_generation/data_model/insertion.py

@@ -0,0 +1,2 @@
+from .class_generator import generate_class
+from .entrypoints_generator import generate_entrypoints