Python 脚本如何通过 SSH 设置 Linux 环境变量？

import paramiko
import os

def set_env_via_ssh(hostname, username, password, env_dict):
    """
    通过SSH设置Linux环境变量
    
    参数:
        hostname: 远程主机地址
        username: SSH用户名
        password: SSH密码
        env_dict: 要设置的环境变量字典 {变量名: 值}
    """
    # 创建SSH客户端
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    
    try:
        # 连接远程主机
        client.connect(hostname, username=username, password=password)
        
        # 对于每个环境变量，写入到~/.bashrc
        for key, value in env_dict.items():
            # 构建设置环境变量的命令
            env_line = f'export {key}="{value}"\n'
            
            # 检查是否已存在该变量设置
            check_cmd = f'grep -n "export {key}=" ~/.bashrc || echo "not found"'
            stdin, stdout, stderr = client.exec_command(check_cmd)
            result = stdout.read().decode().strip()
            
            if result != "not found":
                # 如果已存在，先删除原有行
                line_num = result.split(':')[0]
                delete_cmd = f'sed -i "{line_num}d" ~/.bashrc'
                client.exec_command(delete_cmd)
            
            # 追加新的环境变量设置
            append_cmd = f'echo \'{env_line}\' >> ~/.bashrc'
            client.exec_command(append_cmd)
            
            # 立即生效（当前会话）
            immediate_cmd = f'export {key}="{value}"'
            client.exec_command(immediate_cmd)
            
            print(f"已设置环境变量: {key}={value}")
        
        # 也可以选择写入到/etc/environment（需要sudo权限）
        # 或者写入到~/.profile、~/.bash_profile等
        
    except paramiko.AuthenticationException:
        print("SSH认证失败，请检查用户名和密码")
    except paramiko.SSHException as e:
        print(f"SSH连接错误: {e}")
    except Exception as e:
        print(f"发生错误: {e}")
    finally:
        client.close()

# 使用示例
if __name__ == "__main__":
    # 配置SSH连接信息
    HOST = "192.168.1.100"  # 远程主机IP
    USER = "your_username"  # SSH用户名
    PASS = "your_password"  # SSH密码
    
    # 要设置的环境变量
    environment_vars = {
        "JAVA_HOME": "/usr/lib/jvm/java-11-openjdk",
        "APP_HOME": "/opt/myapp",
        "DATABASE_URL": "postgresql://user:pass@localhost/db"
    }
    
    # 执行设置
    set_env_via_ssh(HOST, USER, PASS, environment_vars)

这个脚本用了paramiko库来SSH连接Linux服务器，主要做了几件事：

1. 先检查~/.bashrc里有没有已经存在的环境变量设置，有的话就删掉原来的
2. 把新的环境变量追加到~/.bashrc文件末尾
3. 同时用export命令让当前会话立即生效

用的时候记得先装paramiko：pip install paramiko，然后把HOST、USER、PASS换成你自己的服务器信息，environment_vars字典里放你要设的变量。

注意这方法修改的是用户级别的环境变量（~/.bashrc），如果要在系统级别设置（/etc/environment）或者给所有用户生效，需要sudo权限，而且得用不同的文件。

总结：用paramiko SSH连接后修改bashrc文件来设置环境变量。