Исправления третьей задачи

1 files changed, 45 insertions, 32 deletions

diff --git a/day4/task3/client.py b/day4/task3/client.py
index 49663c8..18bb2b1 100644
--- a/day4/task3/client.py
+++ b/day4/task3/client.py
@@ -4,6 +4,10 @@ import struct
 import json
 
 
+HOST, PORT = ADDR = 'localhost', 6000
+DATATYPE_SIZE = 4  # int
+
+
 def pack_data(data):
     encoded = data.encode()
     return struct.pack('I', len(encoded)) + encoded
@@ -11,14 +15,15 @@ def pack_data(data):
 
 def get_data(connection):
     buffer = b''
-    while len(buffer) < 4:
-        data = connection.recv(4 - len(buffer))
+    # Считываем размер пакета
+    while len(buffer) < DATATYPE_SIZE:
+        data = connection.recv(DATATYPE_SIZE - len(buffer))
         if not data:
             raise ConnectionError('Connection was closed')
 
         buffer += data
 
-    size_packed, buffer = buffer[:4], buffer[4:]
+    size_packed, buffer = buffer[:DATATYPE_SIZE], buffer[DATATYPE_SIZE:]
     if not size_packed:
         raise ConnectionError('Connection was closed')
 
@@ -31,42 +36,50 @@ def parse_data(data):
     return json.loads(data.decode())
 
 
-HOST, PORT = ADDR = 'localhost', 6000
+def try_connect():
+    cnt = 0
+    sock = socket.socket()
+    while cnt < 10:
+        try:
+            cnt += 1
+            sock.connect(ADDR)
 
-cnt = 0
-sock = socket.socket()
-while cnt < 10:
-    try:
-        cnt += 1
-        sock.connect(ADDR)
+        except ConnectionRefusedError:
+            print('Не удалось подключиться к серверу. Следующая попытка через 3 секунды...')
+            sleep(3)
 
-    except ConnectionRefusedError:
-        print('Не удалось подключиться к серверу. Следующая попытка через 3 секунды...')
-        sleep(3)
+        else:
+            return sock
 
     else:
-        del cnt
-        break
-else:
-    print('Не удалось подключиться к серверу через 10 попыток.\nОстанавливаемся...')
-    sock.close()
-    quit()
+        print('Не удалось подключиться к серверу через 10 попыток.\nОстанавливаемся...')
+        sock.close()
+        return
 
 
-while True:
-    try:
-        data = input('Введите модуль для поиска:\n')
+def main():
+    sock = try_connect()
+    if sock is None:
+        quit()
 
-    except KeyboardInterrupt:
-        break
+    while True:
+        try:
+            data = input('Введите модуль для поиска:\n')
 
-    else:
-        sock.sendall(pack_data(data))
-        response = get_data(sock)
-        arr = parse_data(response)
+        except KeyboardInterrupt:
+            break
+
+        else:
+            sock.sendall(pack_data(data))
+            response = get_data(sock)
+            arr = parse_data(response)
+
+            print(f'Найдено {len(arr)} совпадений:')
+            for line in arr:
+                print(line)
+
+    sock.close()
 
-        print(f'Найдено {len(arr)} совпадений:')
-        for line in arr:
-            print(line)
 
-sock.close()
+if __name__ == '__main__':
+    main()