emphasize transactions that the logged-in user is part of, deemphasize others

[bluechips.git] / bluechips / model / transfer.py

diff --git a/bluechips/model/transfer.py b/bluechips/model/transfer.py
index 21b2445356908c7f6604ec4be2f82e7c967b17c9..512e0f37937d472bd38f3083018da36baab51721 100644 (file)
--- a/bluechips/model/transfer.py
+++ b/bluechips/model/transfer.py
@@ -1,7 +1,17 @@
+from types import Currency
+
 class Transfer(object):
+    def __init__(self, debtor=None, creditor=None, amount=Currency(0)):
+        self.debtor = debtor
+        self.creditor = creditor
+        self.amount = amount
+    
     def __repr__(self):
         return '<Transfer: from %s to %s for %s>' % (self.debtor,
                                                      self.creditor,
                                                      self.amount)
 
+    def involves(self, user):
+        return (self.debtor == user) or (self.creditor == user)
+
 __all__ = ['Transfer']