netbank

   1 #!/usr/bin/python3
   2
   3 import sys, os, getopt, pwd
   4 from fulbank import auth, data, util
   5
   6 sessname = data.defaultsess()
   7 sess = None
   8
   9 def usage(out):
  10     out.write("usage: netbank [-h] [-s SESSION-ID] COMMAND [ARGS...]\n")
  11
  12 def requiresess(fn):
  13     def wrap(cmd, args):
  14         if sess is None:
  15             sys.stderr.write("netbank: %s: no current session\n" % (cmd))
  16             sys.exit(1)
  17         return fn(cmd, args)
  18     return wrap
  19
  20 commands = {}
  21
  22 def cmd_login(cmd, args):
  23     global sess
  24     opts, args = getopt.getopt(args, "t:")
  25     typename = sessname
  26     for o, a in opts:
  27         if o == "-t":
  28             typename = a
  29     if len(args) < 1:
  30         sys.stderr.write("usage: login [-t BANK-ID] TYPE\n")
  31         sys.exit(1)
  32     sess = data.getsessnam(typename).create()
  33     if args[0] == "bankid":
  34         authfun = sess.auth_bankid
  35     elif args[0] == "token":
  36         authfun = sess.auth_token
  37     else:
  38         sys.stderr.write("netbank: %s: unknown authentication type\n" % (args[0]))
  39         sys.exit(1)
  40     if len(args) < 2:
  41         sys.stderr.write("usage: login bankid USER-ID\n")
  42         sys.exit(1)
  43     with auth.ttyconv() as conv:
  44         try:
  45             authfun(args[1], conv)
  46         except auth.autherror as err:
  47             sys.stderr.write("netbank: authentication failed: %s\n" % err)
  48             sys.exit(1)
  49 commands["login"] = cmd_login
  50
  51 @requiresess
  52 def cmd_logout(cmd, args):
  53     global sess
  54     if sess is not None:
  55         sess.close()
  56         sess = None
  57 commands["logout"] = cmd_logout
  58
  59 @requiresess
  60 def cmd_ping(cmd, args):
  61     sess.keepalive()
  62 commands["ping"] = cmd_ping
  63
  64 @requiresess
  65 def cmd_lsacct(cmd, args):
  66     for acct in sess.accounts:
  67         sys.stdout.write("%s (%s): %s\n" % (acct.number, acct.name, acct.balance))
  68 commands["lsacct"] = cmd_lsacct
  69
  70 @requiresess
  71 def cmd_lstxn(cmd, args):
  72     opts, args = getopt.getopt(args, "n:")
  73     num = 10
  74     for o, a in opts:
  75         if o == "-n":
  76             num = int(a)
  77     if len(args) < 1:
  78         sys.stderr.write("usage: lstxn [-n NUM] ACCOUNT\n")
  79         sys.exit(1)
  80     try:
  81         acct = util.find(sess.accounts, item=args[0], key=lambda acct: acct.number, test=util.pfxmatch)
  82     except util.ambiguous as exc:
  83         sys.stderr.write("netbank: %s: ambiguous match between %s and %s\n" % (args[0], exc.a, exc.b))
  84         sys.exit(1)
  85     except LookupError:
  86         sys.stderr.write("netbank: %s: no such account\n" % (args[0]))
  87         sys.exit(1)
  88     for i, txn in zip(range(num), acct.transactions()):
  89         sys.stdout.write("%s %s: %s\n" % (txn.date.isoformat(), txn.value, txn.message))
  90 commands["lstxn"] = cmd_lstxn
  91
  92 def main():
  93     global sess, sessname
  94
  95     opts, args = getopt.getopt(sys.argv[1:], "hs:")
  96     for o, a in opts:
  97         if o == "-h":
  98             usage(sys.stdout)
  99             sys.exit(0)
 100         if o == "-s":
 101             sessname = a
 102     if len(args) < 1:
 103         usage(sys.stderr)
 104         sys.exit(1)
 105
 106     cmd = args[0]
 107     args = args[1:]
 108
 109     sess = data.loadsess(sessname, None)
 110     if cmd in commands:
 111         commands[cmd](cmd, args)
 112     else:
 113         sys.stderr.write("netbank: %s: unknown command\n" % (cmd))
 114         sys.exit(1)
 115     data.savesess(sess, sessname)
 116
 117 try:
 118     if __name__ == "__main__":
 119         main()
 120 except KeyboardInterrupt:
 121     sys.exit(1)