kiwitcms/Kiwi

from pylint import checkers
from pylint_django.utils import node_is_subclass


class FormFieldChecker(checkers.BaseChecker):
    name = "form-field-label-checker"

    msgs = {
        "R4811": (
            "Do not use label in form Field class constructor.",
            "form-field-label-used",
            "Do not use label in form Field class constructor,"
            "place template label in HTML. See:"
            "https://github.com/kiwitcms/Kiwi/issues/738",
        ),
        "R4812": (
            "Do not use help_text in form Field class constructor.",
            "form-field-help-text-used",
            "Do not use help_text in form Field class constructor,"
            "place template help_text in HTML. See:"
            "https://github.com/kiwitcms/Kiwi/issues/738",
        ),
    }

    def visit_call(self, node):
        if node.func.as_string().endswith("Field") and node.keywords:
            for keyword in node.keywords:
                if keyword.arg == "label":
                    self.add_message("form-field-label-used", node=node)
                if keyword.arg == "help_text":
                    self.add_message("form-field-help-text-used", node=node)


class ModelFormChecker(checkers.BaseChecker):
    """
    Check to ensure that all forms inherit from ModelForm.
    """

    name = "model-form-checker"

    msgs = {
        "R4831": (
            "All forms must inherit from ModelForm",
            "must-inherit-from-model-form",
            "All forms must inherit from ModelForm"
            "See: https://github.com/kiwitcms/Kiwi/issues/1384",
        )
    }

    def visit_classdef(self, node):
        if node_is_subclass(
            node, "django.forms.forms.BaseForm", ".BaseForm"
        ) and not node_is_subclass(node, "django.forms.models.ModelForm", ".ModelForm"):
            self.add_message("must-inherit-from-model-form", node=node)