From 653c66e0aa2c02248377fb9f3f1b95ce0cf138a4 Mon Sep 17 00:00:00 2001 From: Konsta Vesterinen Date: Tue, 14 Jan 2014 12:49:07 +0200 Subject: [PATCH] Add types to ischema_names --- sqlalchemy_utils/types/range.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/sqlalchemy_utils/types/range.py b/sqlalchemy_utils/types/range.py index cfddf56..d299b1c 100644 --- a/sqlalchemy_utils/types/range.py +++ b/sqlalchemy_utils/types/range.py @@ -25,6 +25,7 @@ except ImportError: pass import six import sqlalchemy as sa +from sqlalchemy.dialects.postgresql.base import ischema_names from sqlalchemy import types from ..exceptions import ImproperlyConfigured from .scalar_coercible import ScalarCoercible @@ -63,6 +64,14 @@ class TSTZRANGE(types.UserDefinedType): return 'tstzrange' +ischema_names['int4range'] = INT4RANGE +ischema_names['int8range'] = INT8RANGE +ischema_names['numrange'] = NUMRANGE +ischema_names['daterange'] = DATERANGE +ischema_names['tsrange'] = TSRANGE +ischema_names['tstzrange'] = TSTZRANGE + + class RangeComparator(types.TypeEngine.Comparator): @classmethod def coerce_arg(cls, func):