CHIEFSOFT\ameye
50 lines
1.1 KiB
Python
50 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': 'DigiFi',
|
|
'depends_on_past': False,
|
|
'start_date': datetime(2022, 1, 1),
|
|
'email_on_failure': False,
|
|
'email_on_retry': False,
|
|
'retries': 1
|
|
}
|
|
|
|
|
|
def DigiFiTrxDataGenerator(**kwargs):
|
|
url = kwargs["api_host"] + "/transaction/generator"
|
|
querystring= {"testuser":"ameye"}
|
|
response = requests.request("GET", url, params=querystring)
|
|
print(response.text)
|
|
|
|
with DAG(
|
|
dag_id="DIGIFI_CRON_NEWEMPLOYER",
|
|
default_args=default_args,
|
|
start_date=datetime(2024, 9, 1, 10, 00),
|
|
schedule_interval=timedelta(minutes=1),
|
|
catchup=False) as dag:
|
|
t1 = PythonOperator(
|
|
task_id = "digifi_cron_demotrx_generator",
|
|
python_callable= DigiFiTrxDataGenerator,
|
|
op_kwargs={
|
|
"api_host" : Variable.get("DEMO_DIGIFI_HOST")
|
|
}
|
|
)
|
|
|
|
# http://10.10.11.17:6335/transaction/generator
|
|
|
|
t2 = DummyOperator(
|
|
task_id="end"
|
|
)
|
|
|
|
t1 >> t2
|
|
|
|
|