Skip to content

Commit 796fee4

Browse files
committed
Update variables usage in bigquery_github_trends DAG
1 parent 8441236 commit 796fee4

File tree

2 files changed

+15
-5
lines changed

2 files changed

+15
-5
lines changed

examples/gcloud-example/dags/bigquery_github/bigquery_github_trends.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,17 @@
22
from datetime import timedelta, datetime
33

44
from airflow import DAG
5+
from airflow.models import Variable
56
from airflow.contrib.operators.bigquery_operator import BigQueryOperator
67
from airflow.contrib.operators.bigquery_check_operator import BigQueryCheckOperator
78

9+
10+
# Config variables
11+
dag_config = Variable.get("bigquery_github_trends_variables", deserialize_json=True)
12+
BQ_CONN_ID = dag_config["bq_conn_id"]
13+
BQ_PROJECT = dag_config["bq_project"]
14+
BQ_DATASET = dag_config["bq_dataset"]
15+
816
default_args = {
917
'owner': 'airflow',
1018
'depends_on_past': True,
@@ -28,11 +36,6 @@
2836
schedule_interval=schedule_interval
2937
)
3038

31-
# Config variables
32-
BQ_CONN_ID = "my_gcp_conn"
33-
BQ_PROJECT = "my-bq-project"
34-
BQ_DATASET = "my-bq-dataset"
35-
3639
## Task 1: check that the github archive data has a dated table created for that date
3740
# To test this task, run this command:
3841
# docker-compose -f docker-compose-gcloud.yml run --rm webserver airflow test bigquery_github_trends bq_check_githubarchive_day 2018-12-01
Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
{
2+
"bigquery_github_trends_variables": {
3+
"bq_conn_id": "my_gcp_conn",
4+
"bq_project": "my_bq_project",
5+
"bq_dataset": "my_bq_dataset"
6+
}
7+
}

0 commit comments

Comments
 (0)