Skip to content

Commit 18e8cea

Browse files
authored
[AIRFLOW-6428] Fix import path for airflow.utils.dates.days_ago in Example DAGs (#7007)
1 parent e740173 commit 18e8cea

68 files changed

Lines changed: 136 additions & 136 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

airflow/contrib/example_dags/example_databricks_operator.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,23 +32,22 @@
3232
https://docs.databricks.com/api/latest/jobs.html#runstate
3333
"""
3434

35-
import airflow
3635
from airflow import DAG
3736
from airflow.contrib.operators.databricks_operator import DatabricksSubmitRunOperator
37+
from airflow.utils.dates import days_ago
3838

3939
default_args = {
4040
'owner': 'airflow',
4141
'email': ['airflow@example.com'],
4242
'depends_on_past': False,
43-
'start_date': airflow.utils.dates.days_ago(2)
43+
'start_date': days_ago(2)
4444
}
4545

4646
with DAG(
4747
dag_id='example_databricks_operator',
4848
default_args=default_args,
4949
schedule_interval='@daily'
5050
) as dag:
51-
5251
new_cluster = {
5352
'spark_version': '2.1.0-db3-scala2.11',
5453
'node_type_id': 'r3.xlarge',

airflow/contrib/example_dags/example_dingding_operator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,14 +21,14 @@
2121
"""
2222
from datetime import timedelta
2323

24-
import airflow
2524
from airflow import DAG
2625
from airflow.contrib.operators.dingding_operator import DingdingOperator
26+
from airflow.utils.dates import days_ago
2727

2828
args = {
2929
'owner': 'airflow',
3030
'retries': 3,
31-
'start_date': airflow.utils.dates.days_ago(2)
31+
'start_date': days_ago(2)
3232
}
3333

3434

airflow/contrib/example_dags/example_emr_job_flow_automatic_steps.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,15 +21,15 @@
2121
"""
2222
from datetime import timedelta
2323

24-
import airflow
2524
from airflow import DAG
2625
from airflow.contrib.operators.emr_create_job_flow_operator import EmrCreateJobFlowOperator
2726
from airflow.contrib.sensors.emr_job_flow_sensor import EmrJobFlowSensor
27+
from airflow.utils.dates import days_ago
2828

2929
DEFAULT_ARGS = {
3030
'owner': 'airflow',
3131
'depends_on_past': False,
32-
'start_date': airflow.utils.dates.days_ago(2),
32+
'start_date': days_ago(2),
3333
'email': ['airflow@example.com'],
3434
'email_on_failure': False,
3535
'email_on_retry': False

airflow/contrib/example_dags/example_emr_job_flow_manual_steps.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,17 +24,17 @@
2424
"""
2525
from datetime import timedelta
2626

27-
import airflow
2827
from airflow import DAG
2928
from airflow.contrib.operators.emr_add_steps_operator import EmrAddStepsOperator
3029
from airflow.contrib.operators.emr_create_job_flow_operator import EmrCreateJobFlowOperator
3130
from airflow.contrib.operators.emr_terminate_job_flow_operator import EmrTerminateJobFlowOperator
3231
from airflow.contrib.sensors.emr_step_sensor import EmrStepSensor
32+
from airflow.utils.dates import days_ago
3333

3434
DEFAULT_ARGS = {
3535
'owner': 'airflow',
3636
'depends_on_past': False,
37-
'start_date': airflow.utils.dates.days_ago(2),
37+
'start_date': days_ago(2),
3838
'email': ['airflow@example.com'],
3939
'email_on_failure': False,
4040
'email_on_retry': False

airflow/contrib/example_dags/example_gcs_to_gdrive.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,13 +21,13 @@
2121
"""
2222
import os
2323

24-
import airflow
2524
from airflow import models
2625
from airflow.contrib.operators.gcs_to_gdrive_operator import GcsToGDriveOperator
26+
from airflow.utils.dates import days_ago
2727

2828
GCS_TO_GDRIVE_BUCKET = os.environ.get("GCS_TO_DRIVE_BUCKET", "example-object")
2929

30-
default_args = {"start_date": airflow.utils.dates.days_ago(1)}
30+
default_args = {"start_date": days_ago(1)}
3131

3232
with models.DAG(
3333
"example_gcs_to_gdrive", default_args=default_args, schedule_interval=None # Override to match your needs

airflow/contrib/example_dags/example_kubernetes_executor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,13 +21,13 @@
2121
"""
2222
import os
2323

24-
import airflow
2524
from airflow.models import DAG
2625
from airflow.operators.python_operator import PythonOperator
26+
from airflow.utils.dates import days_ago
2727

2828
args = {
2929
'owner': 'airflow',
30-
'start_date': airflow.utils.dates.days_ago(2)
30+
'start_date': days_ago(2)
3131
}
3232

3333
with DAG(

airflow/contrib/example_dags/example_kubernetes_executor_config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,14 +21,14 @@
2121
"""
2222
import os
2323

24-
import airflow
2524
from airflow.contrib.example_dags.libs.helper import print_stuff
2625
from airflow.models import DAG
2726
from airflow.operators.python_operator import PythonOperator
27+
from airflow.utils.dates import days_ago
2828

2929
default_args = {
3030
'owner': 'airflow',
31-
'start_date': airflow.utils.dates.days_ago(2)
31+
'start_date': days_ago(2)
3232
}
3333

3434
with DAG(

airflow/contrib/example_dags/example_papermill_operator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,13 +24,13 @@
2424

2525
from datetime import timedelta
2626

27-
import airflow
2827
from airflow.models import DAG
2928
from airflow.operators.papermill_operator import PapermillOperator
29+
from airflow.utils.dates import days_ago
3030

3131
default_args = {
3232
'owner': 'airflow',
33-
'start_date': airflow.utils.dates.days_ago(2)
33+
'start_date': days_ago(2)
3434
}
3535

3636
with DAG(

airflow/contrib/example_dags/example_qubole_operator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,16 +31,16 @@
3131
import filecmp
3232
import random
3333

34-
import airflow
3534
from airflow import DAG
3635
from airflow.contrib.operators.qubole_operator import QuboleOperator
3736
from airflow.operators.dummy_operator import DummyOperator
3837
from airflow.operators.python_operator import BranchPythonOperator, PythonOperator
38+
from airflow.utils.dates import days_ago
3939

4040
default_args = {
4141
'owner': 'airflow',
4242
'depends_on_past': False,
43-
'start_date': airflow.utils.dates.days_ago(2),
43+
'start_date': days_ago(2),
4444
'email': ['airflow@example.com'],
4545
'email_on_failure': False,
4646
'email_on_retry': False

airflow/contrib/example_dags/example_twitter_dag.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,11 +31,11 @@
3131
"""
3232
from datetime import date, timedelta
3333

34-
import airflow
3534
from airflow import DAG
3635
from airflow.operators.bash_operator import BashOperator
3736
from airflow.operators.hive_operator import HiveOperator
3837
from airflow.operators.python_operator import PythonOperator
38+
from airflow.utils.dates import days_ago
3939

4040
# --------------------------------------------------------------------------------
4141
# Create a few placeholder scripts. In practice these would be different python
@@ -74,7 +74,7 @@ def transfertodb():
7474
default_args = {
7575
'owner': 'Ekhtiar',
7676
'depends_on_past': False,
77-
'start_date': airflow.utils.dates.days_ago(5),
77+
'start_date': days_ago(5),
7878
'email': ['airflow@example.com'],
7979
'email_on_failure': False,
8080
'email_on_retry': False,

0 commit comments

Comments
 (0)