CHIEFSOFT\ameye
48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
|
|
from airflow import DAG
|
|
from airflow.operators.dummy import DummyOperator
|
|
# from airflow.operators.python import PythonOperator
|
|
from airflow.operators.python_operator import PythonOperator
|
|
from datetime import datetime, timedelta
|
|
from airflow.models import Variable
|
|
|
|
import requests
|
|
|
|
default_args = {
|
|
'owner': 'WrenchBoard',
|
|
'depends_on_past': False,
|
|
'start_date': datetime(2022, 1, 1),
|
|
'email_on_failure': False,
|
|
'email_on_retry': False,
|
|
'retries': 1
|
|
}
|
|
|
|
|
|
def WrenchCronNotificationsApi(**kwargs):
|
|
url = kwargs["api_host"] + "/v1/notifications"
|
|
querystring= {"testuser":"ameye"}
|
|
response = requests.request("GET", url, params=querystring)
|
|
print(response.text)
|
|
|
|
with DAG(
|
|
dag_id="WRENCBOARD_LIVE_NOTIFICATIONS",
|
|
default_args=default_args,
|
|
start_date=datetime(2024, 9, 1, 10, 00),
|
|
schedule_interval=timedelta(minutes=300),
|
|
catchup=False) as dag:
|
|
t1 = PythonOperator(
|
|
task_id = "wrenchboard_livenotifications",
|
|
python_callable= WrenchCronNotificationsApi,
|
|
op_kwargs={
|
|
"api_host" : Variable.get("API_WRENCHBOARD_HOST")
|
|
}
|
|
)
|
|
|
|
t2 = DummyOperator(
|
|
task_id="end"
|
|
)
|
|
|
|
t1 >> t2
|
|
|
|
|