import decimal
from django.db.models import fields
from django.db.models.fields.subclassing import SubfieldBase
from django.forms import fields_special
from django.utils.currency import Currency
from django.utils.translation import ugettext as _
class CurrencyField(fields.DecimalField):
__metaclass__ = SubfieldBase
def __init__(self, verbose_name=None, name=None, max_digits=None, *args, **kwargs):
if not kwargs.has_key("help_text"):
kwargs['help_text'] = _('Format: ') + Currency(9999.00).format()
decimal_places = 2
if kwargs.has_key('decimal_places'):
del kwargs['decimal_places']
fields.DecimalField.__init__(self, verbose_name, name, max_digits, decimal_places, *args, **kwargs)
def format(self):
return Currency(self).format()
def format_number(self, value):
return Currency(value).format()
def formfield(self, **kwargs):
defaults = {
'form_class': fields_special.CurrencyField,
}
defaults.update(kwargs)
return super(CurrencyField, self).formfield(**defaults)
def to_python(self, value):
if value is None:
return value
try:
return Currency(value)
except decimal.InvalidOperation:
raise decimal.InvalidOperation(
_("This value must be a decimal number."))
def value_to_string(self, obj):
val = self._get_val_from_obj(obj)
if val is None:
data = ''
else:
data = str(val)
return data
Comments