Use cache in get_ip_address_info. (#494)

domainaware · Mar 24, 2024 · 1e565d9 · 1e565d9
1 parent 3a1360a
commit 1e565d9
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/parsedmarc/utils.py b/parsedmarc/utils.py
@@ -313,10 +313,15 @@ def get_ip_address_info(ip_address, ip_db_path=None, cache=None, offline=False,
 
     """
     ip_address = ip_address.lower()
-    if cache:
+    if cache is not None:
         info = cache.get(ip_address, None)
         if info:
+            logger.debug("IP address " + ip_address + " was found in cache")
             return info
+        else:
+            logger.debug("IP address " + ip_address + " not found in cache")
+    else:
+        logger.debug("IP address cache not specified")
     info = OrderedDict()
     info["ip_address"] = ip_address
     if offline:
@@ -333,6 +338,9 @@ def get_ip_address_info(ip_address, ip_db_path=None, cache=None, offline=False,
         base_domain = get_base_domain(reverse_dns)
         info["base_domain"] = base_domain
 
+    if cache is not None:
+        cache[ip_address] = info
+
     return info