126 lines
3.3 KiB
Python
126 lines
3.3 KiB
Python
|
__all__ = ('sleep', 'sleep_free', 'repeat_sleeping', 'move_on_after', )
|
||
|
|
|
||
|
|
import typing as T
|
||
|
|
import types
|
||
|
|
|
||
|
|
from kivy.clock import Clock
|
||
|
|
from asyncgui import _current_task, _sleep_forever, move_on_when, Task, Cancelled
|
||
|
|
|
||
|
|
|
||
|
|
@types.coroutine
|
||
|
|
def sleep(duration) -> T.Awaitable[float]:
|
||
|
|
'''
|
||
|
|
An async form of :meth:`kivy.clock.Clock.schedule_once`.
|
||
|
|
|
||
|
|
.. code-block::
|
||
|
|
|
||
|
|
dt = await sleep(5) # wait for 5 seconds
|
||
|
|
'''
|
||
|
|
task = (yield _current_task)[0][0]
|
||
|
|
clock_event = Clock.create_trigger(task._step, duration, False, False)
|
||
|
|
clock_event()
|
||
|
|
|
||
|
|
try:
|
||
|
|
return (yield _sleep_forever)[0][0]
|
||
|
|
except Cancelled:
|
||
|
|
clock_event.cancel()
|
||
|
|
raise
|
||
|
|
|
||
|
|
|
||
|
|
@types.coroutine
|
||
|
|
def sleep_free(duration) -> T.Awaitable[float]:
|
||
|
|
'''
|
||
|
|
An async form of :meth:`kivy.clock.Clock.schedule_once_free`.
|
||
|
|
|
||
|
|
.. code-block::
|
||
|
|
|
||
|
|
dt = await sleep_free(5) # wait for 5 seconds
|
||
|
|
'''
|
||
|
|
task = (yield _current_task)[0][0]
|
||
|
|
clock_event = Clock.create_trigger_free(task._step, duration, False, False)
|
||
|
|
clock_event()
|
||
|
|
|
||
|
|
try:
|
||
|
|
return (yield _sleep_forever)[0][0]
|
||
|
|
except Cancelled:
|
||
|
|
clock_event.cancel()
|
||
|
|
raise
|
||
|
|
|
||
|
|
|
||
|
|
class repeat_sleeping:
|
||
|
|
'''
|
||
|
|
Returns an async context manager that provides an efficient way to repeat sleeping.
|
||
|
|
|
||
|
|
When there is a piece of code like this:
|
||
|
|
|
||
|
|
.. code-block::
|
||
|
|
|
||
|
|
while True:
|
||
|
|
await sleep(0)
|
||
|
|
...
|
||
|
|
|
||
|
|
it can be translated to:
|
||
|
|
|
||
|
|
.. code-block::
|
||
|
|
|
||
|
|
async with repeat_sleeping(step=0) as sleep:
|
||
|
|
while True:
|
||
|
|
await sleep()
|
||
|
|
...
|
||
|
|
|
||
|
|
The latter is more suitable for situations requiring frequent sleeps, such as moving an object in every frame.
|
||
|
|
|
||
|
|
**Restriction**
|
||
|
|
|
||
|
|
You are not allowed to perform any kind of async operations inside the with-block except you can
|
||
|
|
``await`` the return value of the function that is bound to the identifier of the as-clause.
|
||
|
|
|
||
|
|
.. code-block::
|
||
|
|
|
||
|
|
async with repeat_sleeping(step=0) as sleep:
|
||
|
|
await sleep() # OK
|
||
|
|
await something_else # NOT ALLOWED
|
||
|
|
async with async_context_manager: # NOT ALLOWED
|
||
|
|
...
|
||
|
|
async for __ in async_iterator: # NOT ALLOWED
|
||
|
|
...
|
||
|
|
'''
|
||
|
|
|
||
|
|
__slots__ = ('_step', '_trigger', )
|
||
|
|
|
||
|
|
@types.coroutine
|
||
|
|
def _sleep(_f=_sleep_forever):
|
||
|
|
return (yield _f)[0][0]
|
||
|
|
|
||
|
|
def __init__(self, *, step=0):
|
||
|
|
self._step = step
|
||
|
|
|
||
|
|
@types.coroutine
|
||
|
|
def __aenter__(self, _sleep=_sleep) -> T.Awaitable[T.Callable[[], T.Awaitable[float]]]:
|
||
|
|
task = (yield _current_task)[0][0]
|
||
|
|
self._trigger = Clock.create_trigger(task._step, self._step, True, False)
|
||
|
|
self._trigger()
|
||
|
|
return _sleep
|
||
|
|
|
||
|
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
||
|
|
self._trigger.cancel()
|
||
|
|
|
||
|
|
|
||
|
|
def move_on_after(seconds: float) -> T.AsyncContextManager[Task]:
|
||
|
|
'''
|
||
|
|
Returns an async context manager that applies a time limit to its code block,
|
||
|
|
like :func:`trio.move_on_after` does.
|
||
|
|
|
||
|
|
.. code-block::
|
||
|
|
|
||
|
|
async with move_on_after(seconds) as bg_task:
|
||
|
|
...
|
||
|
|
if bg_task.finished:
|
||
|
|
print("The code block was interrupted due to a timeout")
|
||
|
|
else:
|
||
|
|
print("The code block exited gracefully.")
|
||
|
|
|
||
|
|
.. versionadded:: 0.6.1
|
||
|
|
'''
|
||
|
|
return move_on_when(sleep(seconds))
|