From dcdbba6529fab8c9037d7e07f1e21a492cad670b Mon Sep 17 00:00:00 2001 From: Gibeom Lim Date: Mon, 30 Oct 2023 00:29:45 +0900 Subject: [PATCH] fix(contact): add matching regex logic (#36) --- content_generator/_utils.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/content_generator/_utils.py b/content_generator/_utils.py index 068c233..f9515f9 100644 --- a/content_generator/_utils.py +++ b/content_generator/_utils.py @@ -1,5 +1,6 @@ import json from os import remove, listdir +import re from typing import List GoogleSheetResult = List[List[str]] @@ -295,8 +296,10 @@ def make_contacts(contact_us: GoogleSheetResult, faq: GoogleSheetResult) -> None } _faqs.append(_faq) + content_file_pattern = re.compile("[0-9]+\.json") + for filename in listdir("./content/contacts"): - if filename.endswith(".json"): + if content_file_pattern.match(filename): remove(f"./content/contacts/{filename}") for idx, _contact in enumerate(_contacts): @@ -304,7 +307,7 @@ def make_contacts(contact_us: GoogleSheetResult, faq: GoogleSheetResult) -> None f.write(json.dumps(_contact, ensure_ascii=False, indent=2)) for filename in listdir("./content/contacts/faq"): - if filename.endswith(".json"): + if content_file_pattern.match(filename): remove(f"./content/contacts/faq/{filename}") for idx, _faq in enumerate(_faqs):