ansys-mechanical-core 0.11.11__py3-none-any.whl → 0.11.13__py3-none-any.whl

ansys/mechanical/core/embedding/rpc/server.py

@@ -0,0 +1,382 @@
+# Copyright (C) 2022 - 2025 ANSYS, Inc. and/or its affiliates.
+# SPDX-License-Identifier: MIT
+#
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+"""Remote Procedure Call (RPC) server."""
+
+import fnmatch
+import os
+import typing
+
+import rpyc
+from rpyc.utils.server import ThreadedServer
+import toolz
+
+from ansys.mechanical.core.embedding.background import BackgroundApp
+from ansys.mechanical.core.mechanical import port_in_use
+
+from .utils import MethodType, get_remote_methods, remote_method
+
+# TODO : implement logging
+
+PYMECHANICAL_DEFAULT_RPC_PORT = 20000
+
+
+class MechanicalService(rpyc.Service):
+    """Starts Mechanical app services."""
+
+    def __init__(self, backgroundapp, functions=[], impl=None):
+        """Initialize the service."""
+        super().__init__()
+        self._backgroundapp = backgroundapp
+        self._install_functions(functions)
+        self._install_class(impl)
+        self.EMBEDDED = True
+
+    def _install_functions(self, methods):
+        """Install the given list of methods."""
+        [self._install_function(method) for method in methods]
+
+    def _install_class(self, impl):
+        """Install methods from the given implemented class."""
+        print("Install class")
+        if impl is None:
+            return
+        print("Installing methods from class")
+        for methodname, method, methodtype in get_remote_methods(impl):
+            print(f"installing {methodname} of {impl}")
+            if methodtype == MethodType.METHOD:
+                self._install_method(method)
+            elif methodtype == MethodType.PROP:
+                self._install_property(method, methodname)
+
+    def on_connect(self, conn):
+        """Handle client connection."""
+        print("Client connected")
+        print(self._backgroundapp.app)
+
+    def on_disconnect(self, conn):
+        """Handle client disconnection."""
+        print("Client disconnected")
+
+    def _curry_property(self, prop, propname, get: bool):
+        """Curry the given property."""
+
+        def posted(*arg):
+            def curried():
+                if get:
+                    return getattr(prop._owner, propname)
+                else:
+                    setattr(prop._owner, propname, *arg)
+
+            return self._backgroundapp.try_post(curried)
+
+        return posted
+
+    def _curry_method(self, method, realmethodname):
+        """Curry the given method."""
+
+        def posted(*args, **kwargs):
+            def curried():
+                original_method = getattr(method._owner, realmethodname)
+                result = original_method(*args, **kwargs)
+                return result
+
+            return self._backgroundapp.try_post(curried)
+
+        return posted
+
+    def _curry_function(self, methodname):
+        """Curry the given function."""
+        wrapped = getattr(self, methodname)
+        curried_method = toolz.curry(wrapped)
+
+        def posted(*args, **kwargs):
+            def curried():
+                return curried_method(self._app, *args, **kwargs)
+
+            return self._backgroundapp.try_post(curried)
+
+        return posted
+
+    def _install_property(self, prop: property, propname: str):
+        """Install property with inner and exposed pairs."""
+        if prop.fget:
+            exposed_get_name = f"exposed_propget_{propname}"
+
+            def exposed_propget():
+                """Convert to exposed getter."""
+                f = self._curry_property(prop.fget, propname, True)
+                result = f()
+                return result
+
+            setattr(self, exposed_get_name, exposed_propget)
+        if prop.fset:
+            exposed_set_name = f"exposed_propset_{propname}"
+
+            def exposed_propset(arg):
+                """Convert to exposed getter."""
+                f = self._curry_property(prop.fset, propname, True)
+                result = f(arg)
+                return result
+
+            setattr(self, exposed_set_name, exposed_propset)
+
+    def _install_method(self, method):
+        methodname = method.__name__
+        self._install_method_with_name(method, methodname, methodname)
+
+    def _install_method_with_name(self, method, methodname, innername):
+        """Install methods of impl with inner and exposed pairs."""
+        exposed_name = f"exposed_{methodname}"
+
+        def exposed_method(*args, **kwargs):
+            """Convert to exposed method."""
+            f = self._curry_method(method, innername)
+            result = f(*args, **kwargs)
+            return result
+
+        setattr(self, exposed_name, exposed_method)
+
+    def _install_function(self, function):
+        """Install a functions with inner and exposed pairs."""
+        print(f"Installing {function}")
+        exposed_name = f"exposed_{function.__name__}"
+        inner_name = f"inner_{function.__name__}"
+
+        def inner_method(app, *args, **kwargs):
+            """Convert to inner method."""
+            return function(app, *args, **kwargs)
+
+        def exposed_method(*args, **kwargs):
+            """Convert to exposed method."""
+            f = self._curry_function(inner_name)
+            return f(*args, **kwargs)
+
+        setattr(self, inner_name, inner_method)
+        setattr(self, exposed_name, exposed_method)
+
+    def exposed_service_upload(self, remote_path, file_data):
+        """Handle file upload request from client."""
+        if not remote_path:
+            raise ValueError("The remote file path is empty.")
+
+        remote_dir = os.path.dirname(remote_path)
+
+        if remote_dir:
+            os.makedirs(remote_dir, exist_ok=True)
+
+        with open(remote_path, "wb") as f:
+            f.write(file_data)
+
+        print(f"File {remote_path} uploaded successfully.")
+
+    def exposed_service_download(self, remote_path):
+        """Handle file download request from client."""
+        # Check if the remote file exists
+        if not os.path.exists(remote_path):
+            raise FileNotFoundError(f"The file {remote_path} does not exist on the server.")
+
+        if os.path.isdir(remote_path):
+            files = []
+            for dirpath, _, filenames in os.walk(remote_path):
+                for filename in filenames:
+                    full_path = os.path.join(dirpath, filename)
+                    relative_path = os.path.relpath(full_path, remote_path)
+                    files.append(relative_path)
+            return {"is_directory": True, "files": files}
+
+        with open(remote_path, "rb") as f:
+            file_data = f.read()
+
+        print(f"File {remote_path} downloaded successfully.")
+        return file_data
+
+    def exposed_service_exit(self):
+        """Exit the server."""
+        self._backgroundapp.stop()
+        self._backgroundapp = None
+
+
+class MechanicalEmbeddedServer:
+    """Start rpc server."""
+
+    def __init__(
+        self,
+        service: typing.Type[rpyc.Service] = MechanicalService,
+        port: int = None,
+        version: int = None,
+        methods: typing.List[typing.Callable] = [],
+        impl=None,
+    ):
+        """Initialize the server."""
+        self._exited = False
+        self._background_app = BackgroundApp(version=version)
+        self._service = service
+        self._methods = methods
+        print("Initializing Mechanical ...")
+
+        self._port = self.get_free_port(port)
+        if impl is None:
+            self._impl = None
+        else:
+            self._impl = impl(self._background_app.app)
+
+        my_service = self._service(self._background_app, self._methods, self._impl)
+        self._server = ThreadedServer(my_service, port=self._port)
+
+    @staticmethod
+    def get_free_port(port=None):
+        """Get free port.
+
+        If port is not given, it will find a free port starting from PYMECHANICAL_DEFAULT_RPC_PORT.
+        """
+        if port is None:
+            port = PYMECHANICAL_DEFAULT_RPC_PORT
+
+        while port_in_use(port):
+            port += 1
+
+        return port
+
+    def start(self) -> None:
+        """Start server on specified port."""
+        print(
+            f"Starting mechanical application in server.\n"
+            f"Listening on port {self._port}\n{self._background_app.app}"
+        )
+        self._server.start()
+        """try:
+            try:
+                conn.serve_all()
+            except KeyboardInterrupt:
+                print("User interrupt!")
+        finally:
+            conn.close()"""
+        self._exited = True
+
+    def stop(self) -> None:
+        """Stop the server."""
+        print("Stopping the server...")
+        self._background_app.stop()
+        self._server.close()
+        self._exited = True
+        print("Server stopped.")
+
+
+class DefaultServiceMethods:
+    """Default service methods for MechanicalEmbeddedServer."""
+
+    def __init__(self, app):
+        """Initialize the DefaultServiceMethods."""
+        self._app = app
+
+    def __repr__(self):
+        """Return the representation of the instance."""
+        return '"ServiceMethods instance"'
+
+    @remote_method
+    def run_python_script(
+        self, script: str, enable_logging=False, log_level="WARNING", progress_interval=2000
+    ):
+        """Run scripts using Internal python engine."""
+        result = self._app.execute_script(script)
+        return result
+
+    @remote_method
+    def run_python_script_from_file(
+        self,
+        file_path: str,
+        enable_logging=False,
+        log_level="WARNING",
+        progress_interval=2000,
+    ):
+        """Run scripts using Internal python engine."""
+        return self._app.execute_script_from_file(file_path)
+
+    @remote_method
+    def clear(self):
+        """Clear the current project."""
+        self._app.new()
+
+    @property
+    @remote_method
+    def project_directory(self):
+        """Get the project directory."""
+        return self._app.execute_script("""ExtAPI.DataModel.Project.ProjectDirectory""")
+
+    @remote_method
+    def list_files(self):
+        """List all files in the project directory."""
+        list = []
+        mechdbPath = self._app.execute_script("""ExtAPI.DataModel.Project.FilePath""")
+        if mechdbPath != "":
+            list.append(mechdbPath)
+        rootDir = self._app.execute_script("""ExtAPI.DataModel.Project.ProjectDirectory""")
+
+        for dirPath, dirNames, fileNames in os.walk(rootDir):
+            for fileName in fileNames:
+                list.append(os.path.join(dirPath, fileName))
+        files_out = "\n".join(list).splitlines()
+        if not files_out:  # pragma: no cover
+            print("No files listed")
+        return files_out
+
+    @remote_method
+    def _get_files(self, files, recursive=False):
+        self_files = self.list_files()  # to avoid calling it too much
+
+        if isinstance(files, str):
+            if files in self_files:
+                list_files = [files]
+            elif "*" in files:
+                list_files = fnmatch.filter(self_files, files)
+                if not list_files:
+                    raise ValueError(
+                        f"The `'files'` parameter ({files}) didn't match any file using "
+                        f"glob expressions in the remote server."
+                    )
+            else:
+                raise ValueError(
+                    f"The `'files'` parameter ('{files}') does not match any file or pattern."
+                )
+
+        elif isinstance(files, (list, tuple)):
+            if not all([isinstance(each, str) for each in files]):
+                raise ValueError(
+                    "The parameter `'files'` can be a list or tuple, but it "
+                    "should only contain strings."
+                )
+            list_files = files
+        else:
+            raise ValueError(
+                f"The `file` parameter type ({type(files)}) is not supported."
+                "Only strings, tuple of strings, or list of strings are allowed."
+            )
+
+        return list_files
+
+
+class MechanicalDefaultServer(MechanicalEmbeddedServer):
+    """Default server with default service methods."""
+
+    def __init__(self, **kwargs):
+        """Initialize the MechanicalDefaultServer."""
+        super().__init__(service=MechanicalService, impl=DefaultServiceMethods, **kwargs)

