vciecounter.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import re

products = [
	(r'(kratten )?bier', 'bier'),
    (r'flessen rode wijn|rood|rode wijn', 'rood'),
    (r'(flessen witte wijn )?zoet|zoete witte wijn', 'zoet'),
    (r'(flessen witte wijn )?droog|droge witte wijn', 'droog'),
    (r'(flessen )?rosé|rose', 'rose' ),
    (r'(flessen )?cola', 'cola'),
    (r'((flessen )?cola )?light', 'light'),
    (r'(flessen )?fanta|sinas', 'fanta'),
    (r'(flessen )?sprite', 'sprite'),
    (r'(flessen )?spa( rood)?', 'spa'),
    (r'((pakken )?appel)?sap', 'sap'),
    (r'(pakken )?(sinaasappelsap|jus)', 'jus'),
    (r'(flessen )?cassis', 'cassis'),
    (r'(flessen )?ice tea', 'ice tea'),
    (r'pakken ice tea|ijsthee|ice tea pak', 'ijsthee'),
    (r'(zakken borrel)?mix', 'mix'),
    (r'(zakken cocktail)?nootjes|noot', 'noot'),
]

def get_matching_value(dictionary, key):
	for item in dictionary:
		if re.match(r'^%s$' % item[0], key):
			return item[1]

if __name__ == '__main__':
    product_amounts = dict((x, 0) for x in list(zip(*products))[1])
    while 1:
        try:
            input_string = input()
            m = re.match('^(- )?(?P<amount>\d+)\s(?P<product>%s)$' % "|".join(list(zip(*products))[0]), input_string)
            if not m:
                print("Did not understand '%s'." % input_string)
                continue
            product = get_matching_value(products, m.group('product'))
            product_amounts[product] += int(m.group('amount'))
        except EOFError:
            break
    print('I counted:'.upper())
    for product in list(zip(*products))[1]:
        if not product_amounts[product] == 0:
            print("%s: %d" % (product, product_amounts[product]))