53 lines
1.2 KiB
Python
53 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, BinaryIO, Generator, Protocol, TypeAlias
|
|
|
|
if TYPE_CHECKING:
|
|
from sqlalchemy.engine.base import Connection as SqlAlchemyConnection
|
|
else:
|
|
SqlAlchemyConnection = object
|
|
|
|
CaseVariable: TypeAlias = bool | int | str
|
|
|
|
|
|
class ReadFileFn(Protocol):
|
|
def __call__(self, key: str, id: int) -> BinaryIO: ...
|
|
|
|
|
|
class WriteFileFn(Protocol):
|
|
def __call__(self, key: str, id: int) -> "WriteHandle": ...
|
|
|
|
|
|
class WriteHandle(Protocol):
|
|
def write(self, data: bytes | str) -> int: ...
|
|
def flush(self) -> None: ...
|
|
|
|
|
|
class ProcessFn(Protocol):
|
|
def __call__(
|
|
self,
|
|
request: int,
|
|
read_file: ReadFileFn,
|
|
write_file: WriteFileFn,
|
|
connection: SqlAlchemyConnection,
|
|
) -> Generator[tuple[int, CaseVariable], None, None]: ...
|
|
|
|
|
|
class StartupFn(Protocol):
|
|
def __call__(
|
|
self,
|
|
read_file: ReadFileFn,
|
|
write_file: WriteFileFn,
|
|
connection: SqlAlchemyConnection,
|
|
) -> Generator[tuple[int, CaseVariable], None, None]: ...
|
|
|
|
|
|
__all__ = [
|
|
"CaseVariable",
|
|
"ReadFileFn",
|
|
"WriteHandle",
|
|
"WriteFileFn",
|
|
"ProcessFn",
|
|
"StartupFn",
|
|
]
|