diff --git a/ansible_mitogen/planner.py b/ansible_mitogen/planner.py index c0913a3e..a17189a3 100644 --- a/ansible_mitogen/planner.py +++ b/ansible_mitogen/planner.py @@ -41,6 +41,7 @@ import json import logging import os import random +import re from ansible.executor import module_common from ansible.collections.list import list_collection_dirs @@ -297,11 +298,11 @@ class NewStylePlanner(ScriptPlanner): preprocessing the module. """ runner_name = 'NewStyleRunner' - marker = b'from ansible.module_utils.' + MARKER = re.compile(b'from ansible(?:_collections|\.module_utils)\.') @classmethod def detect(cls, path, source): - return cls.marker in source + return cls.MARKER.search(source) != None def _get_interpreter(self): return None, None