1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
"""Modules for find which files should be linted."""
import os
import sys
from typing import Tuple, List, Dict, Callable
from git import Repo
import structlog
# Get relative imports to work when the package is not installed on the PYTHONPATH.
if __name__ == "__main__" and __package__ is None:
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(os.path.realpath(__file__)))))
# pylint: disable=wrong-import-position
from buildscripts.linter import git
from buildscripts.patch_builds.change_data import generate_revision_map, \
RevisionMap, find_changed_files_in_repos
# pylint: enable=wrong-import-position
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),
"enterprise": os.environ.get(ENTERPRISE_REVISION_ENV_VAR)
})
return repos, revision_map
def _filter_file(filename: str, is_interesting_file: Callable) -> bool:
"""
Determine if file should be included based on existence and passed in method.
:param filename: Filename to check.
:param is_interesting_file: Function to determine if file is interesting.
:return: True if file exists and is interesting.
"""
return os.path.exists(filename) and is_interesting_file(filename)
def gather_changed_files_for_lint(is_interesting_file: Callable) -> List[str]:
"""
Get the files that have changes since the last git commit.
:param is_interesting_file: Filter for whether a file should be returned.
:return: List of files for linting.
"""
repos, revision_map = _get_repos_and_revisions()
LOGGER.info("revisions", revision=revision_map)
candidate_files = find_changed_files_in_repos(repos, revision_map)
files = [
filename for filename in candidate_files if _filter_file(filename, is_interesting_file)
]
LOGGER.info("Found files to lint", files=files)
return files
|
