monkeypatch AzureBlobFileSystem._strip_protocol

Author: kevinjqliu

pyiceberg/io/fsspec.py

@@ -206,6 +206,16 @@ def _adls(properties: Properties) -> AbstractFileSystem:
     from azure.core.credentials import AccessToken
     from azure.core.credentials_async import AsyncTokenCredential
 
+    # Monkeypatch AzureBlobFileSystem._strip_protocol function
+    # adlfs expects abfs/abfss. For compatibility, convert wasb/wasbs to abfs/abfss
+    _original_strip_protocol = AzureBlobFileSystem._strip_protocol
+
+    def strip_protocol_patch(cls, path: str):  # type: ignore
+        path = path.replace("wasb://", "abfs://").replace("wasbs://", "abfss://")
+        return _original_strip_protocol(path)
+
+    AzureBlobFileSystem._strip_protocol = classmethod(strip_protocol_patch)
+
     for key, sas_token in {
         key.replace(f"{ADLS_SAS_TOKEN}.", ""): value for key, value in properties.items() if key.startswith(ADLS_SAS_TOKEN)
     }.items():