TandoorRecipes/cookbook/helper/unit_conversion_helper.py
2023-02-26 11:56:48 +01:00

52 lines
1.7 KiB
Python

from pint import UnitRegistry, UndefinedUnitError
def base_conversions(ingredient_list):
ingredient_list = [{'amount': 1, 'unit': 'g'}, {'amount': 1, 'unit': 'Gramm'}, {'amount': 1, 'unit': 'Stück'}]
ureg = UnitRegistry()
pint_converted_list = []
for i in ingredient_list:
try:
pint_converted_list.append(ureg.Quantity(f'{i["amount"]} {i["unit"]}'))
except UndefinedUnitError:
pass
print(pint_converted_list)
def get_conversions(amount, unit, food):
conversions = []
if unit:
for c in unit.unit_conversion_base_relation.all():
r = _uc_convert(c, amount, unit, food)
if r not in conversions:
conversions.append(r)
for c in unit.unit_conversion_converted_relation.all():
r = _uc_convert(c, amount, unit, food)
if r not in conversions:
conversions.append(r)
return conversions
def _uc_convert(uc, amount, unit, food):
if uc.food is None or uc.food == food:
if unit == uc.base_unit:
return {
'amount': amount * (uc.converted_amount / uc.base_amount),
'unit': {
'id': uc.converted_unit.id,
'name': uc.converted_unit.name,
'plural_name': uc.converted_unit.plural_name
},
}
else:
return {
'amount': amount * (uc.base_amount / uc.converted_amount),
'unit': {
'id': uc.base_unit.id,
'name': uc.base_unit.name,
'plural_name': uc.base_unit.plural_name
},
}