向Python脚本传参数

from airflow.hooks.oracle_hook import OracleHook
import sys
import argparse
oracle_conn = OracleHook.get_hook(conn_id='etbc_oracle')
#查数据量
def get_count(src_sql):
        return oracle_conn.get_records(src_sql)

if __name__=='__main__':
        parser = argparse.ArgumentParser(description='manual to this script')
        parser.add_argument('--sql', type=str, default = None)
        args = parser.parse_args()
        print(get_count(args.sql))

# python3 oracle_count.py --sql='select count(1) from ci_adj'

 

posted @ 2020-10-30 16:32  Mars.wang  阅读(111)  评论(0编辑  收藏  举报