ansys/mechanical/core/embedding/rpc/utils.py

@@ -0,0 +1,120 @@
+# Copyright (C) 2022 - 2025 ANSYS, Inc. and/or its affiliates.
+# SPDX-License-Identifier: MIT
+#
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+"""Utilities necessary for remote calls."""
+import typing
+
+
+class remote_method:
+    """Decorator for passing remote methods."""
+
+    def __init__(self, func):
+        """Initialize with the given function."""
+        self._func = func
+
+    def __call__(self, *args, **kwargs):
+        """Call the stored function with provided arguments."""
+        return self._func(*args, **kwargs)
+
+    def __call_method__(self, instance, *args, **kwargs):
+        """Call the stored function with the instance and provided arguments."""
+        return self._func(instance, *args, **kwargs)
+
+    def __get__(self, obj, objtype):
+        """Return a partially applied method."""
+        from functools import partial
+
+        func = partial(self.__call_method__, obj)
+        func._is_remote = True
+        func.__name__ = self._func.__name__
+        func._owner = obj
+        return func
+
+
+class MethodType:
+    """Enum for method or property types."""
+
+    METHOD = 0
+    PROP = 1
+
+
+def try_get_remote_method(methodname: str, obj: typing.Any) -> typing.Tuple[str, typing.Callable]:
+    """Try to get a remote method."""
+    method = getattr(obj, methodname)
+    if not callable(method):
+        return None
+    if hasattr(method, "_is_remote") and method._is_remote is True:
+        return (methodname, method)
+
+
+def try_get_remote_property(attrname: str, obj: typing.Any) -> typing.Tuple[str, property]:
+    """Try to get a remote property."""
+    objclass: typing.Type = obj.__class__
+    class_attribute = getattr(objclass, attrname)
+    getmethod = None
+    setmethod = None
+
+    if class_attribute.fget:
+        if isinstance(class_attribute.fget, remote_method):
+            getmethod = class_attribute.fget
+            getmethod._owner = obj
+    if class_attribute.fset:
+        if isinstance(class_attribute.fset, remote_method):
+            setmethod = class_attribute.fset
+            setmethod._owner = obj
+
+    return (attrname, property(getmethod, setmethod))
+
+
+def get_remote_methods(
+    obj,
+) -> typing.Generator[typing.Tuple[str, typing.Callable, MethodType], None, None]:
+    """Yield names and methods of an object's remote methods.
+
+    A remote method is identified by the presence of an attribute `_is_remote` set to `True`.
+
+    Parameters
+    ----------
+    obj: Any
+        The object to inspect for remote methods.
+
+    Yields
+    ------
+    Generator[Tuple[str, Callable], None, None]
+        A tuple containing the method name and the method itself
+        for each remote method found in the object
+    """
+    print(f"Getting remote methods on {obj}")
+    objclass = obj.__class__
+    for attrname in dir(obj):
+        if attrname.startswith("__"):
+            continue
+        print(attrname)
+        if hasattr(objclass, attrname):
+            class_attribute = getattr(objclass, attrname)
+            if isinstance(class_attribute, property):
+                attrname, prop = try_get_remote_property(attrname, obj)
+                yield attrname, prop, MethodType.PROP
+                continue
+        result = try_get_remote_method(attrname, obj)
+        if result != None:
+            attrname, method = result
+            yield attrname, method, MethodType.METHOD

