当在Airflow DAG中使用SSH连接时,可以通过paramiko.SSHClient()方法连接到远程服务器。但在某些情况下,连接后需要切换用户,这就需要使用paramiko库提供的.invoke_shell()方法来创建一个互动式的Shell会话。然而,在使用这种方式时,存在无法切换用户的问题,所以需要使用以下代码段来解决:
import paramiko
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(hostname=, username=, password=, port=22)
transport = ssh.get_transport()
session = transport.open_session()
session.set_combine_stderr(True)
session.get_pty()
session.exec_command(f"sudo su - ")
使用以上代码段来连接到远程服务器并切换用户。其中,通过session.set_combine_stderr(True)将标准输出和标准错误合并,可以避免因为没有权限导致无法执行相关操作。在执行完切换用户的命令后,可以在session中执行需要的命令。