-
Notifications
You must be signed in to change notification settings - Fork 19
Expand file tree
/
Copy pathsegment.py
More file actions
74 lines (62 loc) · 2.63 KB
/
segment.py
File metadata and controls
74 lines (62 loc) · 2.63 KB
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
from typing import Dict, List, Optional
from stream_chat.base.segment import SegmentInterface
from stream_chat.types.base import SortParam
from stream_chat.types.segment import (
QuerySegmentTargetsOptions,
SegmentData,
SegmentUpdatableFields,
)
from stream_chat.types.stream_response import StreamResponse
class Segment(SegmentInterface):
def create(
self, segment_id: Optional[str] = None, data: Optional[SegmentData] = None
) -> StreamResponse:
if segment_id is not None:
self.segment_id = segment_id
if data is not None:
self.data = data
state = self.client.create_segment(
segment_type=self.segment_type, segment_id=self.segment_id, data=self.data
)
if self.segment_id is None and state.is_ok() and "segment" in state: # type: ignore
self.segment_id = state["segment"]["id"] # type: ignore
return state # type: ignore
def get(self) -> StreamResponse:
super().verify_segment_id()
return self.client.get_segment(segment_id=self.segment_id) # type: ignore
def update(self, data: SegmentUpdatableFields) -> StreamResponse:
super().verify_segment_id()
return self.client.update_segment( # type: ignore
segment_id=self.segment_id, data=data
)
def delete(self) -> StreamResponse:
super().verify_segment_id()
return self.client.delete_segment(segment_id=self.segment_id) # type: ignore
def target_exists(self, target_id: str) -> StreamResponse:
super().verify_segment_id()
return self.client.segment_target_exists( # type: ignore
segment_id=self.segment_id, target_id=target_id
)
def add_targets(self, target_ids: list) -> StreamResponse:
super().verify_segment_id()
return self.client.add_segment_targets( # type: ignore
segment_id=self.segment_id, target_ids=target_ids
)
def query_targets(
self,
filter_conditions: Optional[Dict] = None,
sort: Optional[List[SortParam]] = None,
options: Optional[QuerySegmentTargetsOptions] = None,
) -> StreamResponse:
super().verify_segment_id()
return self.client.query_segment_targets( # type: ignore
segment_id=self.segment_id,
sort=sort,
filter_conditions=filter_conditions,
options=options,
)
def remove_targets(self, target_ids: list) -> StreamResponse:
super().verify_segment_id()
return self.client.remove_segment_targets( # type: ignore
segment_id=self.segment_id, target_ids=target_ids
)