diff --git a/BDSM/commands.py b/BDSM/commands.py index 447bee1..1f6dbcb 100644 --- a/BDSM/commands.py +++ b/BDSM/commands.py @@ -16,6 +16,88 @@ def initdb(drop): db.create_all() click.echo('Initialized database.') +@app.cli.command() +def renderfile(): + """render toot""" + from BDSM.models import Toot + from BDSM.views import process_toot + from jinja2 import Environment, FileSystemLoader + + head = ''' + +
+ + + + ''' + def render_toot(toot_id): + _toot = [] + toot = Toot.query.get(toot_id) + + if toot == None: + return "None" + + _toot.append(toot) + _toot = process_toot(_toot) + env = Environment(loader=FileSystemLoader("./BDSM/templates")) + jinja_template = env.get_template("toot.html") + template_string = jinja_template.render(toots=_toot) + return template_string + + def _render(input_file): + env = Environment(loader=FileSystemLoader("./")) + jinja_template = env.get_template(input_file) + jinja_template.globals.update(render_toot=render_toot) + template_string = jinja_template.render() + + with open('./output.html', 'w') as f: + f.write(head + template_string + '