kassomat-count-and-set-coin-levels.py

#!/usr/bin/env python2
import json
import uuid
from redis import StrictRedis


redis = StrictRedis()
pubsub = redis.pubsub()
pubsub.subscribe('hopper-response')
event_pubsub = redis.pubsub()
event_pubsub.subscribe('hopper-event')


def to_int(s):
  try:
    return int(s)
  except ValueError:
    return None

def set_levels(levels):
    """set an array where coin values are the keys and their counts are values
    as new coin leves for the machine"""

    print("Sending the following values to the machine:")
    for coin, count in sorted(levels.items()):
        hopper_request('set-denomination-level',
                       amount=coin, level=count)


def wait_for_message(correlId):
    for msg in pubsub.listen():
        if msg['type'] != 'message':
            continue
        data = json.loads(msg['data'])
        if data['correlId'] == correlId:
 	    return data


def wait_for_event(event):
    for msg in event_pubsub.listen():
        if msg['type'] != 'message':
            continue
        data = json.loads(msg['data'])
        if data['event'] == event:
            return data


def count_coins():
    while True:
        raw_value = raw_input("> to count coins now, please enter 'yes': ")
        if raw_value != 'yes':
    	    break

        smart_empty()

        print("waiting for 'smart emptied' event")  

        smart_emptied = wait_for_event('smart emptied')
        print("amount of money emptied: %d" % (int(smart_emptied['amount'])))

        get_and_set_cashbox_payout_operation_data()


def get_and_set_cashbox_payout_operation_data(): 
    print("Sending cashbox-payout-operation-data to the machine:")

    correlId = str(uuid.uuid4()) 
    redis.publish('hopper-request', json.dumps({
         "cmd": "cashbox-payout-operation-data",
         "msgId": correlId
    }))
    msg = wait_for_message(correlId)
#    status = 'success' if msg['result'] == 'ok' else 'error'
    status = "ok"
 
    levels = {int(level['value']): int(level['level']) for level in msg['levels']}
    print("Quantity of coins emptied:")
    for value, count in sorted(levels.items()):
        print("%3d Eurocent x %3d" % (value, levels[value]))
    
    if 0 in levels and not levels[0] > 0 :
        print("Setting this levels as new value now.")
        set_levels(levels)
    else:
        print("Unknown Coins found, please remove them. Before runnig this again.")


def smart_empty(): 
    print("Sending smart-empty to the machine:")

    correlId = str(uuid.uuid4()) 
    redis.publish('hopper-request', json.dumps({
         "cmd": "smart-empty",
         "msgId": correlId
    }))
    msg = wait_for_message(correlId)
    status = 'success' if msg['result'] == 'ok' else 'error'


if __name__ == '__main__':
    print("Welcome to Count-o-matic!\n")

    count_coins() 
 
    print("Bye.")