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
85
86
87
88
89
90
|
"""PgQ event container.
"""
from typing import Any, Optional, Mapping, KeysView, Iterator, ValuesView, ItemsView
from skytools.basetypes import DictRow
__all__ = ['Event']
_fldmap = {
'ev_id': 'ev_id',
'ev_txid': 'ev_txid',
'ev_time': 'ev_time',
'ev_type': 'ev_type',
'ev_data': 'ev_data',
'ev_extra1': 'ev_extra1',
'ev_extra2': 'ev_extra2',
'ev_extra3': 'ev_extra3',
'ev_extra4': 'ev_extra4',
'ev_retry': 'ev_retry',
'id': 'ev_id',
'txid': 'ev_txid',
'time': 'ev_time',
'type': 'ev_type',
'data': 'ev_data',
'extra1': 'ev_extra1',
'extra2': 'ev_extra2',
'extra3': 'ev_extra3',
'extra4': 'ev_extra4',
'retry': 'ev_retry',
}
class Event(Mapping[str, Any]):
"""Event data for consumers.
Will be removed from the queue by default.
"""
__slots__ = ('_event_row', 'retry_time', 'queue_name')
_event_row: DictRow
retry_time: int
queue_name: str
def __init__(self, queue_name: str, row: DictRow) -> None:
self._event_row = row
self.retry_time = 60
self.queue_name = queue_name
def __getattr__(self, key: str) -> Any:
return self._event_row[_fldmap[key]]
def __iter__(self) -> Iterator[str]:
return iter(self._event_row)
def __len__(self) -> int:
return len(self._event_row)
# would be better in RetriableEvent only since we don't care but
# unfortunately it needs to be defined here due to compatibility concerns
def tag_done(self) -> None:
pass
# be also dict-like
def __getitem__(self, key: str) -> Any:
return self._event_row.__getitem__(key)
def __contains__(self, key: object) -> bool:
return self._event_row.__contains__(key)
def get(self, key: str, default: Optional[Any] = None) -> Any:
return self._event_row.get(key, default)
def has_key(self, key: str) -> bool:
return key in self._event_row
def keys(self) -> KeysView[str]:
return self._event_row.keys()
def values(self) -> ValuesView[Any]:
return self._event_row.values()
def items(self) -> ItemsView[str, Any]:
return self._event_row.items()
def __str__(self) -> str:
return "<id=%d type=%s data=%s e1=%s e2=%s e3=%s e4=%s>" % (
self.id, self.type, self.data, self.extra1, self.extra2, self.extra3, self.extra4)
|
