openstack/deb-python-sqlalchemy-utils
Code Issues Proposed changes

Add types to ischema_names

Browse Source
This commit is contained in:
Konsta Vesterinen
2014-01-14 12:49:07 +02:00
parent e3f56d1841
commit 653c66e0aa
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
sqlalchemy_utils/types/range.py
View File

@@ -25,6 +25,7 @@ except ImportError:
pass pass
import six import six
import sqlalchemy as sa import sqlalchemy as sa
from sqlalchemy.dialects.postgresql.base import ischema_names
from sqlalchemy import types from sqlalchemy import types
from ..exceptions import ImproperlyConfigured from ..exceptions import ImproperlyConfigured
from .scalar_coercible import ScalarCoercible from .scalar_coercible import ScalarCoercible
@@ -63,6 +64,14 @@ class TSTZRANGE(types.UserDefinedType):
return 'tstzrange' 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): class RangeComparator(types.TypeEngine.Comparator):
@classmethod @classmethod
def coerce_arg(cls, func): def coerce_arg(cls, func):