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

Handle 'sa.Boolean' for 'EncryptedType'

Browse Source
This commit is contained in:
Ryan Leckey
2014-10-30 13:57:27 -07:00
parent 90efc25228
commit f1d6ba4ecc
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
sqlalchemy_utils/types/encrypted.py
View File

@@ -223,8 +223,12 @@ class EncryptedType(TypeDecorator, ScalarCoercible):
except AttributeError: except AttributeError:
# Doesn't have 'process_bind_param' # Doesn't have 'process_bind_param'
pass
# Handle 'boolean'
if issubclass(self.underlying_type.python_type, bool):
value = "true" if value else "false"
print("encrypt: ", value)
return self.engine.encrypt(value) return self.engine.encrypt(value)
def process_result_value(self, value, dialect): def process_result_value(self, value, dialect):
@@ -239,6 +243,12 @@ class EncryptedType(TypeDecorator, ScalarCoercible):
except AttributeError: except AttributeError:
# Doesn't have 'process_result_value' # Doesn't have 'process_result_value'
# Handle 'boolean'
if issubclass(self.underlying_type.python_type, bool):
return decrypted_value == "true"
# Handle all others
return self.underlying_type.python_type(decrypted_value) return self.underlying_type.python_type(decrypted_value)
def _coerce(self, value): def _coerce(self, value):