Add timeout to signal awaiting configs.
This commit is contained in:
parent
6f49d819ac
commit
ba5d560a01
1 changed files with 14 additions and 2 deletions
16
i3toolwait
16
i3toolwait
|
@ -6,6 +6,7 @@ import collections.abc
|
||||||
import asyncio
|
import asyncio
|
||||||
import signal
|
import signal
|
||||||
import os
|
import os
|
||||||
|
import time
|
||||||
import functools
|
import functools
|
||||||
import json
|
import json
|
||||||
|
|
||||||
|
@ -313,9 +314,13 @@ async def run(configs: list[dict], *, timeout: int, debug: bool):
|
||||||
'pid': os.getpid(),
|
'pid': os.getpid(),
|
||||||
}
|
}
|
||||||
coroutines = []
|
coroutines = []
|
||||||
|
timeouts = [timeout]
|
||||||
|
started_at = time.monotonic_ns()
|
||||||
for cfg in configs:
|
for cfg in configs:
|
||||||
cfg['filter'] = parse(cfg['filter'])
|
cfg['filter'] = parse(cfg['filter'])
|
||||||
p = cfg['program']
|
p = cfg['program']
|
||||||
|
t = cfg.get('timeout', timeout)
|
||||||
|
timeouts += [t]
|
||||||
if isinstance(p, collections.abc.Iterable) and not isinstance(p, str):
|
if isinstance(p, collections.abc.Iterable) and not isinstance(p, str):
|
||||||
p = ' '.join(p)
|
p = ' '.join(p)
|
||||||
p = p.format(**variables)
|
p = p.format(**variables)
|
||||||
|
@ -323,12 +328,19 @@ async def run(configs: list[dict], *, timeout: int, debug: bool):
|
||||||
if cfg.get('signal_continue', False):
|
if cfg.get('signal_continue', False):
|
||||||
sig = signal.Signals(cfg.get('signal_continue'))
|
sig = signal.Signals(cfg.get('signal_continue'))
|
||||||
coro = coro_wait_signal(coro, sig)
|
coro = coro_wait_signal(coro, sig)
|
||||||
coroutines += [coro]
|
try:
|
||||||
|
await asyncio.wait_for(coro, timeout=t/1000)
|
||||||
|
except asyncio.TimeoutError:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
coroutines += [coro]
|
||||||
await asyncio.gather(*coroutines)
|
await asyncio.gather(*coroutines)
|
||||||
try:
|
try:
|
||||||
if window_configs:
|
if window_configs:
|
||||||
# run main loop only if we wait for something
|
# run main loop only if we wait for something
|
||||||
await asyncio.wait_for(ipc.main(), timeout=timeout/1000)
|
diff = (time.monotonic_ns() - started_at) / (1000*1000)
|
||||||
|
new_timeout = max(max(*timeouts) - diff, 0)
|
||||||
|
await asyncio.wait_for(ipc.main(), timeout=new_timeout/1000)
|
||||||
except asyncio.TimeoutError:
|
except asyncio.TimeoutError:
|
||||||
return 1
|
return 1
|
||||||
return 0
|
return 0
|
||||||
|
|
Loading…
Reference in a new issue