diff --git a/django-stubs/contrib/postgres/fields/ranges.pyi b/django-stubs/contrib/postgres/fields/ranges.pyi index caf923b..b4af716 100644 --- a/django-stubs/contrib/postgres/fields/ranges.pyi +++ b/django-stubs/contrib/postgres/fields/ranges.pyi @@ -2,6 +2,8 @@ from typing import Any from django.db import models +from psycopg2.extras import DateRange, DateTimeTZRange, NumericRange + class RangeField(models.Field): empty_strings_allowed: bool = ... base_field: Any = ... @@ -10,8 +12,17 @@ class RangeField(models.Field): def to_python(self, value: Any): ... def value_to_string(self, obj: Any): ... -class IntegerRangeField(RangeField): ... -class BigIntegerRangeField(RangeField): ... -class FloatRangeField(RangeField): ... -class DateTimeRangeField(RangeField): ... -class DateRangeField(RangeField): ... +class IntegerRangeField(RangeField): + def __get__(self, instance, owner) -> NumericRange: ... + +class BigIntegerRangeField(RangeField): + def __get__(self, instance, owner) -> NumericRange: ... + +class FloatRangeField(RangeField): + def __get__(self, instance, owner) -> NumericRange: ... + +class DateTimeRangeField(RangeField): + def __get__(self, instance, owner) -> DateTimeTZRange: ... + +class DateRangeField(RangeField): + def __get__(self, instance, owner) -> DateRange: ...