У меня очень простой сервер, который читает из подпроцесса и передает данные во все открытые веб-сокеты.Проблема, с которой я сталкиваюсь, заключается в том, что мой подход к чтению из подпроцесса, кажется, ломает aiohttp таким образом, что я не могу следовать:
#!/usr/bin/env python3
import asyncio
from aiohttp import web
import subprocess
async def websocket_handler(request):
ws = web.WebSocketResponse()
await ws.prepare(request)
request.app['websockets'].append(ws)
try:
async for msg in ws:
print(msg)
await asyncio.sleep(1)
finally:
request.app['websockets'].remove(ws)
return ws
async def on_shutdown(app):
for ws in app['websockets']:
await ws.close(code=999, message='Server shutdown')
Вот где что-то идет не так:
async def listen_to_process(app):
print("listen to process")
while True:
print(" looping? ")
await asyncio.sleep(0.1)
# the problem seems to be here
line = await app['process'].stdout.readline()
# if line:
# buffer.append(line)
async def start_background_tasks(app):
app['process_listener'] = app.loop.create_task(listen_to_process(app))
async def cleanup_background_tasks(app):
print('cleanup background tasks...')
app['process_listener'].cancel()
await app['process_listener']
def init():
app = web.Application()
app['websockets'] = []
app.router.add_get('/updates', websocket_handler)
cmd = "very long running subprocess"
app['process'] = subprocess.Popen(cmd.split(" "), stdout=subprocess.PIPE)
app.on_startup.append(start_background_tasks)
app.on_cleanup.append(cleanup_background_tasks)
app.on_shutdown.append(on_shutdown)
return app
web.run_app(init())
Итак, мой вопрос: как я могу читать строки из stdout в цикле на фоне моего приложения?Спасибо за любые ссылки.