perf: ⚡️ speed up decrypt_key

ZhaoQi99 · Nov 5, 2024 · dc928a1 · dc928a1
1 parent a071ad1
commit dc928a1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pyencrypt/decrypt.py b/pyencrypt/decrypt.py
@@ -1,9 +1,12 @@
-from pathlib import Path
 import os
+from functools import lru_cache
+from pathlib import Path
+
 from pyencrypt.aes import aes_decrypt
 from pyencrypt.ntt import intt
 
 
+@lru_cache
 def decrypt_key(cipher_key: str, d: int, n: int) -> str:
     plain_ls = list()
     for num in map(int, cipher_key.split("O")):