|
2 | 2 |
|
3 | 3 | from __future__ import annotations |
4 | 4 |
|
5 | | -from enum import Enum |
6 | | -from typing import Any, Awaitable, Callable, Literal, Tuple |
| 5 | +from typing import Any, Awaitable, Callable |
7 | 6 |
|
| 7 | +from tigrbl_typing.phases import PHASE, HOOK_PHASES as PHASES, HookPhase |
8 | 8 |
|
9 | | -class PHASE(str, Enum): |
10 | | - PRE_TX_BEGIN = "PRE_TX_BEGIN" |
11 | | - START_TX = "START_TX" |
12 | | - PRE_HANDLER = "PRE_HANDLER" |
13 | | - HANDLER = "HANDLER" |
14 | | - POST_HANDLER = "POST_HANDLER" |
15 | | - PRE_COMMIT = "PRE_COMMIT" |
16 | | - END_TX = "END_TX" |
17 | | - POST_COMMIT = "POST_COMMIT" |
18 | | - POST_RESPONSE = "POST_RESPONSE" |
19 | | - ON_ERROR = "ON_ERROR" |
20 | | - ON_PRE_TX_BEGIN_ERROR = "ON_PRE_TX_BEGIN_ERROR" |
21 | | - ON_START_TX_ERROR = "ON_START_TX_ERROR" |
22 | | - ON_PRE_HANDLER_ERROR = "ON_PRE_HANDLER_ERROR" |
23 | | - ON_HANDLER_ERROR = "ON_HANDLER_ERROR" |
24 | | - ON_POST_HANDLER_ERROR = "ON_POST_HANDLER_ERROR" |
25 | | - ON_PRE_COMMIT_ERROR = "ON_PRE_COMMIT_ERROR" |
26 | | - ON_END_TX_ERROR = "ON_END_TX_ERROR" |
27 | | - ON_POST_COMMIT_ERROR = "ON_POST_COMMIT_ERROR" |
28 | | - ON_POST_RESPONSE_ERROR = "ON_POST_RESPONSE_ERROR" |
29 | | - ON_ROLLBACK = "ON_ROLLBACK" |
30 | | - |
31 | | - |
32 | | -HookPhase = Literal[ |
33 | | - "PRE_TX_BEGIN", |
34 | | - "START_TX", |
35 | | - "PRE_HANDLER", |
36 | | - "HANDLER", |
37 | | - "POST_HANDLER", |
38 | | - "PRE_COMMIT", |
39 | | - "END_TX", |
40 | | - "POST_COMMIT", |
41 | | - "POST_RESPONSE", |
42 | | - "ON_ERROR", |
43 | | - "ON_PRE_TX_BEGIN_ERROR", |
44 | | - "ON_START_TX_ERROR", |
45 | | - "ON_PRE_HANDLER_ERROR", |
46 | | - "ON_HANDLER_ERROR", |
47 | | - "ON_POST_HANDLER_ERROR", |
48 | | - "ON_PRE_COMMIT_ERROR", |
49 | | - "ON_END_TX_ERROR", |
50 | | - "ON_POST_COMMIT_ERROR", |
51 | | - "ON_POST_RESPONSE_ERROR", |
52 | | - "ON_ROLLBACK", |
53 | | -] |
54 | | - |
55 | | -PHASES: Tuple[HookPhase, ...] = tuple(p.value for p in PHASE) |
56 | 9 | Ctx = Any |
57 | 10 | StepFn = Callable[[Ctx], Awaitable[Any] | Any] |
58 | 11 | HookPredicate = Callable[[Any], bool] |
|
0 commit comments