dremio_flight.py

from pyarrow import flight
import base64,time,pandas

class HttpDremioClientAuthHandler(flight.ClientAuthHandler): 

  def __init__(self, username, password): 
    super().__init__() 
    self.username = bytes(username)
    self.password = bytes(password) 
    self.token = None

  def authenticate(self, outgoing, incoming): 
    outgoing.write(base64.b64encode(self.username + b':' + self.password))
    self.token = incoming.read()

  def get_token(self): 
    return self.token


c = flight.FlightClient.connect('grpc+tcp://localhost:47470')
c.authenticate(HttpDremioClientAuthHandler(b'dremio',b'dremio123'))

sql = 'SELECT 1'

df = c.do_get(c.get_flight_info(flight.FlightDescriptor.for_command(sql)).endpoints[0].ticket).read_all().to_pandas()

print(df.to_string())