diff options
Diffstat (limited to 'buildscripts/linter/filediff.py')
| -rw-r--r-- | buildscripts/linter/filediff.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/buildscripts/linter/filediff.py b/buildscripts/linter/filediff.py index 64521d9ed47..7dbad6d9d58 100644 --- a/buildscripts/linter/filediff.py +++ b/buildscripts/linter/filediff.py @@ -19,15 +19,18 @@ from buildscripts.patch_builds.change_data import generate_revision_map, \ LOGGER = structlog.get_logger(__name__) MONGO_REVISION_ENV_VAR = "REVISION" +ENTERPRISE_REVISION_ENV_VAR = "ENTERPRISE_REV" def _get_repos_and_revisions() -> Tuple[List[Repo], RevisionMap]: """Get the repo object and a map of revisions to compare against.""" modules = git.get_module_paths() repos = [Repo(path) for path in modules] - revision_map = generate_revision_map(repos, { - "mongo": os.environ.get(MONGO_REVISION_ENV_VAR), - }) + revision_map = generate_revision_map( + repos, { + "mongo": os.environ.get(MONGO_REVISION_ENV_VAR), + "enterprise": os.environ.get(ENTERPRISE_REVISION_ENV_VAR) + }) return repos, revision_map |