pylab

#!/usr/bin/env python3

import getpass
import argparse
from pexpect import pxssh


def connect(host):
    try:
        username = input("Enter username: ")
        password = getpass.getpass("Enter password: ")
        s = pxssh.pxssh()
        s.login(host, username, password)
        print("SSH connection is established!")
        s.sendline("PS1='\\n\\w\\$ '")
        s.prompt()
        while True:
            command = input(s.before.decode() + "\nEnter command: ")
            if command.strip() == "q" or command.strip() == "Q":
                break
            s.sendline(command)
            s.prompt()
            with open("ssh.log", "a") as f:
                f.write(s.before.decode())
    except pxssh.ExceptionPxssh as e:
        print("Connection refused!")
        return
    finally:
        s.logout()


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--host", default="localhost")
    args = parser.parse_args()
    connect(args.host)