junshan-kit 2.1.9__py2.py3-none-any.whl → 2.2.0__py2.py3-none-any.whl

junshan_kit/meta.py

@@ -202,17 +202,21 @@ class JianguoyunDownloaderFirefox:
             print("❌ Failed to find or click the download button:", e)
             raise
 
-
     def wait_for_download(self, timeout=3600):
+        """Wait until all downloads are finished (auto-detects browser type)."""
         print("⏳ Waiting for downloads to finish...")
         start_time = time.time()
+
+        # Determine the temporary file extension based on the browser type
+        temp_ext = ".crdownload" if "chrome" in self.driver.capabilities["browserName"].lower() else ".part"
+
         while time.time() - start_time < timeout:
-            files = os.listdir(self.download_path)
-            if any(not f.endswith(".part") for f in files):
+            downloading = [f for f in os.listdir(self.download_path) if f.endswith(temp_ext)]
+            if not downloading:
                 print("✅ Download completed!")
-                return
-            time.sleep(5)
-        print("⚠️ Timeout: download not completed within 1 hour")
+                return True
+            time.sleep(2)
+
 
     def close(self):
         print("🛑 Closing browser...")
@@ -235,14 +239,14 @@ class JianguoyunDownloaderFirefox:
 #                           MAIN
 # =============================================================
 
-# if __name__ == "__main__":
-#     url = "https://www.jianguoyun.com/p/DdyHJxUQqdHDDRjvtI0GIAA"
+if __name__ == "__main__":
+    url = "https://www.jianguoyun.com/p/DdyHJxUQqdHDDRjvtI0GIAA"
 
-#     use_chrome = True  # Set True to use Chrome, False for Firefox
+    use_chrome = True  # Set True to use Chrome, False for Firefox
 
-#     if use_chrome:
-#         downloader = JianguoyunDownloaderChrome(url)
-#     else:
-#         downloader = JianguoyunDownloaderFirefox(url)
+    if use_chrome:
+        downloader = JianguoyunDownloaderChrome(url)
+    else:
+        downloader = JianguoyunDownloaderFirefox(url)
 
-#     downloader.run()
+    downloader.run()

{junshan_kit-2.1.9.dist-info → junshan_kit-2.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: junshan_kit
-Version: 2.1.9
+Version: 2.2.0
 Summary: This is an optimization tool.
 Author-email: Junshan Yin <junshanyin@163.com>
 Requires-Dist: kaggle==1.7.4.5

{junshan_kit-2.1.9.dist-info → junshan_kit-2.2.0.dist-info}/RECORD

@@ -1,8 +1,8 @@
 junshan_kit/DataProcessor.py,sha256=AW_1jROexC3s41-RgzqzYVwPI0sOf3tzjiph4qa_Vcw,3882
 junshan_kit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 junshan_kit/datahub.py,sha256=I34e26psFS8WK4X6SNucKPLtdBm0Ujzqa0VDIRACah4,5163
-junshan_kit/meta.py,sha256=wzctYAckD3Gex1MMK5J6diV5z66E_rDUb814NzBQ5VY,9816
+junshan_kit/meta.py,sha256=5aHyUPVr3P3yoAdC4DzOZv4AtaO9iX8zGjluwpOly6Q,10017
 junshan_kit/test.py,sha256=uSckjcr_Wgj__YPTwD6x0GY8Hfn5GBEXIpRf9vIYBbU,91
-junshan_kit-2.1.9.dist-info/METADATA,sha256=AjGMtbds79-V-BRJoU-vx0_PYNy9tJ380zOKlI8bOPs,329
-junshan_kit-2.1.9.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-junshan_kit-2.1.9.dist-info/RECORD,,
+junshan_kit-2.2.0.dist-info/METADATA,sha256=aWDiR4w_Z7sVVrLcqjQNYgt3L-iFWSydzcoiUPqDsg8,329
+junshan_kit-2.2.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+junshan_kit-2.2.0.dist-info/RECORD,,