python проблема с сокетами
Суть программы в том, чтобы делать захват экрана и отправлять серверу. Проблема в том, что клиент отправляет команду о новом файле, когда сервер не пишет о ее получении (случайно пишет)
Вот код сервера:
import socket
import os
s = socket.socket()
s.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR, 1)
try: #находиться путь к файлу полученому
path = os.path.abspath('getScreen.png')
except:
pass
s.bind(('', 9090))
s.listen(1)
conn, addr = s.accept()
op = open('getScreen.png', 'wb')
while 1:
data = conn.recv(1024)
#print(data)
if data == b'hlki8':#если клиент отправляет что новый файл удаляеться старый
print("new")
try:
op.close()
os.remove(path)
op = open('getScreen.png', 'wb')#и создается
except:
pass
else:
op.write(data)
op.close()
conn.close()
Вот код клиента:
import socket
import pyautogui
import os
s = socket.socket()
s.connect(('', 9090))
while 1:
print("yes")
screen = pyautogui.screenshot('screenshot.png')
op = open('screenshot.png', 'rb')
n = 1
while 1:
data = op.read(1024)
if not data:
s.send(b'hlki8')#отправляется команда о новом файле
break
s.send(data)
op.close()
s.shutdown(socket.SHUT_WR)