persistant_bash_terminal.py

import pexpect


class BashProcess:
    """A class for running persistent bash processes."""

    def __init__(self):
        self.shell = pexpect.spawn('/bin/bash', encoding='utf-8', timeout=300)
        self.shell.expect_exact('$')
        self.shell.sendline("\n")
        self.shell.expect_exact('$')
        self.remove = str(self.shell.before).strip()

    def run(self, command):
        if command == "exit":
            self.shell.close()
            self.shell = pexpect.spawn('/bin/bash', encoding='utf-8', timeout=300)
            self.shell.expect_exact('$')
            return "Shell restarted."
        commands = command.split('\n')
        output = ""
        for cmd in commands:
            self.shell.sendline(cmd)
            self.shell.expect_exact('$')
            output += str(self.shell.before).strip().replace(self.remove, "$ ")
        return output


def bash_terminal():
    bash = BashProcess()
    return bash