From fab8a2574f2c4369e8d208b870bc8d6e0da0eed6 Mon Sep 17 00:00:00 2001
From: LmeSzinc <lmeszincsales@gmail.com>
Date: Fri, 19 Jun 2020 02:27:15 +0800
Subject: [PATCH] Dev: word template extractor

---
 dev_tools/word_template_extractor.py | 46 ++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 dev_tools/word_template_extractor.py

diff --git a/dev_tools/word_template_extractor.py b/dev_tools/word_template_extractor.py
new file mode 100644
index 000000000..a60036424
--- /dev/null
+++ b/dev_tools/word_template_extractor.py
@@ -0,0 +1,46 @@
+import re
+
+from dev_tools.slpp import slpp
+
+"""
+This file is used to extract `word_template.lua`, aka, the blacklist words file.
+
+Git clone the repository here, https://github.com/Dimbreath/AzurLaneData, to get the decrypted scripts.
+Then put your filepath here, like `<your_folder>/<server>/sharecfg/word_template.lua`
+Server list: en-US, ja-JP, ko-KR, zh-CN, zh-TW
+"""
+file = ''
+count = 0
+with open(file, 'r', encoding='utf-8') as f:
+    text = f.read()
+
+
+def extract(dic, word_list):
+    """
+    Args:
+        dic (dict):
+        word_list (list[str]):
+    """
+    global count
+    for word, data in dic.items():
+        word = str(word)
+        if data.get('this', False):
+            new = word_list + [word]
+            new = ''.join(new)
+            count += 1
+            print(new)
+        else:
+            new = word_list + [word]
+            extract(data, word_list=new)
+
+
+# CN server
+for result in re.findall('word_template = (.*?)return', text, re.DOTALL):
+    pg = slpp.decode(result)
+    extract(pg, word_list=[])
+# Other server
+for result in re.findall('uv0\.{0,1}(.*?)end', text, re.DOTALL):
+    pg = slpp.decode('{%s}' % result)
+    extract(pg, word_list=[])
+
+print(f'Total count: {count}')