pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pass file-like object to Docx.render method

Browse Source
This commit is contained in:
dandds 2018-12-21 14:05:45 -05:00
parent 5cf2534445
commit 6527f72e78
3 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
atst/routes/task_orders/index.py
View File

@ -1,3 +1,4 @@
from io import BytesIO
from flask import Response from flask import Response
from . import task_orders_bp from . import task_orders_bp
@ -8,7 +9,8 @@ from atst.utils.docx import Docx
@task_orders_bp.route("/task_orders/download_summary/<task_order_id>") @task_orders_bp.route("/task_orders/download_summary/<task_order_id>")
def download_summary(task_order_id): def download_summary(task_order_id):
task_order = TaskOrders.get(task_order_id) task_order = TaskOrders.get(task_order_id)
byte_str = Docx.render(data=task_order.to_dictionary()) byte_str = BytesIO()
Docx.render(byte_str, data=task_order.to_dictionary())
filename = "{}.docx".format(task_order.portfolio_name) filename = "{}.docx".format(task_order.portfolio_name)
return Response( return Response(
byte_str, byte_str,

10
atst/utils/docx.py
View File

@ -1,5 +1,4 @@
import os import os
from io import BytesIO
from zipfile import ZipFile from zipfile import ZipFile
from flask import render_template, current_app as app from flask import render_template, current_app as app
@ -31,15 +30,14 @@ class Docx:
@classmethod @classmethod
def render( def render(
cls, cls,
file_like,
doc_template="docx/document.xml", doc_template="docx/document.xml",
file_template="docx/template.docx", file_template="docx/template.docx",
**args, **args,
): ):
document = render_template(doc_template, **args) document = render_template(doc_template, **args)
byte_str = BytesIO() with ZipFile(file_like, mode="w") as docx_file:
docx_file = ZipFile(byte_str, mode="w")
docx_template = Docx._template(file_template) docx_template = Docx._template(file_template)
Docx._write(docx_template, docx_file, document) Docx._write(docx_template, docx_file, document)
docx_file.close() file_like.seek(0)
byte_str.seek(0) return file_like
return byte_str

3
tests/utils/test_docx.py
View File

@ -6,7 +6,8 @@ from atst.utils.docx import Docx
def test_render_docx(): def test_render_docx():
data = {"droid_class": "R2"} data = {"droid_class": "R2"}
docx_file = Docx.render(data=data) byte_str = BytesIO()
docx_file = Docx.render(byte_str, data=data)
zip_ = ZipFile(docx_file, mode="r") zip_ = ZipFile(docx_file, mode="r")
document = zip_.read(Docx.DOCUMENT_FILE) document = zip_.read(Docx.DOCUMENT_FILE)
assert b"droid_class: R2" in document assert b"droid_class: R2" in document