new file mode 100755

@@ -0,0 +1,52 @@

+#!/usr/bin/env python

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

+

+import sys

+import json

+import time

+import socket

+

+

+def main():

+    accounts = get_accounts()

+    timestamp = time.time()

+    socket = graphite_connect()

+    for account in accounts.get('data', []):

+        send_account(socket, account, timestamp)

+    socket.close()

+

+def graphite_connect():

+    if len(sys.argv) == 4:

+        socket_args = (sys.argv[2], int(sys.argv[3]))

+    elif len(sys.argv) == 3:

+        socket_args = (sys.argv[2], 2003)

+    else:

+        socket_args = ('::1', 2003)

+    graphite_socket = socket.socket()

+    graphite_socket.connect(socket_args)

+    return graphite_socket

+

+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_path(account):

+    (id, bank) = account['id'].split('@')

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

+    return 'bank.%s.%s-balance-%s' % (bank, id, currency)

+

+def send_account(socket, account, timestamp):

+    pattern = '%s %.2f %d\n'

+    values = (get_path(account), account['balance'], int(timestamp))

+    line = pattern % values

+    socket.send(line)

+

+if __name__ == '__main__':

+    main()