Работа с сокетом в python?

Question

[Владимир Куц]

Есть код на php:

if ( ! feof($this->_socket))
        {
            $packet->clean();
            $this->read($packet);
        }

Я его переписываю на Python, немного изменяя под свои нужды

while True:
            self.packet.clean()
            self.packet.read()
            ...

Где в self.packet.read - последовательное чтение из сокета и распаковка нужных данных что-то вроде

...
self.code = int(struct.unpack('B', self.sock.recv(1))[0])
version = int(struct.unpack('B', self.sock.recv(1))[0])
...

после чтения которых, должна вычитываться очередная порция, если таковую сервер высылает, и продолжать выполнение, если данных от сервера более нет.

И тут сталкиваюсь с проблемой. Код на php вычитывает нужную порцию данных, успешно их обрабатывает до конца, и успешно продолжает выполнение программы. Код на python же, вычитывает до конца, обрабатывает их, но ожидаемо, - останавливается в этом месте, продолжая ожидать данные, которые сервер уже не отправит. Вылет и обработка исключения по таймауту ломает логику программы.
Каким образом корректно отработать эту ситуацию под Python 3.x? Кто сталкивался?

Answer 1

[Сергей]

обычно так делается

def recv_basic(the_socket):
    total_data=[]
    while True:
        data = the_socket.recv(8192)
        if not data: break
        total_data.append(data)
    return ''.join(total_data)

Comments

[Владимир Куц]

Спасибо! Попробую с буфером. Там же если пакет меньшим размером придет - в data будет пакет нужного размера? А то там пакеты в бинарном виде определенного (разного размера) приходят. И данные буквально по байтам разбирать надо.

[Владимир Куц]

Дошли руки проверить. Нет - не работает. На моменте когда пытаюсь вычитать очередной байт - работа кода останавливается. Следующий байт уже не приходит.

Answer 2

[MechanicZelenyy]

Может быть проблема в том, что while True: - это бесконечный цикл и нужно вызвать break после того, как данные вычитаны?

Comments

[Владимир Куц]

Да - именно в этом проблема. Мне каким-то образом нужно перехватить момент, когда все данные вычитаны. Проблема в том, что после чтения текущего пакета, следующий пакет может придти, а может и нет. PHPс этим справляется, просто ожидая eof. В Питоне я не нашел как это сделать.

[Владимир Куц]

Если я просто вызову break после вычитки первого пакета, я не получу данных из пакетов последующих.

[Tark]

У вас есть self.packet.read(), который наверняка возвращает полученные данные (?). Если данных нет - цикл можно прерывать.

[Владимир Куц]

Tark: я читаю пакет побайтово, делаю парсинг данных. Далее если сервер отправляет данные - я должен получить следующий пакет. Если нет - то прерваться и продолжить дальнейшее выполнение.

[Владимир Куц]

Tark: Если тупо выйду после вычитки первого пакета - данные остальных пакетов потеряются. Если я попытаюсь вычитать пакет после прекращения отсылки данных то в этом месте процесс останавливается, поскольку процедура self.packet.read() не сможет вычитать пакет и программа отвалится по таймауту, поскольку сервер в ответ ничего отсылать уже не будет. PHP тупо выходит по EOF. Как сделать это на Питоне - я не нашел.