feat: adding muted context manager

src/ansys/mapdl/core/mapdl_core.py

@@ -1181,28 +1181,24 @@ def _mesh(self) -> "Archive":
         ``ansys.mapdl.core.Archive``"""
         from ansys.mapdl.reader import Archive
 
+        from ansys.mapdl.core.misc import ChangeProperty
+
         if self._archive_cache is None:
             # write database to an archive file
             arch_filename = os.path.join(self.directory, "_tmp.cdb")
             nblock_filename = os.path.join(self.directory, "nblock.cdb")
 
             # must have all nodes elements are using selected
-            if hasattr(self, "mute"):
-                old_mute = self.mute
-                self.mute = True
-
-            self.cm("__NODE__", "NODE", mute=True)
-            self.nsle("S", mute=True)
-            self.cdwrite("db", arch_filename, mute=True)
-            self.cmsel("S", "__NODE__", "NODE", mute=True)
-
-            self.cm("__ELEM__", "ELEM", mute=True)
-            self.esel("NONE", mute=True)
-            self.cdwrite("db", nblock_filename, mute=True)
-            self.cmsel("S", "__ELEM__", "ELEM", mute=True)
-
-            if hasattr(self, "mute"):
-                self.mute = old_mute
+            with ChangeProperty(mapdl, "mute", True):
+                self.cm("__NODE__", "NODE", mute=True)
+                self.nsle("S", mute=True)
+                self.cdwrite("db", arch_filename, mute=True)
+                self.cmsel("S", "__NODE__", "NODE", mute=True)
+
+                self.cm("__ELEM__", "ELEM", mute=True)
+                self.esel("NONE", mute=True)
+                self.cdwrite("db", nblock_filename, mute=True)
+                self.cmsel("S", "__ELEM__", "ELEM", mute=True)
 
             self._archive_cache = Archive(arch_filename, parse_vtk=False, name="Mesh")
             grid = self._archive_cache._parse_vtk(additional_checking=True)

src/ansys/mapdl/core/misc.py

@@ -34,6 +34,7 @@
 from threading import Thread
 from typing import Callable, Dict, Iterable, List, Tuple, Union
 from warnings import warn
+import weakref
 
 import numpy as np
 
@@ -623,3 +624,18 @@ def deco(f):
         return f
 
     return deco
+
+
+class ChangeProperty:
+    def __init__(self, obj, prop, new_value):
+        self.obj = weakref.ref(obj)
+        self.property = prop
+        self.new_value = new_value
+        self.old_value = None
+
+    def __enter__(self, *args, **kwargs):
+        self.old_value = getattr(self.obj(), self.property)
+        setattr(self.obj(), self.property, self.new_value)
+
+    def __exit__(self, *args, **kwargs):
+        setattr(self.obj(), self.property, self.old_value)

tests/test_misc.py

@@ -28,6 +28,7 @@
 import pytest
 
 from ansys.mapdl.core.misc import (
+    ChangeProperty,
     check_valid_ip,
     check_valid_port,
     check_valid_routine,
@@ -272,3 +273,22 @@ def test_create_temp_dir():
     dir_ = path.parts[-1]
 
     assert str(path) != create_temp_dir(parent, dir_)
+
+
+@pytest.mark.parametrize("prop", ["mute"])
+def test_change_property_context(mapdl, prop):
+    assert not mapdl.mute
+    with ChangeProperty(mapdl, prop, True):
+        assert getattr(mapdl, prop) is True
+
+    assert not mapdl.mute
+
+
+def test_change_property_context_method(mapdl):
+    type_ = type(mapdl._log)
+    assert mapdl._log
+    with ChangeProperty(mapdl, "_log", True):
+        assert getattr(mapdl, "_log") is True
+
+    assert mapdl._log
+    assert isinstance(mapdl._log, type_)