37 lines
992 B
Python
37 lines
992 B
Python
import decimal
|
|
from .._gae import ndb
|
|
|
|
|
|
#TODO Needs more testing
|
|
class NDBDecimalProperty(ndb.StringProperty):
|
|
"""
|
|
NDB decimal implementation
|
|
"""
|
|
data_type = decimal.Decimal
|
|
|
|
def __init__(self, precision, scale, **kwargs):
|
|
d = '1.'
|
|
for x in range(scale):
|
|
d += '0'
|
|
self.round = decimal.Decimal(d)
|
|
|
|
def _to_base_type(self, value):
|
|
if value is None or value == '':
|
|
return None
|
|
else:
|
|
return str(value)
|
|
|
|
def _from_base_type(self, value):
|
|
if value is None or value == '':
|
|
return None
|
|
else:
|
|
return decimal.Decimal(value).quantize(self.round)
|
|
|
|
def _validate(self, value):
|
|
if value is None or isinstance(value, decimal.Decimal):
|
|
return value
|
|
elif isinstance(value, basestring):
|
|
return decimal.Decimal(value)
|
|
raise TypeError("Property %s must be a Decimal or string."
|
|
% self._name)
|