1919from .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