def test_category(self):
category_dict = ItemCategoryDict()
category_dict.item_category_dict = {'Pfand': 'Pfand',
'Pesto': 'Pesto',
'Parmesan': 'Käse',
'Heumilch': 'Milch',
'Milch': 'Milch',
}
cat_milch = category_dict.get_category('Milch')
assert cat_milch == 'Milch'
def test_category(self):
category_dict = ItemCategoryDict()
category_dict.item_category_dict = {
'Pfand': 'Pfand',
'Pesto': 'Pesto',
'Parmesan': 'Käse',
'Heumilch': 'Milch',
'Milch': 'Milch',
}
cat_milch = category_dict.get_category('Milch')
assert cat_milch == 'Milch'
class ReceiptCollection(object):
'''
Collection of purchases with evaluation and output options.
'''
def __init__(self, purchases=[]):
self.categories = defaultdict(lambda: [0.0, set(), 0.0])
self.purchases = purchases
self.unsane_items = []
self.unsane_categories = []
self.category_dict = ItemCategoryDict()
self.total = 0.0
def collect_items(self):
'''
sort all positions in our stored receipts/purchases into their
respective categories
'''
for purchase in self.purchases:
self.unsane_items.extend(purchase.unsane_items)
for item in purchase.positions:
self.categories[item.category][1].add(item.name)
self.categories[item.category][0] += item.price
self.check_sanity()
self.calculate_total()
def totalize_categories(self):
self.initialize_super_categories()
for category in self.categories.keys():
catsum = 0
length = len(category)
for cat in self.categories.keys():
if (cat[0:length] == category):
catsum += self.categories[cat][0]
self.categories[category][2] = catsum
def initialize_super_categories(self):
missing_super_categories = []
for category in self.categories.keys():
if (category[:category.rfind(':')] not in self.categories.keys()):
missing_super_categories.append(category[:category.rfind(':')])
for missing in missing_super_categories:
while True:
if missing not in self.categories:
self.categories[missing][0] = 0
if missing.rfind(':') == -1:
break
missing = missing[:missing.rfind(':')]
def check_category(self, category, item):
'''
make list of conflicting categories
'''
stored_category = self.category_dict.get_category(item)
if category != stored_category:
self.unsane_categories.append((item, category, stored_category))
def check_sanity(self):
'''
make list of items belonging to more than one category
'''
all_items = set()
for category in self.categories:
if category is '':
continue
for item in self.categories[category][1]:
if item in all_items:
self.unsane_items.append(item)
self.check_category(category, item)
all_items.add(item)
def calculate_total(self):
'''
calculate the grand total across all categories
'''
self.total = 0.0
for category in self.categories:
self.total += self.categories[category][0]
def get_ledger(self, date='1900-01-01'):
'''
create output in the format of the ledger application
'''
ledger_output = ""
for receipt in sorted(self.purchases, key=lambda t: t.date):
if receipt.date >= date:
ledger_output += receipt.get_ledger()
return ledger_output
class ReceiptCollection(object):
'''
Collection of purchases with evaluation and output options.
'''
def __init__(self, purchases=[]):
self.categories = defaultdict(lambda: [0.0, set(), 0.0])
self.purchases = purchases
self.unsane_items = []
self.unsane_categories = []
self.category_dict = ItemCategoryDict()
self.total = 0.0
def collect_items(self):
'''
sort all positions in our stored receipts/purchases into their
respective categories
'''
for purchase in self.purchases:
self.unsane_items.extend(purchase.unsane_items)
for item in purchase.positions:
self.categories[item.category][1].add(item.name)
self.categories[item.category][0] += item.price
self.check_sanity()
self.calculate_total()
def totalize_categories(self):
self.initialize_super_categories()
for category in self.categories.keys():
catsum = 0
length = len(category)
for cat in self.categories.keys():
if (cat[0:length] == category):
catsum += self.categories[cat][0]
self.categories[category][2] = catsum
def initialize_super_categories(self):
missing_super_categories = []
for category in self.categories.keys():
if (category[:category.rfind(':')] not in self.categories.keys()):
missing_super_categories.append(category[:category.rfind(':')])
for missing in missing_super_categories:
while True:
if missing not in self.categories:
self.categories[missing][0] = 0
if missing.rfind(':') == -1:
break
missing = missing[:missing.rfind(':')]
def check_category(self, category, item):
'''
make list of conflicting categories
'''
stored_category = self.category_dict.get_category(item)
if category != stored_category:
self.unsane_categories.append((item, category, stored_category))
def check_sanity(self):
'''
make list of items belonging to more than one category
'''
all_items = set()
for category in self.categories:
if category is '':
continue
for item in self.categories[category][1]:
if item in all_items:
self.unsane_items.append(item)
self.check_category(category, item)
all_items.add(item)
def calculate_total(self):
'''
calculate the grand total across all categories
'''
self.total = 0.0
for category in self.categories:
self.total += self.categories[category][0]
def get_ledger(self, date='1900-01-01'):
'''
create output in the format of the ledger application
'''
ledger_output = ""
for receipt in sorted(self.purchases, key=lambda t: t.date):
if receipt.date >= date:
ledger_output += receipt.get_ledger()
return ledger_output