bluechips/widgets/__init__.py

   1 from tw import forms
   2
   3 from tw.forms import validators
   4
   5 from bluechips import model
   6 from bluechips.model import meta
   7
   8 from decimal import Decimal
   9
  10 class UserSelect(forms.SingleSelectField):
  11     @staticmethod
  12     def getUserList():
  13         for u in meta.Session.query(model.User):
  14             yield (u.id, u.name)
  15
  16     options = getUserList
  17     validator = validators.Wrapper(
  18         to_python=(lambda x: meta.Session.query(model.User).get(int(x))),
  19         from_python=(lambda x: str(x.id)))
  20
  21     def _is_option_selected(self, option_value, value):
  22         return option_value == value.id
  23
  24 class AmountField(forms.TextField):
  25     size = 8
  26     validator = validators.All(
  27         validators.Wrapper(
  28             to_python=Decimal,
  29             from_python=str),
  30         validators.Regex(r'^[0-9]*(\.[0-9]{2})?$', not_empty=True))
  31
  32 __all__ = ['UserSelect', 'AmountField']