如何从python调用Snowsql客户端

How to call snowsql client from python(如何从python调用Snowsql客户端)

本文介绍了如何从python调用Snowsql客户端的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

我从shell脚本调用Snowsql客户端。我正在通过执行源代码导入属性文件。并调用Snowsql客户端。我怎样才能在Python中做同样的事情呢?如有任何帮助,我们将不胜感激。

调用Snowsql客户端的Shell脚本:

source /opt/data/airflow/config/cyrus_de/snowflake_config.txt


sudo /home/user/snowsql -c $connection --config=/home/user/.snowsql/config -w $warehouse  --variable database_name=$dbname --variable stage=$stagename  --variable env=$env  -o exit_on_error=true -o variable_substitution=True -q /data/snowsql/queries.sql

Python

假设您切换到使用推荐答案纯粹是为了改进控制流,并且仍然希望继续使用Shell功能,则直接转换将需要编写一个acts as the source command导入环境变量的函数,然后在与Shell一起执行的子流程调用中使用它们allow environment variable substitution:

import os, shlex, subprocess

def source_file_into_env():
  command = shlex.split("env -i bash -c 'source /opt/data/airflow/config/cyrus_de/snowflake_config.txt && env'")
  proc = subprocess.Popen(command, stdout = subprocess.PIPE)
  for line in proc.stdout:
    (key, _, value) = line.partition("=")
    os.environ[key] = value
  proc.communicate()

def run():
  source_file_into_env()

  subprocess.run("""sudo /home/user/snowsql 
      -c $connection 
      --config=/home/user/.snowsql/config 
      -w $warehouse 
      --variable database_name=$dbname 
      --variable stage=$stagename 
      --variable env=$env 
      -o exit_on_error=true 
      -o variable_substitution=True 
      -q /data/snowsql/queries.sql""", 
    shell=True, 
    env=os.environ)

if __name__ == '__main__':
  run()

如果您希望使用纯Python而不使用任何shell调用,那么可以使用更多的native connector offered by Snowflake而不是snowsql。这将是一个更具侵略性的更改,但connection examples将帮助您入门。

这篇关于如何从python调用Snowsql客户端的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本文标题为:如何从python调用Snowsql客户端

基础教程推荐