mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-18 23:02:11 +00:00
33 lines
883 B
Python
33 lines
883 B
Python
|
from datetime import datetime
|
||
|
|
|
||
|
|
from airflow import DAG
|
||
|
|
from airflow.providers.snowflake.operators.snowflake import SnowflakeOperator
|
||
|
|
|
||
|
|
SNOWFLAKE_COST_TABLE = "costs"
|
||
|
|
SNOWFLAKE_PROCESSED_TABLE = "processed_costs"
|
||
|
|
|
||
|
|
with DAG(
|
||
|
|
"snowflake_operator",
|
||
|
|
start_date=datetime(2023, 1, 1),
|
||
|
|
schedule_interval=None,
|
||
|
|
catchup=False,
|
||
|
|
) as dag:
|
||
|
|
transform_cost_table = SnowflakeOperator(
|
||
|
|
snowflake_conn_id="my_snowflake",
|
||
|
|
task_id="transform_cost_table",
|
||
|
|
sql="""
|
||
|
|
CREATE OR REPLACE TABLE {{ params.out_table_name }} AS
|
||
|
|
SELECT
|
||
|
|
id,
|
||
|
|
month,
|
||
|
|
total_cost,
|
||
|
|
area,
|
||
|
|
total_cost / area as cost_per_area
|
||
|
|
FROM {{ params.in_table_name }}
|
||
|
|
""",
|
||
|
|
params={
|
||
|
|
"in_table_name": SNOWFLAKE_COST_TABLE,
|
||
|
|
"out_table_name": SNOWFLAKE_PROCESSED_TABLE,
|
||
|
|
},
|
||
|
|
)
|