improve code

dnstapir · Nov 15, 2024 · 6ad04db · 6ad04db
1 parent b3c937b
commit 6ad04db
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/dnstapir/dns/mozpsl.py b/dnstapir/dns/mozpsl.py
@@ -43,12 +43,15 @@ def insert(self, array: list[str], nlbl: int, icann: bool) -> None:
 
     def search(self, key: list[str]) -> tuple[int, int]:
         """Search Trie"""
+        core = 0
+        pcore = 0
         current = self.root
         for label in key:
-            if current.icann:
+            if current.icann is True:
                 core = current.count
-            else:
+            elif current.icann is False:
                 pcore = current.count
+            # # If current.icann is None, do not update core or pcore
             if label not in current.children:
                 if current.count != 0:
                     break