From 08bf5660bf2818d86e1af0acda52709464469c7c Mon Sep 17 00:00:00 2001 From: Aleksander Vognild Burkow Date: Mon, 4 Feb 2019 16:48:47 +0100 Subject: [PATCH] Add additional django field __get__ types These are taken from the corresponding django/contrib/postgres/fields/ranges.py file. --- .../contrib/postgres/fields/ranges.pyi | 21 ++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) 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: ...