Python-socketio простой сервер-клиент. Должен ли работать sio.wait()?

Question

[vutmuk123]

Здравствуйте. Изучаю вебсокеты. Скопипастил простой сервер-клиент из документации к python-socketio. Это работает, коннект между сервером и клиентом есть. Дальше я добавил отдельный процесс на сервер, чтобы отправлять клиенту сообщение с клавиатуры. И вот сообщение не приходит клиенту) Не хотел бы утруждать вас кодом, но есть вопрос. В главном потоке клиента после коннекта работает sio.wait():

async def main():
    await sio.connect('http://localhost:8080/')
    await sio.wait()

В документации написано:
Если приложению нечего делать в основном потоке и он просто хочет дождаться завершения соединения с сервером, оно может вызвать wait()метод:
sio.wait()
Или в asyncioверсии:
await sio.wait()
В подсказке пайчарма написано:
Wait until the connection with the server ends.
Client applications can use this function to block the main thread during the life of the connection.
Т.е. метод wait тянет работу клиента до завершения соединения? Значит это типо заглушка для демонстрации работоспособности? Значит ли это, что он должен быть заменен на метод ожидающий какой-то запрос, или wait должен быть всегда в потоке?

Comments

[shurshur]

А как определено, что "не доходит"? Можно создать клиента с логгированием: socketio.Client(logger=True, engineio_logger=True) и попробовать поизучать происходящее более внимательно.

Насколько я понимаю, надо повесить с помощью sio.on обработчик для входящих запросов, иначе все прилетающие данные просто проигнорируются.

Мне потребовался socketio для парсинга одного сайта, и я там не использовал wait таким образом, тем более что у меня был синхронный изначально код (который уже делает много другой работы и его переписывать теперь несколько лениво). Я вместо этого делал короткие (по 0.1 с) sio.sleep, пока не получу ответ, если за 100 итераций (эквивалент 10 секунд) ничего не придёт - скрипт вываливался (это случалось в единичных случаях, а при повторном запуске не повторялось, что меня устраивало). sio.wait я делал только после sio.disconnect.

Вот такой класс я тогда набросал и он вполне работал в моей простой задаче:

cont_socketio = 'https://foo.bar'
cont_namespace = '/comments'

cont_connect_sleep = 0.1
cont_connect_iterations = 50
cont_query_sleep = 0.1
cont_query_iterations = 100

class SIOClient:
  def __init__(self):
    #self.sio = socketio.Client(logger=True, engineio_logger=True)
    self.sio = socketio.Client()
    self.ready = False
    def handle():
      self.ready = True
    self.sio.on("connect", handle, namespace=cont_namespace)
    self.sio.connect(cont_socketio, namespaces=[cont_namespace], transports=['websocket'])
    i = 0
    while i<cont_connect_iterations and not self.ready:
      i = i+1
      self.sio.sleep(cont_connect_sleep)
    assert self.ready

  def query(self, method, data):
    self.result = None
    def handle(data):
      self.result = data
    self.sio.on(method, handle, namespace=cont_namespace)
    self.sio.emit(method, data, namespace=cont_namespace)
    i = 0
    while i<cont_query_iterations and not self.result:
      i += 1
      self.sio.sleep(cont_query_sleep)
    assert self.result
    return self.result

  def quit(self):
    if self.ready:
      self.sio.disconnect()
      self.sio.wait()

[vutmuk123]

shurshur, да, извините, у меня проблема во втором процессе оказывается.... sio.wait не мешает принимать сообщения от сервера...

[vutmuk123]

Наконец-то получилось. Не знаю зачем, но я хотел отправлять сообщения прям из сервера...
это сервер:

from aiohttp import web
import socketio
from aioconsole import ainput

sio = socketio.AsyncServer()
app = web.Application()
sio.attach(app)

@sio.event
def connect(sid, environ):
    print("connect ", sid)

@sio.event
def disconnect(sid):
    print('disconnect ', sid)

async def chat_message():
    while True:
        text = await ainput(">>>")
        if text == 'exit':
            break
        else:
            await sio.emit("test", text)

sio.start_background_task(chat_message)
if __name__ == '__main__':
    web.run_app(app)

это клиент:

import asyncio
import socketio

sio = socketio.AsyncClient()

@sio.event
async def connect():
    print('connection established')

@sio.event
def test(data):
    print('message received with ', data)

@sio.event
async def disconnect():
    print('disconnected from server')


async def main():
    await sio.connect('http://localhost:8080')
    await sio.wait()

if __name__ == '__main__':
    asyncio.run(main())