refactor: cache load_rules()

ninoseki · May 14, 2022 · d58361c · d58361c
1 parent e28cdc6
commit d58361c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/abuse_whois/matchers/shared_hosting/rules.py b/abuse_whois/matchers/shared_hosting/rules.py
@@ -1,11 +1,13 @@
 import pathlib
+from functools import lru_cache
 from typing import List
 
 from abuse_whois.utils import load_yaml
 
 from .rule import SharedHostingRule
 
 
+@lru_cache(maxsize=1)
 def load_rules() -> List[SharedHostingRule]:
     rules: List[SharedHostingRule] = []
     for path in pathlib.Path(__file__).parent.glob("./rules/*.yaml"):

diff --git a/abuse_whois/matchers/whois/rules.py b/abuse_whois/matchers/whois/rules.py
@@ -1,11 +1,13 @@
 import pathlib
+from functools import lru_cache
 from typing import List
 
 from abuse_whois.utils import load_yaml
 
 from .rule import WhoisRule
 
 
+@lru_cache(maxsize=1)
 def load_rules() -> List[WhoisRule]:
     rules: List[WhoisRule] = []
     for path in pathlib.Path(__file__).parent.glob("./rules/*.yaml"):