74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
from aiosched.util.debugging import BaseDebugger
|
|
from selectors import EVENT_READ, EVENT_WRITE
|
|
|
|
|
|
class Debugger(BaseDebugger):
|
|
"""
|
|
A simple debugger for aiosched
|
|
"""
|
|
|
|
def on_start(self):
|
|
print("## Started running")
|
|
|
|
def on_exit(self):
|
|
print("## Finished running")
|
|
|
|
def on_task_schedule(self, task, delay: int):
|
|
print(
|
|
f">> A task named '{task.name}' was scheduled to run in {delay:.2f} seconds"
|
|
)
|
|
|
|
def on_task_spawn(self, task):
|
|
print(f">> A task named '{task.name}' was spawned")
|
|
|
|
def on_task_exit(self, task):
|
|
print(f"<< Task '{task.name}' exited")
|
|
|
|
def before_task_step(self, task):
|
|
print(f"-> About to run a step for '{task.name}'")
|
|
|
|
def after_task_step(self, task):
|
|
print(f"<- Ran a step for '{task.name}'")
|
|
|
|
def before_sleep(self, task, seconds):
|
|
print(f"# About to put '{task.name}' to sleep for {seconds:.2f} seconds")
|
|
|
|
def after_sleep(self, task, seconds):
|
|
print(f"# Task '{task.name}' slept for {seconds:.2f} seconds")
|
|
|
|
def before_io(self, timeout):
|
|
if timeout is None:
|
|
timeout = float("inf")
|
|
print(f"!! About to check for I/O for up to {timeout:.2f} seconds")
|
|
|
|
def after_io(self, timeout):
|
|
print(f"!! Done I/O check (waited for {timeout:.2f} seconds)")
|
|
|
|
def before_cancel(self, task):
|
|
print(f"// About to cancel '{task.name}'")
|
|
|
|
def after_cancel(self, task):
|
|
print(f"// Cancelled '{task.name}'")
|
|
|
|
def on_exception_raised(self, task, exc):
|
|
print(f"== '{task.name}' raised {repr(exc)}")
|
|
|
|
def on_context_creation(self, ctx):
|
|
print(f"=> A new context was created by {ctx.entry_point.name!r}")
|
|
|
|
def on_context_exit(self, ctx):
|
|
print(f"=> A context was closed by {ctx.entry_point.name}")
|
|
|
|
def on_io_schedule(self, stream, event: int):
|
|
evt = ""
|
|
if event == EVENT_READ:
|
|
evt = "reading"
|
|
elif event == EVENT_WRITE:
|
|
evt = "writing"
|
|
elif event == EVENT_WRITE | EVENT_READ:
|
|
evt = "reading or writing"
|
|
print(f"|| Stream {stream!r} was scheduled for {evt}")
|
|
|
|
def on_io_unschedule(self, stream):
|
|
print(f"|| Stream {stream!r} was unscheduled")
|