fix(contact): add matching regex logic (#36)

Nexters · Oct 29, 2023 · dcdbba6 · dcdbba6
1 parent 0a334fc
commit dcdbba6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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,16 +296,18 @@ 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):
         with open(f"./content/contacts/{idx}.json", mode="w", encoding="utf-8") as f:
             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):