Skip to content

Commit 2d7bea6

Browse files
committed
fix(export): Set default language to en on pdf export
1 parent dd941de commit 2d7bea6

1 file changed

Lines changed: 21 additions & 8 deletions

File tree

api/tasks.py

Lines changed: 21 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from .utils import DebugPlaywright
2020

2121

22-
def build_storage_state(tmp_dir, user, token, language):
22+
def build_storage_state(tmp_dir, user, token, language="en"):
2323
temp_file = pathlib.Path(tmp_dir, "storage_state.json")
2424
temp_file.touch()
2525

@@ -40,7 +40,7 @@ def build_storage_state(tmp_dir, user, token, language):
4040
}
4141
),
4242
},
43-
{"name": "language", "value": json.dumps(language)}, # NOTE: Use the language from the request for PDF export
43+
{"name": "language", "value": json.dumps(language)},
4444
],
4545
}
4646
]
@@ -88,12 +88,25 @@ def generate_url(url, export_id, user, title, language):
8888
with tempfile.TemporaryDirectory() as tmp_dir:
8989
with sync_playwright() as p:
9090
browser = p.chromium.connect(settings.PLAYWRIGHT_SERVER_URL)
91-
storage_state = build_storage_state(
92-
tmp_dir,
93-
user,
94-
token,
95-
language,
96-
)
91+
# NOTE: DREF Export use the language from request
92+
if export.export_type in [
93+
Export.ExportType.DREF,
94+
Export.ExportType.OPS_UPDATE,
95+
Export.ExportType.FINAL_REPORT,
96+
]:
97+
storage_state = build_storage_state(
98+
tmp_dir,
99+
user,
100+
token,
101+
language,
102+
)
103+
else:
104+
# NOTE: Other Export types use default language (en)
105+
storage_state = build_storage_state(
106+
tmp_dir,
107+
user,
108+
token,
109+
)
97110
context = browser.new_context(storage_state=storage_state)
98111
page = context.new_page()
99112
if settings.DEBUG_PLAYWRIGHT:

0 commit comments

Comments
 (0)