Airflow 動(dòng)態(tài)創(chuàng)建Task

import airflow
import MySQLdb
from impala.util import as_pandas
from airflow import DAG
from airflow.operators.dummy_operator import DummyOperator
from airflow.operators.bash_operator import BashOperator

# todo 優(yōu)化腳本

def get_datas():
    db = MySQLdb.connect(host="xxx.xx.xx.xx", port=4313, user="xxxx", password="xxxx",
                         database="xxxx", charset='utf8')
    cur = db.cursor()
    cur.execute(
        "select * from xxx ")
    return as_pandas(cur).fillna(0).to_dict(orient='records')


def multitasking_task(xx, dag):
    python_command = "python3"
    py_path = "/data/airflow/dag_scripts/xxxx.py"
    return BashOperator(
        task_id='handle_task_with_id_{}'.format(xx['id']),
        bash_command="{} {} {} {}".format(python_command, py_path, "{{ ds }}", xx['id']),
        dag=dag,
    )


def running():
    default_args = {
        'owner': 'xx',
        'start_date': airflow.utils.dates.days_ago(1),
        'email': ['xx@xx.cn'],
        'email_on_failure': False,
        'email_on_retry': False,
    }

    dag = DAG(dag_id='xx',
              default_args=default_args,
              schedule_interval='10 05 * * *', )

    start = DummyOperator(
        task_id="start",
        dag=dag
    )

    end = DummyOperator(
        task_id="end",
        dag=dag
    )

    # todo 優(yōu)化task過多情況,可以動(dòng)態(tài)創(chuàng)建dag任務(wù)

    for data in get_datas():
        start >> multitasking_task(data , dag) >> end

    return dag


dag = running()

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市差油,隨后出現(xiàn)的幾起案子伤溉,更是在濱河造成了極大的恐慌孽惰,老刑警劉巖,帶你破解...
    沈念sama閱讀 212,816評(píng)論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件聚蝶,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)秽澳,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,729評(píng)論 3 385
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來戏羽,“玉大人担神,你說我怎么就攤上這事∈蓟ǎ” “怎么了妄讯?”我有些...
    開封第一講書人閱讀 158,300評(píng)論 0 348
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)酷宵。 經(jīng)常有香客問我亥贸,道長(zhǎng),這世上最難降的妖魔是什么浇垦? 我笑而不...
    開封第一講書人閱讀 56,780評(píng)論 1 285
  • 正文 為了忘掉前任砌函,我火速辦了婚禮,結(jié)果婚禮上溜族,老公的妹妹穿的比我還像新娘讹俊。我一直安慰自己,他們只是感情好煌抒,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,890評(píng)論 6 385
  • 文/花漫 我一把揭開白布仍劈。 她就那樣靜靜地躺著,像睡著了一般寡壮。 火紅的嫁衣襯著肌膚如雪贩疙。 梳的紋絲不亂的頭發(fā)上讹弯,一...
    開封第一講書人閱讀 50,084評(píng)論 1 291
  • 那天,我揣著相機(jī)與錄音这溅,去河邊找鬼组民。 笑死,一個(gè)胖子當(dāng)著我的面吹牛悲靴,可吹牛的內(nèi)容都是我干的臭胜。 我是一名探鬼主播,決...
    沈念sama閱讀 39,151評(píng)論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼癞尚,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼耸三!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起浇揩,我...
    開封第一講書人閱讀 37,912評(píng)論 0 268
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤仪壮,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后胳徽,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體积锅,經(jīng)...
    沈念sama閱讀 44,355評(píng)論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,666評(píng)論 2 327
  • 正文 我和宋清朗相戀三年养盗,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了乏沸。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 38,809評(píng)論 1 341
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡蹬跃,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出铆铆,到底是詐尸還是另有隱情蝶缀,我是刑警寧澤,帶...
    沈念sama閱讀 34,504評(píng)論 4 334
  • 正文 年R本政府宣布薄货,位于F島的核電站翁都,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏谅猾。R本人自食惡果不足惜柄慰,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 40,150評(píng)論 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望税娜。 院中可真熱鬧坐搔,春花似錦、人聲如沸敬矩。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,882評(píng)論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)弧岳。三九已至凳忙,卻和暖如春业踏,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背涧卵。 一陣腳步聲響...
    開封第一講書人閱讀 32,121評(píng)論 1 267
  • 我被黑心中介騙來泰國(guó)打工勤家, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人柳恐。 一個(gè)月前我還...
    沈念sama閱讀 46,628評(píng)論 2 362
  • 正文 我出身青樓伐脖,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親胎撤。 傳聞我的和親對(duì)象是個(gè)殘疾皇子晓殊,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,724評(píng)論 2 351

推薦閱讀更多精彩內(nèi)容