需要借助exe_command函数，该函数通过subprocess执行 shell 命令，并实时打印输出。
from subprocess import Popen, PIPE, STDOUT
def exe_command(command):
"""
执行 shell 命令并实时打印输出
:param command: shell 命令
:return: process, exitcode
"""
print(command)
process = Popen(command, stdout=PIPE, stderr=STDOUT, shell=True)
with process.stdout:
for line in iter(process.stdout.readline, b''):
print(line.decode().strip())
exitcode = process.wait()
return process, exitcode
def exe_sql_command(sql_command, redirect=None):
"""
执行 hive sql 命令并转储到文件
:param sql_command: sql 命令
:param redirect: 是否重定向输出到文件
:return: process, exitcode
"""
command = "hive -e "%s"" % sql_command
if redirect:
command += f" > {redirect}"
return exe_command(command)
def exe_sql_file(sql_file, redirect=None):
"""
执行 hive sql 文件并转储到文件
:param sql_file: sql 文件
:param redirect: 是否重定向输出到文件
:return: process, exitcode
"""
command = f"hive -f {sql_file}"
if redirect:
command += f" > {redirect}"
return exe_command(command)