ansys/mechanical/core/embedding/runtime.py

@@ -22,6 +22,7 @@
 
 """Runtime initialize for pythonnet in embedding."""
 
+from importlib.metadata import distribution
 import os
 
 from ansys.mechanical.core.embedding.logger import Logger
@@ -44,6 +45,25 @@ def __register_function_codec():
     Ansys.Mechanical.CPython.Codecs.FunctionCodec.Register()
 
 
+def _bind_assembly_for_explicit_interface(assembly_name: str):
+    """Bind the assembly for explicit interface implementation."""
+    # if pythonnet is not installed, we can't bind the assembly
+    try:
+        distribution("pythonnet")
+        return
+    except ModuleNotFoundError:
+        pass
+
+    import clr
+
+    assembly = clr.AddReference(assembly_name)
+    from Python.Runtime import BindingManager, BindingOptions
+
+    binding_options = BindingOptions()
+    binding_options.AllowExplicitInterfaceImplementation = True
+    BindingManager.SetBindingOptions(assembly, binding_options)
+
+
 def initialize(version: int) -> None:
     """Initialize the runtime.
 
@@ -59,3 +79,5 @@ def initialize(version: int) -> None:
         # function codec is distributed with pymechanical on linux only
         # at version 242 or later
         __register_function_codec()
+
+    _bind_assembly_for_explicit_interface("Ansys.ACT.WB1")

ansys/mechanical/core/feature_flags.py

@@ -31,6 +31,7 @@ class FeatureFlags:
 
     ThermalShells = "Mechanical.ThermalShells"
     MultistageHarmonic = "Mechanical.MultistageHarmonic"
+    CPython = "Mechanical.CPython.Capability"
 
 
 def get_feature_flag_names() -> typing.List[str]:

ansys/mechanical/core/ide_config.py

@@ -88,7 +88,7 @@ def _vscode_impl(
         The type of settings to update. Either "user" or "workspace" in VS Code.
         By default, it's ``user``.
     revision: int
-        The Mechanical revision number. For example, "242".
+        The Mechanical revision number. For example, "251".
         If unspecified, it finds the default Mechanical version from ansys-tools-path.
     """
     # Update the user or workspace settings
