flux-proxy-ssh.py

#!/usr/bin/env python3

import os
import sys
import subprocess

def find_split_point(argv):
    for idx, val in enumerate(argv):
        if val == "relay":
            return idx - 1

def main():
    ld_lib_path = os.getenv("LD_LIBRARY_PATH")
    new_argv = sys.argv.copy()
    new_argv[0] = os.getenv("FLUX_SSH_SSH", "mrsh")
    if ld_lib_path:
        split_point = find_split_point(sys.argv)
        new_argv.insert(split_point, "LD_LIBRARY_PATH={}".format(ld_lib_path))
    subprocess.Popen(new_argv).communicate()

if __name__ == "__main__":
    main()