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

Made NumberRangeType use ScalarCoercible

Browse Source
This commit is contained in:
Konsta Vesterinen
2013-10-24 11:29:04 +03:00
parent 2950abb5f6
commit 7dc71d2719
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
sqlalchemy_utils/types/number_range.py
View File

@@ -1,5 +1,6 @@
import six import six
from sqlalchemy import types from sqlalchemy import types
from .scalar_coercible import ScalarCoercible
class NumberRangeRawType(types.UserDefinedType): class NumberRangeRawType(types.UserDefinedType):
@@ -10,7 +11,7 @@ class NumberRangeRawType(types.UserDefinedType):
return 'int4range' return 'int4range'
class NumberRangeType(types.TypeDecorator): class NumberRangeType(types.TypeDecorator, ScalarCoercible):
impl = NumberRangeRawType impl = NumberRangeRawType
def process_bind_param(self, value, dialect): def process_bind_param(self, value, dialect):
@@ -26,7 +27,7 @@ class NumberRangeType(types.TypeDecorator):
return NumberRange.from_normalized_str(value) return NumberRange.from_normalized_str(value)
return value return value
def coercion_listener(self, target, value, oldvalue, initiator): def _coerce(self, value):
if value is not None and not isinstance(value, NumberRange): if value is not None and not isinstance(value, NumberRange):
if isinstance(value, six.string_types): if isinstance(value, six.string_types):
value = NumberRange.from_normalized_str(value) value = NumberRange.from_normalized_str(value)