Improve remote driver log matching performance

morganstanley · Oct 24, 2024 · ee0f8ac · ee0f8ac
1 parent a1c2861
commit ee0f8ac
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/testplan/common/utils/match.py b/testplan/common/utils/match.py
@@ -196,6 +196,14 @@ def _prepare_regexp(self, regexp: Regex) -> Pattern[AnyStr]:
 
         if isinstance(regexp, (str, bytes)):
             regexp = re.compile(regexp)
+        else:
+            try:
+                import rpyc
+                if isinstance(regexp, rpyc.core.netref.BaseNetref):
+                    regexp = re.compile(regexp.pattern, regexp.flags)
+            except ImportError:
+                pass
+
         try:
             if self.binary and isinstance(regexp.pattern, str):
                 raise TypeError(