athena_create_table_generator.py

.py

import boto3

def get_glue_table_schema(database_name, table_name):
    client = boto3.client('glue')

    response = client.get_table(
        DatabaseName=database_name,
        Name=table_name
    )

    schema = response['Table']['StorageDescriptor']['Columns']
    return schema

def generate_athena_create_ddl(database_name, table_name):
    schema = get_glue_table_schema(database_name, table_name)

    ddl_columns = [f"`{col['Name']}` {col['Type']}" for col in schema]
    ddl_statement = f"CREATE EXTERNAL TABLE IF NOT EXISTS `{table_name}` (\n\t{',\n\t'.join(ddl_columns)}\n)\nSTORED AS PARQUET\nLOCATION 's3://your-bucket/{table_name}/'\n"

    return ddl_statement

# Example usage
database_name = 'your-database-name'
table_name = 'your-table-name'

ddl = generate_athena_create_ddl(database_name, table_name)
print(ddl)