Azure support about tap-spreadsheets-anywhere HOT 8 CLOSED

def get_streamreader(uri, universal_newlines=True,newline='',open_mode='r'):
    transport_params = None

    if uri.startswith('azure://'):
        connect_str = os.environ['AZURE_STORAGE_CONNECTION_STRING']
        transport_params = {
            'client': BlobServiceClient.from_connection_string(connect_str),
        }

    streamreader = smart_open.open(uri, open_mode, newline=newline, errors='surrogateescape', transport_params=transport_params)
    if not universal_newlines and isinstance(streamreader, StreamReader):
        return monkey_patch_streamreader(streamreader)
    return streamreader

kwarg_dispatch = {
    "azure": lambda: {
        "transport_params": {
            "client": BlobServiceClient.from_connection_string(
                os.environ['AZURE_STORAGE_CONNECTION_STRING'],
            )
        }
    },
    "gcs": lambda: {
        "transport_params": {
            "client": storage.Client.from_service_account_json(
                os.environ['GOOGLE_APPLICATION_CREDENTIALS'],
                # We can add more nuanced transport params here
            )
        }
    },
   # Adding support for more is intuitive...
}

SCHEME_SEP = "://"
kwargs = kwarg_dispatch.get(uri.split(SCHEME_SEP, 1)[0], lambda: {})()
    
streamreader = smart_open.open(uri, open_mode, newline=newline, errors='surrogateescape', **kwargs)
if not universal_newlines and isinstance(streamreader, StreamReader):
    return monkey_patch_streamreader(streamreader)
return streamreader

def get_streamreader(uri: str, universal_newlines: bool = True, newline: str = "", open_mode: str = "r"):
  return (lambda rdr: rdr if not universal_newlines and isinstance(rdr, StreamReader) 
                          else monkey_patch_streamreader(rdr))(
    smart_open.open(uri, open_mode, newline=newline, errors="surrogateescape", **{
        "azure": lambda: {
            "transport_params": {
                "client": BlobServiceClient.from_connection_string(
                    os.environ["AZURE_STORAGE_CONNECTION_STRING"],
                )
            }
        },
        "gcs": lambda: {
            "transport_params": {
                "client": storage.Client.from_service_account_json(
                    os.environ["GOOGLE_APPLICATION_CREDENTIALS"],
                    # We can add more nuanced transport params here
                )
            }
        },
       # Adding support for more is intuitive...
    }.get(uri.split("://", 1)[0], lambda: {})())
  )