pycoze 0.1.187__py3-none-any.whl → 0.1.189__py3-none-any.whl

pycoze/automation/browser/edge_driver_manager.py

@@ -0,0 +1,77 @@
+import os
+import requests
+from selenium.webdriver.edge.service import Service
+from webdriver_manager.microsoft import EdgeChromiumDriverManager
+from selenium import webdriver
+
+
+# Function to check network availability
+def is_network_available():
+    try:
+        # Attempt to access the download URL for Edge Chromium driver to check network connection
+        response = requests.get("https://msedgedriver.azureedge.net/", timeout=5)
+        return response.status_code == 200
+    except requests.ConnectionError:
+        return False
+
+
+# Function to get the default download path for Edge Chromium driver
+def get_default_driver_path():
+    manager = EdgeChromiumDriverManager()
+    return manager.install()
+
+
+# Function to get the driver version from cache
+def get_cached_driver_version(manager):
+    cache_dir = manager._cache_manager._root_dir  # Get the root path of the cache directory
+    os_type = manager._os_system_manager.get_os_type()  # Get the OS type
+    driver_name = manager.driver.get_name()
+
+    # List all versions in the cache directory
+    versions_dir = os.path.join(cache_dir, "drivers", driver_name, os_type)
+    if os.path.exists(versions_dir):
+        versions = os.listdir(versions_dir)
+        if versions:
+            # Return the latest version found in the cache
+            return max(versions)
+    return None
+
+
+# Function to check if the driver has already been downloaded
+def get_driver_path():
+    network_available = is_network_available()
+    print("Network status:", "Available" if network_available else "Unavailable")
+    manager = EdgeChromiumDriverManager()
+    if network_available:
+        return get_default_driver_path()
+    else:
+        # If no network, check the cache directory
+        driver_version = get_cached_driver_version(manager)
+        if driver_version:
+            driver_name = manager.driver.get_name()
+            os_type = manager._os_system_manager.get_os_type()
+            cache_dir = manager._cache_manager._root_dir
+
+            # Determine the correct driver filename based on the OS
+            if os.name == 'nt':  # Windows
+                driver_filename = "msedgedriver.exe"
+            else:  # macOS and Linux
+                driver_filename = "msedgedriver"
+
+            # Construct the driver path
+            driver_path = os.path.join(cache_dir, "drivers", driver_name, os_type, driver_version, driver_filename)
+            # Check if the driver path exists
+            if os.path.exists(driver_path):
+                return driver_path
+        raise Exception("Network unavailable and no downloaded driver found")
+
+
+# Function to get the Edge driver service
+def get_edge_driver_service():
+    driver_path = get_driver_path()
+    print("driver_path", driver_path)
+    # Check if the driver path exists
+    if os.path.exists(driver_path):
+        return Service(driver_path)
+    else:
+        raise Exception("Network unavailable and no downloaded driver found")

{pycoze-0.1.187.dist-info → pycoze-0.1.189.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.187
+Version: 0.1.189
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.187.dist-info → pycoze-0.1.189.dist-info}/RECORD

@@ -1,6 +1,9 @@
 pycoze/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/ai/__init__.py,sha256=odM2lgYSnApxw4AzLV_5AyaxL5_MLfydOSB1VmLGFyA,75
 pycoze/ai/vram_reserve.py,sha256=jtNsXamhwquyL9_qknZVu-8vQWaPfUdmvjLbDZaZ4vU,4262
+pycoze/automation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pycoze/automation/browser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pycoze/automation/browser/edge_driver_manager.py,sha256=tJEGXab1XTk2LPjhpQ6NfAbRCdrsuyCDexMvYxZPUEM,3022
 pycoze/bot/__init__.py,sha256=6HHMxDQVOyZM9dtSjQm9tjGnhj4h7CixD0JOvEwTi48,41
 pycoze/bot/agent_chat.py,sha256=KshLVuXSnkWsp1J9xKKhoIMvAfQXg5ZyzSkPZdwcRkM,2785
 pycoze/bot/bot.py,sha256=CCnBmxtmpmz7aLR_MIIiOuKR9IACo5P4uSkQXb5d0KU,3061
@@ -24,8 +27,8 @@ pycoze/utils/__init__.py,sha256=Gi5EnrWZGMD2JRejgV4c_VLCXyvA2wwBFI_niDF5MUE,110
 pycoze/utils/arg.py,sha256=GtfGbMTMdaK75Fwh6MpUe1pCA5X6Ep4LFG7a72YrzjI,525
 pycoze/utils/env.py,sha256=W04lhvTHhAAC6EldP6kk2xrctqtu8K6kl1vDLZDNeh8,561
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.187.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.187.dist-info/METADATA,sha256=Ejm-h9bGJdLQe97ljW7nKFHL2MgwMbPQs_1T5uyqZNg,726
-pycoze-0.1.187.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-pycoze-0.1.187.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.187.dist-info/RECORD,,
+pycoze-0.1.189.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.189.dist-info/METADATA,sha256=yGD8p-JPeousEHUQZv7VQEr9ReGC51lMUgNsnK9Ny7k,726
+pycoze-0.1.189.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+pycoze-0.1.189.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.189.dist-info/RECORD,,