From: Nicki Křížek <nicki@isc.org>
Date: Tue, 21 Oct 2025 16:58:26 +0000 (+0200)
Subject: Add pylint check for re.compile() alias
X-Git-Tag: v9.21.17~53^2~2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=a8bf53411d14b8b009cb3a6b3c5bc12a5fd7a2ea;p=thirdparty%2Fbind9.git

Add pylint check for re.compile() alias

Ensure that Re() is used consistently across our code base.
---

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 9a0c47b7b1e..89463e5f6f2 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -671,7 +671,7 @@ vulture:
   <<: *python_triggering_rules
   needs: []
   script:
-    - vulture --exclude "*ans.py,conftest.py,isctest" --ignore-names "after_servers_start,bootstrap,pytestmark" bin/tests/system/
+    - vulture --exclude "*ans.py,conftest.py,re_compile_checker.py,isctest" --ignore-names "after_servers_start,bootstrap,pytestmark" bin/tests/system/
 
 ci-variables:
   <<: *precheck_job
@@ -780,7 +780,7 @@ pylint:
   script:
     - pylint --rcfile $CI_PROJECT_DIR/.pylintrc $(git ls-files '*.py' | grep -vE '(ans\.py|dangerfile\.py|^bin/tests/system/|^contrib/)')
       # Ignore Pylint wrong-import-position error in system test to enable use of pytest.importorskip
-    - pylint --rcfile $CI_PROJECT_DIR/.pylintrc --disable=wrong-import-position $(git ls-files 'bin/tests/system/*.py' | grep -vE '(ans\.py|vulture_ignore_list\.py)')
+    - pylint --rcfile $CI_PROJECT_DIR/.pylintrc --load-plugins re_compile_checker --disable=wrong-import-position $(git ls-files 'bin/tests/system/*.py' | grep -vE '(ans\.py|vulture_ignore_list\.py)')
 
 reuse:
   <<: *precheck_job
diff --git a/bin/tests/system/re_compile_checker.py b/bin/tests/system/re_compile_checker.py
new file mode 100644
index 00000000000..efa0e9a038c
--- /dev/null
+++ b/bin/tests/system/re_compile_checker.py
@@ -0,0 +1,46 @@
+# Copyright (C) Internet Systems Consortium, Inc. ("ISC")
+#
+# SPDX-License-Identifier: MPL-2.0
+#
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0.  If a copy of the MPL was not distributed with this
+# file, you can obtain one at https://mozilla.org/MPL/2.0/.
+#
+# See the COPYRIGHT file distributed with this work for additional
+# information regarding copyright ownership.
+
+# pylint: disable=unknown-option-value,re-compile-alias
+
+import re
+
+from astroid import nodes
+
+from pylint.checkers import BaseRawFileChecker
+from pylint.lint import PyLinter
+
+
+class ReCompileChecker(BaseRawFileChecker):
+
+    name = "custom_raw"
+    msgs = {
+        "R9901": (
+            "Replace re.compile() with Re() using `from re import compile as Re`",
+            "re-compile-alias",
+            (
+                "Use a Re() alias instead of re.compile() by importing the "
+                "re.compile() function as Re()"
+            ),
+        ),
+    }
+    options = ()
+
+    def process_module(self, node: nodes.Module) -> None:
+        pattern = re.compile(r"re\.compile\(")
+        with node.stream() as stream:
+            for lineno, line in enumerate(stream):
+                if pattern.search(line.decode("utf-8")):
+                    self.add_message("re-compile-alias", line=lineno)
+
+
+def register(linter: PyLinter) -> None:
+    linter.register_checker(ReCompileChecker(linter))