File: custom_atexit.py

package info (click to toggle)
knot-resolver 6.0.17-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 16,376 kB
  • sloc: javascript: 42,732; ansic: 40,311; python: 12,580; cpp: 2,121; sh: 1,988; xml: 193; makefile: 181
file content (20 lines) | stat: -rw-r--r-- 480 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
 
"""
Custom replacement for standard module `atexit`. We use `atexit` behind the scenes, we just add the option
to invoke the exit functions manually.
"""

import atexit
from typing import Callable, List

_at_exit_functions: List[Callable[[], None]] = []


def register(func: Callable[[], None]) -> None:
    _at_exit_functions.append(func)
    atexit.register(func)


def run_callbacks() -> None:
    for func in _at_exit_functions:
        func()
        atexit.unregister(func)