add transfers watching jabber bot

1 files changed, 91 insertions, 0 deletions

diff --git a/transfers.py b/transfers.py
new file mode 100755
index 0000000..c14055f
--- /dev/null
+++ b/transfers.py
@@ -0,0 +1,91 @@
+#!/usr/bin/python3
+import configparser
+import argparse
+import datetime
+import logging
+
+from fints.client import FinTS3PinTanClient
+import peewee as p
+from slixmpp import ClientXMPP
+
+logging.basicConfig(level=logging.INFO)
+
+parser = argparse.ArgumentParser()
+parser.add_argument('--config', nargs=1, required=True, help='Account .ini config file')
+parser.add_argument('--db', nargs=1, default='log.db', help='Database')
+parser.add_argument('--fetch-last', metavar='DAYS', default=None, type=int, help='Fetch all last DAYS')
+args = parser.parse_args()
+
+config = configparser.ConfigParser()
+config.read(args.config)
+db = p.SqliteDatabase(args.db)
+
+
+class Transaction(p.Model):
+    created_date = p.DateField(default=datetime.date.today)
+    accountnumber = p.CharField()
+    sepa_entry_date = p.DateField()
+    sepa_amount_str = p.CharField()
+    sepa_transaction_details = p.CharField()
+
+    class Meta:
+        database = db
+        indexes = ((('accountnumber', 'sepa_entry_date', 'sepa_transaction_details'), True),)
+
+
+db.create_tables([Transaction], True)
+
+if args.fetch_last is None:
+    # Look for last date of a saved transaction or fallback default timesamp
+    last_transaction_date = Transaction \
+                                .select(p.fn.max(Transaction.sepa_entry_date)) \
+                                .where(Transaction.accountnumber == config['bank']['accountnumber']) \
+                                .scalar(convert=True) or datetime.date.today() - datetime.timedelta(30)
+else:
+    last_transaction_date = datetime.date.today() - datetime.timedelta(args.fetch_last)
+
+f = FinTS3PinTanClient(config['bank']['blz'], config['bank']['username'], config['bank']['password'],
+                       config['bank']['url'])
+accounts = f.get_sepa_accounts()
+account = list(filter(lambda a: a.accountnumber == config['bank']['accountnumber'], accounts))[0]
+
+logging.info("select transactions on account %s starting from %s", account.accountnumber, last_transaction_date)
+transactions = f.get_statement(account, last_transaction_date, datetime.date.today())
+
+new_transactions = []
+for t in transactions:
+    y, m, d, *_ = t.data['entry_date'].timetuple()
+    entry_date = datetime.date(y, m, d)
+    filter_pred = {
+        'accountnumber': config['bank']['accountnumber'],
+        'sepa_entry_date': entry_date,
+        'sepa_transaction_details': t.data['transaction_details'],
+    }
+    data = {'sepa_amount_str': str(t.data['amount'])}
+    try:
+        record = Transaction.get(**filter_pred)
+        record.update(data)
+        logging.info("Updated %s", record.id)
+    except Transaction.DoesNotExist:
+        data.update(filter_pred)
+        record = Transaction.create(**data)
+        new_transactions += [record]
+        logging.info("Created %s", record.id)
+    record.save()
+
+cl = ClientXMPP(config['jabber']['jid'], config['jabber']['password'])
+
+
+def session_start(event):
+    cl.send_presence()
+    for t in new_transactions:
+        message = "{} {} {}".format(t.accountnumber, t.sepa_amount_str, t.sepa_transaction_details)
+        cl.send_message(config['jabber']['recipient'], message, mtype='chat')
+        logging.info("Sent info %s", message)
+    cl.disconnect()
+
+
+cl.add_event_handler('session_start', session_start)
+cl.add_event_handler('killed', lambda *_: cl.loop.stop())
+cl.connect()
+cl.process()