| # |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| |
| |
| from typing import Any, Dict |
| |
| import requests |
| |
| from airflow.decorators import dag, task |
| from airflow.models.baseoperator import BaseOperator |
| from airflow.operators.email import EmailOperator |
| from airflow.utils.dates import days_ago |
| |
| DEFAULT_ARGS = {"owner": "airflow"} |
| |
| |
| class GetRequestOperator(BaseOperator): |
| """Custom operator to sand GET request to provided url""" |
| |
| def __init__(self, *, url: str, **kwargs): |
| super().__init__(**kwargs) |
| self.url = url |
| |
| def execute(self, context): |
| return requests.get(self.url).json() |
| |
| |
| # [START dag_decorator_usage] |
| @dag(default_args=DEFAULT_ARGS, schedule_interval=None, start_date=days_ago(2), tags=['example']) |
| def example_dag_decorator(email: str = 'example@example.com'): |
| """ |
| DAG to send server IP to email. |
| |
| :param email: Email to send IP to. Defaults to example@example.com. |
| :type email: str |
| """ |
| get_ip = GetRequestOperator(task_id='get_ip', url="http://httpbin.org/get") |
| |
| @task(multiple_outputs=True) |
| def prepare_email(raw_json: Dict[str, Any]) -> Dict[str, str]: |
| external_ip = raw_json['origin'] |
| return { |
| 'subject': f'Server connected from {external_ip}', |
| 'body': f'Seems like today your server executing Airflow is connected from IP {external_ip}<br>', |
| } |
| |
| email_info = prepare_email(get_ip.output) |
| |
| EmailOperator( |
| task_id='send_email', to=email, subject=email_info['subject'], html_content=email_info['body'] |
| ) |
| |
| |
| dag = example_dag_decorator() |
| # [END dag_decorator_usage] |