From 927579f661997b74410cf32977ae641e3f74399f Mon Sep 17 00:00:00 2001 From: Adam Johnson Date: Wed, 31 Aug 2022 15:03:29 +0100 Subject: [PATCH] Update db function __init__'s to pass output_field --- src/django_mysql/models/functions.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/django_mysql/models/functions.py b/src/django_mysql/models/functions.py index 23db10c8..bc044f29 100644 --- a/src/django_mysql/models/functions.py +++ b/src/django_mysql/models/functions.py @@ -23,8 +23,7 @@ class SingleArgFunc(Func): output_field_class: type[DjangoField] def __init__(self, expression: ExpressionArgument) -> None: - super().__init__(expression) - self.output_field = self.output_field_class() + super().__init__(expression, output_field=self.output_field_class()) # Control Flow Functions @@ -223,11 +222,9 @@ class LastInsertId(Func): def __init__(self, expression: ExpressionArgument | None = None) -> None: if expression is not None: - super().__init__(expression) + super().__init__(expression, output_field=IntegerField()) else: - super().__init__() - - self.output_field = IntegerField() + super().__init__(output_field=IntegerField()) @classmethod def get(cls, using: str = DEFAULT_DB_ALIAS) -> int: