diff --git a/sysaudit/__init__.pyi b/sysaudit/__init__.pyi index cd66550..a423560 100644 --- a/sysaudit/__init__.pyi +++ b/sysaudit/__init__.pyi @@ -4,3 +4,15 @@ def audit(event: str, *args: typing.Any) -> None: ... def addaudithook( hook: typing.Callable[[str, typing.Tuple[typing.Any, ...]], None] ) -> None: ... + +_audit_fn = typing.Callable[[str, typing.Any], None] +_addaudithook_fn = typing.Callable[ + [typing.Callable[[str, typing.Tuple[typing.Any, ...]], None]], None +] + +std_audit = typing.Optional[_audit_fn] +std_addaudithook = typing.Optional[_addaudithook_fn] +csysaudit_audit = typing.Optional[_audit_fn] +csysaudit_addaudithook = typing.Optional[_addaudithook_fn] +py_audit = typing.Optional[_audit_fn] +py_addaudithook = typing.Optional[_addaudithook_fn] diff --git a/tests/test_import.py b/tests/test_import.py index 4fefc54..c602172 100644 --- a/tests/test_import.py +++ b/tests/test_import.py @@ -8,8 +8,8 @@ def test_module(): # type: () -> None assert sysaudit.addaudithook is not None if sys.version_info >= (3, 8, 0): - assert sysaudit.std_audit == sys.audit - assert sysaudit.std_addaudithook == sys.addaudithook + assert sysaudit.std_audit == sys.audit # type: ignore + assert sysaudit.std_addaudithook == sys.addaudithook # type: ignore assert sysaudit.audit == sys.audit # type: ignore [attr-defined] assert sysaudit.addaudithook == sys.addaudithook # type: ignore [attr-defined] else: