feature/python module

example/create_sphere.py

@@ -0,0 +1,3 @@
+import maya.cmds as cmds
+
+cmds.polySphere()

example/exampleCharacter.py

@@ -79,6 +79,11 @@ def __init__(self, name: str = "ExampleRig") -> None:
             # "MotionModuleParenting": utility.MotionModuleParenting(
             #     self,
             # ),
+            "PythonCreateSphere": utility.PythonCode(
+                self,
+                pythonFile=os.path.join(self.exampleCharacterFolder, "create_sphere.py"),
+                label="PythonCreateSphere",
+            )
         }
         self.exportModules = {
             # "FBXExport": export.FBXExport(

rigsys/modules/utility/runPythonCode.py

@@ -0,0 +1,35 @@
+"""Python Utility module."""
+
+import logging
+import os
+
+from rigsys.modules.utility.utilityBase import UtilityModuleBase
+
+
+logger = logging.getLogger(__name__)
+
+
+class PythonCode(UtilityModuleBase):
+    """Python utility module.
+
+    Takes in a path to a python file and executes it."""
+
+    def __init__(self, rig, pythonFile: str, label: str = "", buildOrder: int = 3000, isMuted: bool = False) -> None:
+        side = "M"
+        mirror = False
+        super().__init__(rig, side, label, buildOrder, isMuted, mirror)
+
+        self.pythonFile = pythonFile
+
+    def run(self):
+        """Run the module (execute the python file)."""
+        if not os.path.exists(self.pythonFile):
+            raise ValueError(f"File does not exist: {self.pythonFile}")
+
+        with open(self.pythonFile) as f:
+            try:
+                code = compile(f.read(), self.pythonFile, 'exec')
+                exec(code, globals(), locals())
+            except Exception as e:
+                logger.error(f"Error executing python file: {self.pythonFile}")
+                raise e

rigsys/test/modules/utility/test_utility_runPythonCode.py

@@ -0,0 +1,55 @@
+"""Test the PythonCode module."""
+
+
+import os
+import unittest
+
+import maya.cmds as cmds
+
+from rigsys import Rig
+import rigsys.modules.utility as utility
+
+
+class TestPythonCode(unittest.TestCase):
+    """Test the PythonCode module."""
+
+    def setUp(self) -> None:
+        cmds.file(new=True, force=True)
+
+        self.rig = Rig()
+
+        self.resourcesFolder = os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, "resources")
+
+        return super().setUp()
+
+    def tearDown(self) -> None:
+        return super().tearDown()
+
+    def test_general(self):
+        """Ensure the module runs properly."""
+        pythonFilePath = os.path.join(self.resourcesFolder, "create_sphere.py")
+
+        self.rig.utilityModules = {
+            "PythonCode": utility.PythonCode(
+                self.rig,
+                pythonFilePath,
+            )
+        }
+
+        self.rig.build()
+
+        self.assertTrue(cmds.objExists("pSphere1"))
+
+    def test_errorHandling(self):
+        """Ensure the module handles errors properly."""
+        pythonFilePath = os.path.join(self.resourcesFolder, "invalid_python.py")
+
+        self.rig.utilityModules = {
+            "PythonCode": utility.PythonCode(
+                self.rig,
+                pythonFilePath,
+            )
+        }
+
+        with self.assertRaises(IndentationError):
+            self.rig.build()

rigsys/test/resources/create_sphere.py

@@ -0,0 +1,3 @@
+import maya.cmds as cmds
+
+cmds.polySphere()

rigsys/test/resources/invalid_python.py

@@ -0,0 +1 @@
+for i in range(10):

rigsys/test/testRunner.py

@@ -1,21 +1,55 @@
 """Run all tests in the test folder."""
 
+import logging
 import os
 import pytest
+import sys
 
 import maya.cmds as cmds
 
 from rigsys.utils.unload import unloadPackages, nukePycFiles
 
 
+logger = logging.getLogger(__name__)
+
+
+class StdOutWrapper:
+    """Wrapper for stdout that fixes pytest capturing."""
+
+    def __init__(self, stdout):
+        """Initialize the wrapper."""
+        self._stdout = stdout
+
+    def __getattr__(self, item):
+        """Get the attribute from the wrapped stdout."""
+        if item == "isatty":
+            return self.isatty
+        else:
+            return getattr(self._stdout, item)
+
+    def isatty(self):
+        """Return False."""
+        return False
+
+
+def fix_stdout():
+    """Fix Maya stdout so that pytest can capture it."""
+    sys.stdout = StdOutWrapper(sys.stdout)
+    sys.stdout._stdout
+
+
 def runTests(pattern="test_"):
     """Run all tests in the test folder."""
     unloadPackages(silent=False, packages=["rigsys"])
     nukePycFiles()
 
     testPath = os.path.dirname(__file__)
 
-    pytest.main([testPath, "-k", pattern])
+    try:
+        pytest.main([testPath, "-k", pattern])
+    except Exception:
+        logger.warning("Error running tests. Trying again with fixed stdout.")
+        fix_stdout()
 
     cmds.file(new=True, force=True)