Abstracted some common data-types.

[fulbank.git] / fulbank / data.py

diff --git a/fulbank/data.py b/fulbank/data.py
new file mode 100644 (file)
index 0000000..36efe20
--- /dev/null
+++ b/fulbank/data.py
@@ -0,0 +1,48 @@
+import hashlib
+
+def _localname(type):
+    mod = type.__module__
+    if mod.startswith("fulbank."):
+        mod = mod[8:]
+    return "%s.%s" % (mod, type.__name__)
+
+class account(object):
+    @property
+    def number(self): raise NotImplementedError("account.number")
+    @property
+    def name(self): raise NotImplementedError("account.name")
+    def transactions(self): raise NotImplementedError("account.transactions")
+
+    def __repr__(self):
+        return "#<%s %s: %r>" % (_localname(type(self)), self.number, self.name)
+
+class txnaccount(account):
+    @property
+    def balance(self): raise NotImplementedError("txnaccount.balance")
+    @property
+    def clearing(self): raise NotImplementedError("txnaccount.clearing")
+    @property
+    def fullnumber(self): raise NotImplementedError("txnaccount.fullnumber")
+
+class cardaccount(account):
+    pass
+
+class transaction(object):
+    @property
+    def value(self): raise NotImplementedError("transaction.value")
+    @property
+    def message(self): raise NotImplementedError("transaction.message")
+    @property
+    def date(self): raise NotImplementedError("transaction.date")
+
+    @property
+    def hash(self):
+        dig = hashlib.sha256()
+        dig.update(str(self.date.toordinal()).encode("ascii") + b"\0")
+        dig.update(self.message.encode("utf-8") + b"\0")
+        dig.update(str(self.value.amount).encode("ascii") + b"\0")
+        dig.update(self.value.currency.symbol.encode("ascii") + b"\0")
+        return dig.hexdigest()
+
+    def __repr__(self):
+        return "#<%s %s: %r>" % (_localname(type(self)), self.value, self.message)