@@ -144,7 +144,7 @@ def _cli_impl(
         The type of settings to update. Either "user" or "workspace" in VS Code.
         By default, it's ``user``.
     revision: int
-        The Mechanical revision number. For example, "242".
+        The Mechanical revision number. For example, "251".
         If unspecified, it finds the default Mechanical version from ansys-tools-path.
     """
     # Get the ansys-mechanical-stubs install location
@@ -178,7 +178,7 @@ def _cli_impl(
     "--revision",
     default=None,
     type=int,
-    help='The Mechanical revision number, e.g. "242" or "241". If unspecified,\
+    help='The Mechanical revision number, e.g. "251" or "242". If unspecified,\
 it finds and uses the default version from ansys-tools-path.',
 )
 def cli(ide: str, target: str, revision: int) -> None:
@@ -192,14 +192,14 @@ def cli(ide: str, target: str, revision: int) -> None:
         The type of settings to update. Either "user" or "workspace" in VS Code.
         By default, it's ``user``.
     revision: int
-        The Mechanical revision number. For example, "242".
+        The Mechanical revision number. For example, "251".
         If unspecified, it finds the default Mechanical version from ansys-tools-path.
 
     Usage
     -----
     The following example demonstrates the main use of this tool:
 
-        $ ansys-mechanical-ideconfig --ide vscode --target user --revision 242
+        $ ansys-mechanical-ideconfig --ide vscode --target user --revision 251
 
     """
     exe = atp.get_mechanical_path(allow_input=False, version=revision)