39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
import bleach
|
|
import markdown as md
|
|
from bleach_whitelist import markdown_tags, markdown_attrs
|
|
from jinja2 import Template
|
|
|
|
from cookbook.helper.mdx_attributes import MarkdownFormatExtension
|
|
from cookbook.helper.mdx_urlize import UrlizeExtension
|
|
|
|
|
|
class IngredientObject(object):
|
|
amount = None
|
|
unit = None
|
|
food = None
|
|
|
|
def __init__(self, ingredient):
|
|
self.amount = f'[[calculateAmount({ingredient.amount})]]'
|
|
self.unit = ingredient.unit
|
|
self.food = ingredient.food
|
|
|
|
def __str__(self):
|
|
return f'{self.amount} {self.unit} {self.food}'
|
|
|
|
|
|
def render_instructions(step): # TODO deduplicate markdown cleanup code
|
|
|
|
ingredients = []
|
|
|
|
for i in step.ingredients.all():
|
|
ingredients.append(IngredientObject(i))
|
|
|
|
template = Template(step.instruction)
|
|
instructions = template.render(ingredients=ingredients)
|
|
|
|
tags = markdown_tags + ['pre', 'table', 'td', 'tr', 'th', 'tbody', 'style', 'thead']
|
|
parsed_md = md.markdown(instructions, extensions=['markdown.extensions.fenced_code', 'tables', UrlizeExtension(), MarkdownFormatExtension()])
|
|
markdown_attrs['*'] = markdown_attrs['*'] + ['class']
|
|
|
|
return bleach.clean(parsed_md, tags, markdown_attrs)
|