diff --git a/Whatsapp_Chat_Exporter/extract.py b/Whatsapp_Chat_Exporter/extract.py index 8dbaca1..1ed72b6 100644 --- a/Whatsapp_Chat_Exporter/extract.py +++ b/Whatsapp_Chat_Exporter/extract.py @@ -538,11 +538,13 @@ def create_html( template_dir = os.path.dirname(template) template_file = os.path.basename(template) templateLoader = jinja2.FileSystemLoader(searchpath=template_dir) - templateEnv = jinja2.Environment(loader=templateLoader) - templateEnv.globals.update(determine_day=determine_day) - templateEnv.globals.update(no_avatar=no_avatar) - templateEnv.filters['sanitize_except'] = sanitize_except - template = templateEnv.get_template(template_file) + template_env = jinja2.Environment(loader=templateLoader) + template_env.globals.update( + determine_day=determine_day, + no_avatar=no_avatar + ) + template_env.filters['sanitize_except'] = sanitize_except + template = template_env.get_template(template_file) total_row_number = len(data) print(f"\nGenerating chats...(0/{total_row_number})", end="\r") diff --git a/Whatsapp_Chat_Exporter/extract_iphone.py b/Whatsapp_Chat_Exporter/extract_iphone.py index 8a0f739..707015b 100644 --- a/Whatsapp_Chat_Exporter/extract_iphone.py +++ b/Whatsapp_Chat_Exporter/extract_iphone.py @@ -263,11 +263,13 @@ def create_html( template_dir = os.path.dirname(template) template_file = os.path.basename(template) templateLoader = jinja2.FileSystemLoader(searchpath=template_dir) - templateEnv = jinja2.Environment(loader=templateLoader) - templateEnv.globals.update(determine_day=determine_day) - templateEnv.globals.update(no_avatar=no_avatar) - templateEnv.filters['sanitize_except'] = sanitize_except - template = templateEnv.get_template(template_file) + template_env = jinja2.Environment(loader=templateLoader) + template_env.globals.update( + determine_day=determine_day, + no_avatar=no_avatar + ) + template_env.filters['sanitize_except'] = sanitize_except + template = template_env.get_template(template_file) total_row_number = len(data) print(f"\nGenerating chats...(0/{total_row_number})", end="\r")