3 import sys, os, getopt, pwd, operator
4 from fulbank import auth, data
6 sessname = data.defaultsess()
9 def pfxmatch(pfx, item):
10 return str(item)[:len(pfx)] == pfx
12 class ambiguous(LookupError):
13 def __init__(self, a, b):
14 super().__init__("ambigous match: %s and %s" % (a, b))
18 def find(seq, *, item=None, test=None, match=None, key=None, default=LookupError):
21 if match is None and item is not None:
22 match = lambda o: test(item, o)
31 if default is LookupError:
32 raise ambiguous(key(found), key(thing))
37 if default is LookupError:
43 out.write("usage: netbank [-h] [-s SESSION-ID] COMMAND [ARGS...]\n")
48 sys.stderr.write("netbank: %s: no current session\n" % (cmd))
55 def cmd_login(cmd, args):
57 opts, args = getopt.getopt(args, "t:")
63 sys.stderr.write("usage: login [-t BANK-ID] TYPE\n")
65 sess = data.getsessnam(typename).create()
66 if args[0] == "bankid":
67 authfun = sess.auth_bankid
68 elif args[0] == "token":
69 authfun = sess.auth_token
71 sys.stderr.write("netbank: %s: unknown authentication type\n" % (args[0]))
74 sys.stderr.write("usage: login bankid USER-ID\n")
76 with auth.ttyconv() as conv:
78 authfun(args[1], conv)
79 except auth.autherror as err:
80 sys.stderr.write("netbank: authentication failed: %s\n" % err)
82 commands["login"] = cmd_login
85 def cmd_logout(cmd, args):
90 commands["logout"] = cmd_logout
93 def cmd_ping(cmd, args):
95 commands["ping"] = cmd_ping
98 def cmd_lsacct(cmd, args):
99 for acct in sess.accounts:
100 sys.stdout.write("%s (%s): %s\n" % (acct.number, acct.name, acct.balance))
101 commands["lsacct"] = cmd_lsacct
104 def cmd_lstxn(cmd, args):
105 opts, args = getopt.getopt(args, "n:")
111 sys.stderr.write("usage: lstxn [-n NUM] ACCOUNT\n")
114 acct = find(sess.accounts, item=args[0], key=lambda acct: acct.number, test=pfxmatch)
115 except ambiguous as exc:
116 sys.stderr.write("netbank: %s: ambiguous match between %s and %s\n" % (args[0], exc.a, exc.b))
119 sys.stderr.write("netbank: %s: no such account\n" % (args[0]))
121 for i, txn in zip(range(num), acct.transactions()):
122 sys.stdout.write("%s %s: %s\n" % (txn.date.isoformat(), txn.value, txn.message))
123 commands["lstxn"] = cmd_lstxn
126 global sess, sessname
128 opts, args = getopt.getopt(sys.argv[1:], "hs:")
142 sess = data.loadsess(sessname, None)
144 commands[cmd](cmd, args)
146 sys.stderr.write("netbank: %s: unknown command\n" % (cmd))
148 data.savesess(sess, sessname)
151 if __name__ == "__main__":
153 except KeyboardInterrupt: