File size: 2,207 Bytes
d1ceb73 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
# mypy: allow-untyped-defs
class CompilationCallbackHandler:
def __init__(self):
self.start_callbacks = []
self.end_callbacks = []
def register_start_callback(self, callback):
"""
Register a callback function to be called when the compilation starts.
Args:
- callback (callable): The callback function to register.
"""
self.start_callbacks.append(callback)
return callback
def register_end_callback(self, callback):
"""
Register a callback function to be called when the compilation ends.
Args:
- callback (callable): The callback function to register.
"""
self.end_callbacks.append(callback)
return callback
def remove_start_callback(self, callback):
"""
Remove a registered start callback function.
Args:
- callback (callable): The callback function to remove.
"""
self.start_callbacks.remove(callback)
def remove_end_callback(self, callback):
"""
Remove a registered end callback function.
Args:
- callback (callable): The callback function to remove.
"""
self.end_callbacks.remove(callback)
def run_start_callbacks(self):
"""
Execute all registered start callbacks.
"""
for callback in self.start_callbacks:
callback()
def run_end_callbacks(self):
"""
Execute all registered end callbacks.
"""
for callback in self.end_callbacks:
callback()
def clear(self):
"""
Clear all registered callbacks.
"""
self.start_callbacks.clear()
self.end_callbacks.clear()
callback_handler = CompilationCallbackHandler()
def on_compile_start(callback):
"""
Decorator to register a callback function for the start of the compilation.
"""
callback_handler.register_start_callback(callback)
return callback
def on_compile_end(callback):
"""
Decorator to register a callback function for the end of the compilation.
"""
callback_handler.register_end_callback(callback)
return callback
|