Example of Pylint custom checker that verifies tasks definition names

task_name_checker.py

class FlowObjectNameChecker(checkers.BaseChecker):
    __implements__ = interfaces.IAstroidChecker

    name = "invalid-flow-object-definition-name"

    msgs = {
        "C9002": (
            "The task name is not suffixed by `_task`",
            "invalid-task-definition-name",
            "Emitted when a flow function definition is not suffixed by `_task` as per the naming convention requires",
        ),
    }

    def visit_functiondef(self, node):
        if not node.decorators:
            return

        for decorator in node.decorators.get_children():
            if not hasattr(decorator, "func") or not decorator.func:
                return

            if is_prefect_task_decorator(decorator.func):
                if not has_task_valid_suffix(node.name):
                    self.add_message("invalid-task-definition-name", node=node)


def register(linter):
    linter.register_checker(FlowObjectNameChecker(linter))