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'