Recipient inherits from Currency

This commit is contained in:
Romain Bignon 2013-08-02 12:52:01 +02:00
commit e2a67a954a

View file

@ -43,18 +43,23 @@ class TransferError(UserError):
""" """
class Recipient(CapBaseObject): class Recipient(CapBaseObject, Currency):
""" """
Recipient of a transfer. Recipient of a transfer.
""" """
label = StringField('Name') label = StringField('Name')
currency = IntField('Currency', default=Currency.CUR_UNKNOWN)
def __init__(self): def __init__(self):
CapBaseObject.__init__(self, 0) CapBaseObject.__init__(self, 0)
@property
def currency_text(self):
return Currency.currency2txt(self.currency)
class Account(Recipient, Currency):
class Account(Recipient):
""" """
Bank account. Bank account.
@ -72,15 +77,10 @@ class Account(Recipient, Currency):
type = IntField('Type of account', default=TYPE_UNKNOWN) type = IntField('Type of account', default=TYPE_UNKNOWN)
balance = DecimalField('Balance on this bank account') balance = DecimalField('Balance on this bank account')
coming = DecimalField('Coming balance') coming = DecimalField('Coming balance')
currency = IntField('Currency', default=Currency.CUR_UNKNOWN)
def __repr__(self): def __repr__(self):
return u"<Account id=%r label=%r>" % (self.id, self.label) return u"<Account id=%r label=%r>" % (self.id, self.label)
@property
def currency_text(self):
return Currency.currency2txt(self.currency)
class Transaction(CapBaseObject): class Transaction(CapBaseObject):
""" """