Update model_field_checks.py

kalekseev · Dec 9, 2024 · 91e69b3 · 91e69b3
1 parent a4b241c
commit 91e69b3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/extra_checks/checks/model_field_checks.py b/src/extra_checks/checks/model_field_checks.py
@@ -306,8 +306,9 @@ def apply(
                         ):
                             return
             check = f'models.Q({in_name}=[{", ".join([self._repr_choice(c) for c in field_choices])}])'
+            arg_name = "condition" if django.VERSION >= (5, 1) else "check"
             yield self.message(
                 "Field with choices must have companion CheckConstraint to enforce choices on database level.",
-                hint=f'Add to Meta.constraints: `models.CheckConstraint(name="%(app_label)s_%(class)s_{field.name}_valid", condition={check})`',
+                hint=f'Add to Meta.constraints: `models.CheckConstraint(name="%(app_label)s_%(class)s_{field.name}_valid", {arg_name}={check})`',
                 obj=field,
             )