Add `--color-palette` option to `pyreverse` (#8223)

3 files changed, 55 insertions, 0 deletions

diff --git a/tests/pyreverse/functional/class_diagrams/colorized_output/custom_colors.puml b/tests/pyreverse/functional/class_diagrams/colorized_output/custom_colors.puml
new file mode 100644
index 000000000..34dc814ff
--- /dev/null
+++ b/tests/pyreverse/functional/class_diagrams/colorized_output/custom_colors.puml
@@ -0,0 +1,42 @@
+@startuml classes
+set namespaceSeparator none
+class "CheckerCollector" as custom_colors.CheckerCollector #red {
+  checker1
+  checker2
+  checker3
+}
+class "ElseifUsedChecker" as pylint.extensions.check_elif.ElseifUsedChecker #green {
+  msgs : dict
+  name : str
+  leave_module(_: nodes.Module) -> None
+  process_tokens(tokens: list[TokenInfo]) -> None
+  visit_if(node: nodes.If) -> None
+}
+class "ExceptionsChecker" as pylint.checkers.exceptions.ExceptionsChecker #yellow {
+  msgs : dict
+  name : str
+  options : tuple
+  open() -> None
+  visit_binop(node: nodes.BinOp) -> None
+  visit_compare(node: nodes.Compare) -> None
+  visit_raise(node: nodes.Raise) -> None
+  visit_tryexcept(node: nodes.TryExcept) -> None
+}
+class "StdlibChecker" as pylint.checkers.stdlib.StdlibChecker #yellow {
+  msgs : dict[str, MessageDefinitionTuple]
+  name : str
+  deprecated_arguments(method: str) -> tuple[tuple[int | None, str], ...]
+  deprecated_classes(module: str) -> Iterable[str]
+  deprecated_decorators() -> Iterable[str]
+  deprecated_methods() -> set[str]
+  visit_boolop(node: nodes.BoolOp) -> None
+  visit_call(node: nodes.Call) -> None
+  visit_functiondef(node: nodes.FunctionDef) -> None
+  visit_if(node: nodes.If) -> None
+  visit_ifexp(node: nodes.IfExp) -> None
+  visit_unaryop(node: nodes.UnaryOp) -> None
+}
+pylint.checkers.exceptions.ExceptionsChecker --* custom_colors.CheckerCollector : checker1
+pylint.checkers.stdlib.StdlibChecker --* custom_colors.CheckerCollector : checker3
+pylint.extensions.check_elif.ElseifUsedChecker --* custom_colors.CheckerCollector : checker2
+@enduml
diff --git a/tests/pyreverse/functional/class_diagrams/colorized_output/custom_colors.py b/tests/pyreverse/functional/class_diagrams/colorized_output/custom_colors.py
new file mode 100644
index 000000000..b46c97e75
--- /dev/null
+++ b/tests/pyreverse/functional/class_diagrams/colorized_output/custom_colors.py
@@ -0,0 +1,10 @@
+from pylint.checkers.exceptions import ExceptionsChecker
+from pylint.checkers.stdlib import StdlibChecker
+from pylint.extensions.check_elif import ElseifUsedChecker
+
+
+class CheckerCollector:
+    def __init__(self):
+        self.checker1 = ExceptionsChecker(None)
+        self.checker2 = ElseifUsedChecker(None)
+        self.checker3 = StdlibChecker(None)
diff --git a/tests/pyreverse/functional/class_diagrams/colorized_output/custom_colors.rc b/tests/pyreverse/functional/class_diagrams/colorized_output/custom_colors.rc
new file mode 100644
index 000000000..ea30ad4a4
--- /dev/null
+++ b/tests/pyreverse/functional/class_diagrams/colorized_output/custom_colors.rc
@@ -0,0 +1,3 @@
+[testoptions]
+output_formats=puml
+command_line_args=-S --colorized --max-color-depth=2 --color-palette=red,green,yellow