allow using a single string as an ingredient instead of a dict

This commit is contained in:
Emi Vasilek 2023-11-28 12:28:52 +00:00
parent 072aa03a53
commit 69e578dba4
5 changed files with 59 additions and 17 deletions

View file

@ -3,4 +3,4 @@ disable=
missing-module-docstring,
missing-class-docstring,
missing-function-docstring,
too-few-public-methods
too-few-public-methods

View file

@ -139,7 +139,8 @@ class Builder:
def collect_unitnames(rec: parsing.Recipe) -> List[str]:
results: List[str] = []
for ing in rec.ingredients:
results.append(ing.unit.name)
if ing.unit is not None:
results.append(ing.unit.name)
return results
unitnamelists = self.foreach_recipe(collect_unitnames)

View file

@ -1,5 +1,6 @@
from abc import abstractmethod
import collections
import re
from typing import Any, Dict, List, Optional, Self
import comfyrecipes.settings as settings
@ -178,8 +179,8 @@ class IngredientInstance(Element):
self,
ctx: Context,
name: str,
amount: float,
unit: Unit,
amount: Optional[float],
unit: Optional[Unit],
alternatives: List["IngredientInstance"],
note: str,
price: Optional["PriceDB"],
@ -193,7 +194,37 @@ class IngredientInstance(Element):
self.price = price
@classmethod
def from_dict(cls, ctx: Context, dct: Dict[str, Any]) -> Self:
def from_dict(cls, ctx: Context, dct: str | Dict[str, Any]) -> Self:
if isinstance(dct, str):
string = dct.strip()
p = re.compile(r"^(?:([0-9\.]+) ([a-zA-Z]+) )?([\w ]+)(?: \((.*)\))?$")
match = p.match(string)
if match is None:
raise RuntimeError(
"ingredient {string} regex not matched, it should be in the format [amount(num) unit(string, one word)] name(string, any number of words) [(note(string))]"
)
amount = float(match.group(1))
unitstr = match.group(2)
unit = ctx.default_unit
if unit is not None:
unitx = ctx.units.get(unitstr)
if unitx is None:
ctx.issues.error(issues.ISSUE_UNKNOWN_UNIT, "unknown unit {unitstr}")
else:
unit = unitx
name = match.group(3)
note = match.group(4)
if note is None:
note = ""
return cls(
ctx=ctx,
name=name,
amount=amount,
unit=unit,
alternatives=[],
note=note,
price=None,
)
name = dct["name"]
ingredient = ctx.ingredients.get(name)

View file

@ -13,17 +13,22 @@
"ingredients": {
"type": "array",
"items": {
"$id": "https://example.com/ingredient.json",
"type": "object",
"additionalProperties": false,
"required": [ "name" ],
"properties": {
"name": { "type": "string" },
"amount": { "type": "number" },
"unit": { "type": "string" },
"or": { "items": { "$ref": "/ingredient.json" } },
"note": { "type": "string" }
}
"oneOf": [
{ "type": "string" },
{
"$id": "https://example.com/ingredient.json",
"type": "object",
"additionalProperties": false,
"required": [ "name" ],
"properties": {
"name": { "type": "string" },
"amount": { "type": "number" },
"unit": { "type": "string" },
"or": { "items": { "$ref": "/ingredient.json" } },
"note": { "type": "string" }
}
}
]
}
},
"steps": {

View file

@ -1,6 +1,11 @@
{% extends "base.html" %}
{% macro ingredientpart(ing) -%}
{% if recipe.price != None %}{{price(ing.price)}} {%endif%}{{ing.amount|amountprint}} {{ing["unit"].name}} {{ ing.name }} {{ing.note}}
{% if recipe.price != None %}{{price(ing.price)}} {%endif%}
{% if ing.amount != None %} {{ing.amount|amountprint}}
{% if ing.unit != None %} {{ing.unit.name}}{% endif %}
{%endif%}
{{ ing.name }}
{% if ing.note != "" %} {{ing.note}}{% endif %}
{%- endmacro %}
{% macro ingredient(ing) -%}