new file mode 100755

@@ -0,0 +1,73 @@

+#!/usr/bin/env python

+# -*- coding: utf-8 -*-

+

+import sys

+import json

+from prettytable import PrettyTable

+from termcolor import colored

+

+UNUSUAL_BALANCE_THRESHOLD  = 500.0

+WORRYING_BALANCE_THRESHOLD = 100.0

+HOLYSHIT_BALANCE_THRESHOLD = 0

+

+CURRENCIES = {

+    'EUR': '€',

+    'USD': '$',

+    'GBP': '£',

+    'YEN': '¥'

+}

+

+def main():

+    accounts = get_accounts()

+    table = get_table()

+    total = 0

+    for account in accounts:

+        add_account(table, account)

+        total += account.get('balance')

+    print(table)

+    print (' ' * 60) + colored('Total: ', 'white', attrs=['bold']) + format_balance({'balance': total})

+

+def get_accounts():

+    try:

+        accounts_json_path = sys.argv[1]

+    except IndexError:

+        accounts_json_path = '-'

+    if accounts_json_path == '-':

+        return json.load(sys.stdin)

+    else:

+        with open(accounts_json_path, 'r') as filedesc:

+            return json.load(filedesc)

+

+def get_table():

+    pt = PrettyTable()

+    pt.field_names = ['Account ID', 'Account name', 'Balance']

+    pt.field_names = [colored(x, 'white', attrs=['bold']) for x in pt.field_names]

+    pt.align = 'r'

+    pt.align[pt.field_names[1]] = 'l'

+    return pt

+

+def add_account(table, account):

+    id = colored(account.get('id'), 'cyan')

+    label = colored(account.get('label'), 'cyan')

+    balance = format_balance(account)

+    table.add_row([id, label, balance])

+

+def format_balance(account):

+    balance = account.get('balance')

+    currency = account.get('currency', 'EUR')

+    balance_format = '%12.2f '

+    balance_format += CURRENCIES.get(currency, currency)

+

+    balance_color = 'green'

+    balance_attr = []

+    if balance < HOLYSHIT_BALANCE_THRESHOLD:

+        balance_color = 'red'

+        balance_attr = ['blink']

+    elif balance < WORRYING_BALANCE_THRESHOLD:

+        balance_color = 'red'

+    elif balance < UNUSUAL_BALANCE_THRESHOLD:

+        balance_color = 'yellow'

+    return colored(balance_format % balance, balance_color, attrs=balance_attr)

+

+if __name__ == '__main__':

+    main()