Make codeowners rules optional by group

Author: KyleFromNVIDIA

src/rapids_pre_commit_hooks/codeowners.py

@@ -120,13 +120,16 @@ def required_codeowners_list(
         ),
     ],
 )
-REQUIRED_CODEOWNERS_LINES = [
-    *REQUIRED_CI_CODEOWNERS_LINES,
-    *REQUIRED_PACKAGING_CODEOWNERS_LINES,
-    *REQUIRED_CPP_CODEOWNERS_LINES,
-    *REQUIRED_PYTHON_CODEOWNERS_LINES,
-    *REQUIRED_CMAKE_CODEOWNERS_LINES,
-]
+
+
+def required_codeowners_lines(args: argparse.Namespace) -> list[RequiredCodeownersLine]:
+    return [
+        *(REQUIRED_CI_CODEOWNERS_LINES if args.ci else []),
+        *(REQUIRED_PACKAGING_CODEOWNERS_LINES if args.packaging else []),
+        *(REQUIRED_CPP_CODEOWNERS_LINES if args.cpp else []),
+        *(REQUIRED_PYTHON_CODEOWNERS_LINES if args.python else []),
+        *(REQUIRED_CMAKE_CODEOWNERS_LINES if args.cmake else []),
+    ]
 
 
 def parse_codeowners_line(line: str, skip: int) -> CodeownersLine | None:
@@ -164,7 +167,7 @@ def check_codeowners_line(
     codeowners_line: CodeownersLine,
     found_files: list[tuple[RequiredCodeownersLine, tuple[int, int]]],
 ) -> None:
-    for required_codeowners_line in REQUIRED_CODEOWNERS_LINES:
+    for required_codeowners_line in required_codeowners_lines(args):
         if required_codeowners_line.file == codeowners_line.file.filename:
             required_owners = [
                 required_owner(project_prefix=args.project_prefix)
@@ -229,7 +232,7 @@ def check_codeowners(linter: Linter, args: argparse.Namespace) -> None:
             check_codeowners_line(linter, args, codeowners_line, found_files)
 
     new_text = ""
-    for required_codeowners_line in REQUIRED_CODEOWNERS_LINES:
+    for required_codeowners_line in required_codeowners_lines(args):
         if required_codeowners_line.file not in map(
             lambda line: line[0].file, found_files
         ):
@@ -258,6 +261,36 @@ def main() -> None:
         help="project prefix to insert for project-specific team names",
         required=True,
     )
+    m.argparser.add_argument(
+        "--ci",
+        help="enforce rules for CI codeowners",
+        action=argparse.BooleanOptionalAction,
+        default=True,
+    )
+    m.argparser.add_argument(
+        "--packaging",
+        help="enforce rules for packaging codeowners",
+        action=argparse.BooleanOptionalAction,
+        default=True,
+    )
+    m.argparser.add_argument(
+        "--cpp",
+        help="enforce rules for C++ codeowners",
+        action=argparse.BooleanOptionalAction,
+        default=True,
+    )
+    m.argparser.add_argument(
+        "--python",
+        help="enforce rules for Python codeowners",
+        action=argparse.BooleanOptionalAction,
+        default=True,
+    )
+    m.argparser.add_argument(
+        "--cmake",
+        help="enforce rules for CMake codeowners",
+        action=argparse.BooleanOptionalAction,
+        default=True,
+    )
     with m.execute() as ctx:
         ctx.add_check(check_codeowners)
 

test/rapids_pre_commit_hooks/test_codeowners.py

@@ -38,8 +38,8 @@
 ]
 
 patch_required_codeowners_lines = patch(
-    "rapids_pre_commit_hooks.codeowners.REQUIRED_CODEOWNERS_LINES",
-    MOCK_REQUIRED_CODEOWNERS_LINES,
+    "rapids_pre_commit_hooks.codeowners.required_codeowners_lines",
+    lambda _args: MOCK_REQUIRED_CODEOWNERS_LINES,
 )