TLDR Quiero matar un subproceso como top mientras aún se está ejecutando
Estoy usando Fastapi para ejecutar un comando en la entrada. Por ejemplo, si entro arriba, mi programa ejecuta el comando pero como no regresa, en este momento tengo que usar un retraso de tiempo y luego matarlo/terminarlo. Sin embargo, quiero poder matarlo mientras aún se está ejecutando. Sin embargo, por el momento no ejecutará mi comando de eliminación hasta que se acabe el tiempo. Aquí está el código actual para ejecutar un proceso:
@app.put("/command/{command}") async def run_command(command: str): subprocess.run([command], shell=True, timeout=10) return {"run command"}y para matarlo
@app.get("/stop") async def stop(): proc.kill() return{"Stop"}soy nuevo en fastapi asi que agradeceria cualquier ayuda
Es porque subprocess.run se está bloqueando a sí mismo: debe ejecutar el comando Shell en segundo plano, por ejemplo, si ya tiene activado el bucle asnycio, podría usar subprocesos
import asyncio process = None @app.get("/command/{command}") async def run_command(command: str): global process process = await asyncio.create_subprocess_exec( command, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE ) return {"run command"} @app.get("/stop") async def stop(): process.kill() return {"Stop"}O con Popen
from subprocess import Popen process = None @app.get("/command/{command}") async def run_command(command: str): global process process = Popen([command]) # something long running return {"run command"}Agregar la opción de tiempo de espera puede ser complicado, ya que no desea esperar hasta que se complete (donde de hecho podría usar la función wait_for ), sino eliminar el proceso después de un tiempo específico. Hasta donde sé, la mejor opción sería programar otro proceso que sea responsable de matar al principal. El código con asyncio podría verse así:
@app.get("/command/{command}") async def run_command(command: str): global process process = await asyncio.create_subprocess_exec( command, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE ) loop = asyncio.get_event_loop() # We schedule here stop coroutine execution that runs after TIMEOUT seconds loop.call_later(TIMEOUT, asyncio.create_task, stop(process.pid)) return {"run command"} @app.get("/stop") async def stop(pid=None): global process # We need to make sure we won't kill different process if process and (pid is None or process.pid == pid): process.kill() process = None return {"Stop"}