airflow.providers.common.sql.datafusion.format_handlers

Classes

ParquetFormatHandler

Parquet format handler.

CsvFormatHandler

CSV format handler.

AvroFormatHandler

Avro format handler.

IcebergFormatHandler

Iceberg format handler for DataFusion.

Functions

get_format_handler(datasource_config)

Get a format handler based on the format type.

Module Contents

class airflow.providers.common.sql.datafusion.format_handlers.ParquetFormatHandler(datasource_config)[source]

Bases: airflow.providers.common.sql.datafusion.base.FormatHandler

Parquet format handler.

property get_format: airflow.providers.common.sql.config.FormatType[source]

Return the format type.

register_data_source_format(ctx)[source]

Register a data source format.

class airflow.providers.common.sql.datafusion.format_handlers.CsvFormatHandler(datasource_config)[source]

Bases: airflow.providers.common.sql.datafusion.base.FormatHandler

CSV format handler.

property get_format: airflow.providers.common.sql.config.FormatType[source]

Return the format type.

register_data_source_format(ctx)[source]

Register a data source format.

class airflow.providers.common.sql.datafusion.format_handlers.AvroFormatHandler(datasource_config)[source]

Bases: airflow.providers.common.sql.datafusion.base.FormatHandler

Avro format handler.

property get_format: airflow.providers.common.sql.config.FormatType[source]

Return the format type.

register_data_source_format(ctx)[source]

Register a data source format.

class airflow.providers.common.sql.datafusion.format_handlers.IcebergFormatHandler(datasource_config)[source]

Bases: airflow.providers.common.sql.datafusion.base.FormatHandler

Iceberg format handler for DataFusion.

Loads an Iceberg table from a catalog using IcebergHook and registers it with a DataFusion SessionContext via register_table_provider.

property get_format: airflow.providers.common.sql.config.FormatType[source]

Return the format type.

register_data_source_format(ctx)[source]

Register an Iceberg table with the DataFusion session context.

airflow.providers.common.sql.datafusion.format_handlers.get_format_handler(datasource_config)[source]

Get a format handler based on the format type.

Was this entry helpful?