Make session saving and lookup more reusable.

[fulbank.git] / netbank

#!/usr/bin/python3

import sys, os, getopt, pwd, operator
from fulbank import auth, data

sessname = data.defaultsess()
sess = None

def pfxmatch(pfx, item):
    return str(item)[:len(pfx)] == pfx

class ambiguous(LookupError):
    def __init__(self, a, b):
        super().__init__("ambigous match: %s and %s" % (a, b))
        self.a = a
        self.b = b

def find(seq, *, item=None, test=None, match=None, key=None, default=LookupError):
    if key is None:
        key = lambda o: o
    if match is None and item is not None:
        match = lambda o: test(item, o)
    if test is None:
        test = operator.eq
    found = None
    for thing in seq:
        if match(key(thing)):
            if found is None:
                found = thing
            else:
                if default is LookupError:
                    raise ambiguous(key(found), key(thing))
                else:
                    return default
    if found is not None:
        return found
    if default is LookupError:
        raise LookupError()
    else:
        return default

def usage(out):
    out.write("usage: netbank [-h] [-s SESSION-ID] COMMAND [ARGS...]\n")

def requiresess(fn):
    def wrap(cmd, args):
        if sess is None:
            sys.stderr.write("netbank: %s: no current session\n" % (cmd))
            sys.exit(1)
        return fn(cmd, args)
    return wrap

commands = {}

def cmd_login(cmd, args):
    global sess
    opts, args = getopt.getopt(args, "t:")
    typename = sessname
    for o, a in opts:
        if o == "-t":
            typename = a
    if len(args) < 1:
        sys.stderr.write("usage: login [-t BANK-ID] TYPE\n")
        sys.exit(1)
    sess = data.getsessnam(typename).create()
    if args[0] == "bankid":
        authfun = sess.auth_bankid
    elif args[0] == "token":
        authfun = sess.auth_token
    else:
        sys.stderr.write("netbank: %s: unknown authentication type\n" % (args[0]))
        sys.exit(1)
    if len(args) < 2:
        sys.stderr.write("usage: login bankid USER-ID\n")
        sys.exit(1)
    with auth.ttyconv() as conv:
        try:
            authfun(args[1], conv)
        except auth.autherror as err:
            sys.stderr.write("netbank: authentication failed: %s\n" % err)
            sys.exit(1)
commands["login"] = cmd_login

@requiresess
def cmd_logout(cmd, args):
    global sess
    if sess is not None:
        sess.close()
        sess = None
commands["logout"] = cmd_logout

@requiresess
def cmd_ping(cmd, args):
    sess.keepalive()
commands["ping"] = cmd_ping

@requiresess
def cmd_lsacct(cmd, args):
    for acct in sess.accounts:
        sys.stdout.write("%s (%s): %s\n" % (acct.number, acct.name, acct.balance))
commands["lsacct"] = cmd_lsacct

@requiresess
def cmd_lstxn(cmd, args):
    opts, args = getopt.getopt(args, "n:")
    num = 10
    for o, a in opts:
        if o == "-n":
            num = int(a)
    if len(args) < 1:
        sys.stderr.write("usage: lstxn [-n NUM] ACCOUNT\n")
        sys.exit(1)
    try:
        acct = find(sess.accounts, item=args[0], key=lambda acct: acct.number, test=pfxmatch)
    except ambiguous as exc:
        sys.stderr.write("netbank: %s: ambiguous match between %s and %s\n" % (args[0], exc.a, exc.b))
        sys.exit(1)
    except LookupError:
        sys.stderr.write("netbank: %s: no such account\n" % (args[0]))
        sys.exit(1)
    for i, txn in zip(range(num), acct.transactions()):
        sys.stdout.write("%s %s: %s\n" % (txn.date.isoformat(), txn.value, txn.message))
commands["lstxn"] = cmd_lstxn

def main():
    global sess, sessname

    opts, args = getopt.getopt(sys.argv[1:], "hs:")
    for o, a in opts:
        if o == "-h":
            usage(sys.stdout)
            sys.exit(0)
        if o == "-s":
            sessname = a
    if len(args) < 1:
        usage(sys.stderr)
        sys.exit(1)

    cmd = args[0]
    args = args[1:]

    sess = data.loadsess(sessname, None)
    if cmd in commands:
        commands[cmd](cmd, args)
    else:
        sys.stderr.write("netbank: %s: unknown command\n" % (cmd))
        sys.exit(1)
    data.savesess(sess, sessname)

try:
    if __name__ == "__main__":
        main()
except KeyboardInterrupt:
    sys.exit(1)