Update api/validate.py

api/validate.py

@@ -1,11 +1,10 @@
 import requests
 import re
 import time
+from api.config import BASE_URL, headers
+from api.logger import setup_logger
 
-base_url = "https://www.blackbox.ai"
-headers = {
-    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
-}
+logger = setup_logger(__name__)
 
 # Cache variables
 cached_hid = None
@@ -16,14 +15,14 @@ def getHid(force_refresh=False):
     global cached_hid, cache_time
     current_time = time.time()
 
-    # Check if we need to refresh the hid
+    # Check if we need to refresh or cache is valid
     if not force_refresh and cached_hid and (current_time - cache_time) < CACHE_DURATION:
-        print("Using cached hid:", cached_hid)
+        logger.info(f"Using cached_hid: {cached_hid}")
         return cached_hid
 
     try:
-        # Get the initial HTML content
-        response = requests.get(base_url, headers=headers)
+        # Get initial HTML content
+        response = requests.get(BASE_URL, headers=headers)
         response.raise_for_status()
         content = response.text
 
@@ -32,31 +31,31 @@ def getHid(force_refresh=False):
         match = re.search(pattern, content)
 
         if match:
-            # Construct the full URL of the JS file
+            # Construct full URL of JS file
             js_path = match.group()
-            full_url = f"{base_url}/_next/{js_path}"
+            full_url = f"{BASE_URL}/_next/{js_path}"
 
-            # Request the JS file content
+            # Get JS file content
             js_response = requests.get(full_url, headers=headers)
             js_response.raise_for_status()
 
-            # Search for h-value in the JS content
+            # Search for h-value in JS content
             h_pattern = r'h="([0-9a-f-]+)"'
             h_match = re.search(h_pattern, js_response.text)
 
             if h_match:
                 h_value = h_match.group(1)
-                print("Found h-value:", h_value)
+                logger.info(f"Found h-value: {h_value}")
                 # Update cache
                 cached_hid = h_value
                 cache_time = current_time
                 return h_value
             else:
-                print("Did not find h-value in JS content")
+                logger.error("h-value not found in JS content")
                 return None
         else:
-            print("Did not find specified JS file path in HTML content")
+            logger.error("Specified JS file path not found in HTML content")
             return None
     except requests.exceptions.RequestException as e:
-        print(f"Error occurred: {e}")
+        logger.error(f"An error occurred: {e}")
         return None