ellas: dont crash if name is not uppercase

trnila · Nov 15, 2024 · 53d27f6 · 53d27f6
1 parent 69a54ae
commit 53d27f6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lunches.py b/lunches.py
@@ -290,9 +290,14 @@ def ellas(dom):
         for food in foods[1:]:
             if food.text():
                 parsed = re.match(
-                    r"\s*(?P<num>[0-9]+)\s*\.\s*(?P<name>[A-Z -]+)\s+(?P<ingredients>.*?)\s*(\([0-9 ,]+\))?\s*(?P<price>[0-9]+),-",  # noqa: E501
+                    r"\s*(?P<num>[0-9]+)\s*\.\s*(?P<name>.*?)\s*(\([0-9 ,]+\))?\s*(?P<price>[0-9]+),-",  # noqa: E501
                     food.text(),
                 ).groupdict()
+
+                m = re.match(r"^(?P<name>[A-Z -]+)\s+(?P<ingredients>.*?)$", parsed["name"])
+                if m:
+                    parsed.update(m.groupdict())
+
                 yield Lunch(**parsed)