Compartilhar via


Gerenciar arquivos em volumes do Catálogo do Unity com o Simba (Driver ODBC) do Databricks

Esta página descreve como carregar, baixar e excluir arquivos em volumes do Catálogo do Unity usando o Driver ODBC do Databricks.

Requisitos

  • Databricks ODBC Driver versão 2.8.2 ou superior
  • Modo de consulta nativo habilitado (padrão). Se estiver desabilitado, adicione UseNativeQuery=1 ou UseNativeQuery=2 à cadeia de conexão.

Para obter um exemplo completo do Python com configuração de autenticação, consulte Conectar Python e pyodbc ao Azure Databricks.

Fazer upload de um arquivo

Para carregar um arquivo, adicione a StagingAllowedLocalPaths propriedade à cadeia de conexão com o caminho do arquivo a ser carregado. Para vários locais de origem, use uma lista separada por vírgulas (por exemplo, /tmp/,/usr/tmp/).

Importante

Em ambientes multilocatários em que os usuários controlam a URL ODBC (como ferramentas de BI ou serviços de desenvolvedor), defina o StagingAllowedLocalPaths para uma área restrita ou caminho inexistente. Isso impede que os usuários gravem arquivos arbitrários e interfiram na implantação interna do serviço.

Para substituir um arquivo existente, adicione OVERWRITE à instrução.

conn_string = "".join([
    "DRIVER=", os.getenv("ODBC_DRIVER", "/Library/simba/spark/lib/libsparkodbc_sbu.dylib"),
    ";Host=", os.getenv("ODBC_HOST_NAME", "<<HOST_NAME>>"),
    ";PORT=443",
    ";HTTPPath=", os.getenv("ODBC_HTTP_PATH", "/sql/1.0/endpoints/1234567890"),
    ";AuthMech=11",
    ";SSL=1",
    ";ThriftTransport=2",
    ";SparkServerType=3",
    ";Auth_Flow=0",
    ";Auth_AccessToken=", os.getenv("API_TOKEN", "<<NO_ACCESS_TOKEN_IS_SET>>"),
    ";StagingAllowedLocalPaths=", "/tmp"),
    os.getenv("ODBC_OPTIONS", ""),
])

conn = pyodbc.connect(conn_string, autocommit=True)
cursor = conn.cursor()
cursor.execute("PUT '" +
    "/tmp/my-data.csv" +
    "' INTO '" +
    "/Volumes/main/default/my-volume/my-data.csv" +
    "' OVERWRITE")

Baixar um arquivo

Use GET para baixar um arquivo de um volume para um caminho local:

conn = pyodbc.connect(conn_string, autocommit=True)
cursor = conn.cursor()
cursor.execute("GET '" +
    "/Volumes/main/default/my-volume/my-data.csv" +
    "' TO '" +
    "/tmp/my-downloaded-data.csv" +
    "'")

Excluir um arquivo

Use REMOVE para excluir um arquivo de um volume:

conn = pyodbc.connect(conn_string, autocommit=True)
cursor = conn.cursor()
cursor.execute("REMOVE '" +
    "/Volumes/main/default/my-volume/my-data.csv" +
    "'")