Name	Size	Mode	Actions
__pycache__/	-	0755	rm
auth_user.py	1392	0644	editdlrm
django_installed.py	1319	0644	editdlrm
foreign_key_strings.py	6321	0644	editdlrm
forms.py	1517	0644	editdlrm
json_response.py	2701	0644	editdlrm
migrations.py	6428	0644	editdlrm
models.py	4609	0644	editdlrm
__init__.py	802	0644	editdlrm

"""Models.""" from astroid.nodes import Assign, AssignName, ClassDef from pylint.checkers import BaseChecker from pylint.checkers.utils import check_messages from pylint.interfaces import IAstroidChecker from pylint_django.__pkginfo__ import BASE_ID from pylint_django.utils import node_is_subclass def _get_child_meta(node): for child in node.get_children(): if isinstance(child, ClassDef) and child.name == "Meta": return child return None class FormChecker(BaseChecker): """Django model checker.""" __implements__ = IAstroidChecker name = "django-form-checker" msgs = { f"W{BASE_ID}04": ( "Use explicit fields instead of exclude in ModelForm", "modelform-uses-exclude", "Prevents accidentally allowing users to set fields, especially when adding new fields to a Model", ) } @check_messages("modelform-uses-exclude") def visit_classdef(self, node): """Class visitor.""" if not node_is_subclass(node, "django.forms.models.ModelForm", ".ModelForm"): # we only care about forms return meta = _get_child_meta(node) if not meta: return for child in meta.get_children(): if not isinstance(child, Assign) or not isinstance(child.targets[0], AssignName): continue if child.targets[0].name == "exclude": self.add_message(f"W{BASE_ID}04", node=child) break