feat: add missing APIs (#93)

Co-authored-by: pyansys-ci-bot <[email protected]>
ansys · Dec 20, 2024 · c6d1ad0 · c6d1ad0
1 parent dfd29e4
commit c6d1ad0
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/doc/changelog.d/93.added.md b/doc/changelog.d/93.added.md
@@ -0,0 +1 @@
+add missing APIs
diff --git a/src/ansys/mechanical/stubs/stub_generator/create_files.py b/src/ansys/mechanical/stubs/stub_generator/create_files.py
@@ -32,7 +32,11 @@
 
 import System  # isort: skip
 
-ACCEPTED_TYPES = {"Ansys.Core.Units.Quantity"}
+ACCEPTED_TYPES = {
+    "Ansys.Core.Units.Quantity",
+    "Ansys.ACT.Interfaces.Common",
+    "Ansys.Mechanical.DataModel.Interfaces.IDataModelObject",
+}
 
 
 def get_version():
@@ -199,8 +203,16 @@ def make(base_dir, outdir, assemblies, str_version):
                         contents = contents.replace(
                             "import typing", f"import typing\n{''.join(import_statements)}"
                         )
+
                         f.write(contents)
 
+                        datamodel_interfaces = (
+                            pathlib.Path("Ansys") / "Mechanical" / "DataModel" / "Interfaces"
+                        )
+                        if str(datamodel_interfaces) in str(init_path):
+                            f.write("class DataModelObject(IDataModelObject):\n")
+                            f.write("    pass\n")
+
     print("Done processing all mechanical stubs.")
 
 
@@ -246,6 +258,7 @@ def main():
     assemblies = [
         "Ansys.Mechanical.DataModel",
         "Ansys.Mechanical.Interfaces",
+        "Ansys.ACT.Interfaces",
         "Ansys.ACT.WB1",
         "Ans.Core",
     ]

diff --git a/src/ansys/mechanical/stubs/stub_generator/generate_content.py b/src/ansys/mechanical/stubs/stub_generator/generate_content.py
@@ -884,6 +884,8 @@ def make(outdir: str, assembly_name: str, type_filter: typing.Callable = None) -
 
     if assembly_name == "Ans.Core":
         namespaces = {"Ansys.Core.Units": namespaces["Ansys.Core.Units"]}
+    elif assembly_name == "Ansys.ACT.Interfaces":
+        namespaces = {"Ansys.ACT.Interfaces.Common": namespaces["Ansys.ACT.Interfaces.Common"]}
 
     dump_types(namespaces)
     doc = get_doc(assembly)