final commit

summary.py

@@ -0,0 +1,151 @@
+from pathlib import Path
+from datetime import datetime
+from collections import Counter
+from functools import reduce
+import re
+import string
+
+import sys
+
+def parse_foods_file():
+    path = Path.home() / 'projects' / 'open-journal' / 'foods'
+    text = path.read_text()
+    foods, recipes = text.split('---')
+
+    def parse_macro(macro):
+        if macro == '...':
+            return ('INVALID', 0.0)
+
+        name, value = macro.split()
+        value = float(value.removesuffix('g').removesuffix('kcal'))
+        return (name, value)
+
+    foods = {
+        macros[0]: dict(parse_macro(macro) for macro in macros[1:])
+        for macros in [food.split('\n') for food in foods.strip().split('\n\n')]
+    }
+
+    def combine_values(fst, snd):
+        result = fst.copy()
+        for k,v in snd.items():
+            if k in fst:
+                result[k] += v
+            else:
+                result[k] = v
+        return result
+
+    def evaluate_ingredients(ingredients):
+        result = {}
+
+        total_weight = 0.0
+        for ingredient in ingredients:
+            k,v = parse_macro(ingredient)
+            if k == 'TOTAL':
+                result[k] = v
+                break
+            else:
+                total_weight += v
+
+
+            food = foods[k]
+
+            for kk,vv in food.items():
+                if kk not in result:
+                    result[kk] = 0.0
+
+                result[kk] += vv * (v/100.0)
+
+        if 'TOTAL' not in result:
+            result['TOTAL'] = total_weight
+
+        return result
+
+    recipes = {
+        ingredients[0]: evaluate_ingredients(ingredients[1:])
+        for ingredients in [
+            recipe.split('\n') for recipe in recipes.strip().split('\n\n')
+        ]
+    }
+
+    def get_calories_from_macros(mm):
+        calories = 0.0
+        for k,v in mm.items():
+            calories += v * {
+                'Carbs': 4,
+                'Fat': 9,
+                'Protein': 4
+            }.get(k, 0.0)
+        return calories
+
+    #for k,v in foods.items():
+    #    print(round(v.get('Energy') - get_calories_from_macros(v)), k)
+
+    return foods, recipes
+
+foods, recipes = parse_foods_file()
+
+entry_re = re.compile(r'^(\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}) ', re.MULTILINE)
+diet_re = re.compile(r'@diet (\d+g) ([a-zA-Z]+)')
+
+current_day = list(sorted((Path.home() / 'workspace' /
+    'journal').glob('*.md')))[-1]
+
+header, *tmp = entry_re.split(current_day.read_text())
+entries = list(zip(tmp[::2], tmp[1::2]))
+
+daily_grams = 0.0
+daily_calories = 0.0
+daily_protein = 0.0
+
+for (timestamp, content) in sorted(entries, key=lambda x: x[0]):
+    content = '\n'.join(
+        part.replace('\n', ' ')
+        for part in content.split('\n\n')
+    )
+
+    has_printed = False
+    entry_calories = 0.0
+    entry_protein = 0.0
+    for diet in diet_re.finditer(content):
+        if not has_printed:
+            print(f'-- {timestamp}')
+            has_printed = True
+            
+        value, name = diet.groups()
+        value = float(value.removesuffix('g'))
+
+        if name in recipes:
+            food = recipes[name]
+
+            if value == 0.0:
+                value = food['TOTAL']
+
+            food = {k: v*(value/food['TOTAL']) for k,v in food.items()}
+        elif name in foods:
+            if value == 0.0:
+                value = 100
+
+            food = {k: v*(value/100.0) for k,v in foods[name].items()}
+        else:
+            breakpoint()
+            print(f'ERROR: Invalid diet entry: {content}')
+            continue
+
+        protein = round(food.get('Protein', 0.0), 2)
+        calories = round(food.get('Energy', 0.0), 2)
+
+        entry_calories += calories
+        entry_protein += protein
+
+        print(f'{name:<20} {value:<6}g, {calories:<6}kcal, {protein:<6}g protein')
+
+    if has_printed:
+        entry_calories = round(entry_calories, 2)
+        entry_protein = round(entry_protein, 2)
+        print(f'-- TOTAL: {entry_calories}kcal, {entry_protein}g protein')
+        print()
+
+    daily_calories += entry_calories
+    daily_protein += entry_protein
+
+print(f'-- DAILY TOTAL ({daily_calories}kcal, {daily_protein}g protein)')