summaryrefslogtreecommitdiff
path: root/day4/task3/client.py
blob: 7f5c298bed5356759be3e2ff36ad7dc672decaff (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import socket
from time import sleep
import struct
import json


def pack_data(data):
    encoded = data.encode()
    return struct.pack('I', len(encoded)) + encoded


def get_data(connection):
    buffer = b''
    while len(buffer) < 4:
        data = connection.recv(4 - len(buffer))
        if not data:
            raise ConnectionError('Connection was closed')

        buffer += data

    size_packed, buffer = buffer[:4], buffer[4:]
    if not size_packed:
        raise ConnectionError('Connection was closed')

    (size,) = struct.unpack('I', size_packed)
    data = connection.recv(size - len(buffer))
    return buffer + data


def parse_data(data):
    data = json.loads(data.decode())
    arr = [f'Найдено {len(data)} совпадений:']
    for error in data:
        arr.append(f"[{error['date']}][{error['type']}][{error['module']}][{error['id']}] {error['text']}")
    return '\n'.join(arr)


HOST, PORT = ADDR = 'localhost', 6001

cnt = 0
sock = socket.socket()
while cnt < 10:
    try:
        cnt += 1
        sock.connect(ADDR)

    except ConnectionRefusedError:
        print('Не удалось подключиться к серверу. Следующая попытка через 3 секунды...')
        sleep(3)

    else:
        del cnt
        break
else:
    print('Не удалось подключиться к серверу через 10 попыток.\nОстанавливаемся...')
    sock.close()
    quit()


while True:
    try:
        data = input('Введите модуль для поиска:\n')

    except KeyboardInterrupt:
        break

    else:
        sock.sendall(pack_data(data))
        response = get_data(sock)
        print(parse_data(response))

sock.close()