import socket
import os
host = '127.0.0.1'
port = 10000

client_socket = socket.socket()
client_socket.connect((host, port))

while True:
    command = input('enter a command: ')
    if command == 'screenshot':
        print("taking screenshot...")
        client_socket.send(b'screenshot')
        path_to_save = os.path.expanduser("~/Desktop")+'/MyImage.png'
        with open(path_to_save, 'wb') as target_file:
            data = client_socket.recv(2048)
            while not data == b'finish':
                target_file.write(data)
                data = client_socket.recv(2048)

        print(f"screenshot saved successfully at {path_to_save}")