update

count_ip_data.py

@@ -29,7 +29,6 @@ WHITELIST_IPS = [
 
 logging.basicConfig(level=logging.WARNING)
 log = logging.getLogger(__name__)
-# log.disabled = True
 
 def get_ip_from_jsonl(file_path):
     """Extract IP from the first line of a JSONL file"""
@@ -149,7 +148,7 @@ def count_files_per_ip(smb_url, start_date_str="2025_02_18"):
     password = unquote(password)
     
     # Register the SMB session
-    smbclient.register_session(server, username=username, password=password)
+    smbclient.register_session(server, username=username, password=password, connection_timeout=1000)
     
     # Convert start date string to datetime
     start_date = datetime.strptime(start_date_str, "%Y_%m_%d")