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

Fix WeekDaysType support for py33

Browse Source
This commit is contained in:
Konsta Vesterinen
2014-04-21 11:11:18 +03:00
parent 18a40f68f3
commit 02c60f1d28
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sqlalchemy_utils/types/bit.py
View File

@@ -16,5 +16,7 @@ class BitType(sa.types.TypeDecorator):
# Use the native BIT type for drivers that has it. # Use the native BIT type for drivers that has it.
if dialect.name == 'postgresql': if dialect.name == 'postgresql':
return dialect.type_descriptor(BIT(self.length)) return dialect.type_descriptor(BIT(self.length))
elif dialect.name == 'sqlite':
return dialect.type_descriptor(sa.String(self.length))
else: else:
return dialect.type_descriptor(type(self.impl)(self.length)) return dialect.type_descriptor(type(self.impl)(self.length))

4
sqlalchemy_utils/types/weekdays.py
View File

@@ -48,6 +48,10 @@ class WeekDaysType(types.TypeDecorator, ScalarCoercible):
impl = BitType(WeekDay.NUM_WEEK_DAYS) impl = BitType(WeekDay.NUM_WEEK_DAYS)
@property
def comparator_factory(self):
return self.impl.comparator_factory
def process_bind_param(self, value, dialect): def process_bind_param(self, value, dialect):
if isinstance(value, WeekDays): if isinstance(value, WeekDays):
return value.as_bit_string() return value.as_bit_string()