`airflow.providers.amazon.aws.hooks.glue_catalog`¶

This module contains AWS Glue Catalog Hook

Module Contents¶

class airflow.providers.amazon.aws.hooks.glue_catalog.AwsGlueCatalogHook(*args, **kwargs)[source]¶

Bases: airflow.providers.amazon.aws.hooks.base_aws.AwsBaseHook

Interact with AWS Glue Catalog

Additional arguments (such as aws_conn_id) may be specified and are passed down to the underlying AwsBaseHook.

See also

AwsBaseHook

get_partitions(self, database_name: str, table_name: str, expression: str = '', page_size: Optional[int] = None, max_items: Optional[int] = None)[source]¶

Retrieves the partition values for a table.

Parameters

database_name (str) -- The name of the catalog database where the partitions reside.
table_name (str) -- The name of the partitions' table.
expression (str) -- An expression filtering the partitions to be returned. Please see official AWS documentation for further information. https://docs.aws.amazon.com/glue/latest/dg/aws-glue-api-catalog-partitions.html#aws-glue-api-catalog-partitions-GetPartitions
page_size (int) -- pagination size
max_items (int) -- maximum items to return

Returns

set of partition values where each value is a tuple since a partition may be composed of multiple columns. For example: {('2018-01-01','1'), ('2018-01-01','2')}

check_for_partition(self, database_name: str, table_name: str, expression: str)[source]¶

Checks whether a partition exists

Parameters

database_name (str) -- Name of hive database (schema) @table belongs to
table_name (str) -- Name of hive table @partition belongs to

Expression

Expression that matches the partitions to check for (eg a = 'b' AND c = 'd')

Return type

bool

>>> hook = AwsGlueCatalogHook()
>>> t = 'static_babynames_partitioned'
>>> hook.check_for_partition('airflow', t, "ds='2015-01-01'")
True

get_table(self, database_name: str, table_name: str)[source]¶

Get the information of the table

Parameters

database_name (str) -- Name of hive database (schema) @table belongs to
table_name (str) -- Name of hive table

Return type

dict

>>> hook = AwsGlueCatalogHook()
>>> r = hook.get_table('db', 'table_foo')
>>> r['Name'] = 'table_foo'

get_table_location(self, database_name: str, table_name: str)[source]¶

Get the physical location of the table

Parameters

database_name (str) -- Name of hive database (schema) @table belongs to
table_name (str) -- Name of hive table

Returns

str

airflow.providers.amazon.aws.hooks.glue_catalog¶

Module Contents¶

`airflow.providers.amazon.aws.hooks.glue_catalog`¶