-
Notifications
You must be signed in to change notification settings - Fork 37
Expand file tree
/
Copy pathmessage.py
More file actions
144 lines (129 loc) · 6.23 KB
/
Copy pathmessage.py
File metadata and controls
144 lines (129 loc) · 6.23 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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
# coding: utf-8
"""
实时通讯消息相关操作。
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
from datetime import datetime
from typing import Any
from typing import Dict
from typing import Generator
from typing import List
from typing import Optional
from typing import Union
import six
from leancloud import client
class Message(object):
def __init__(self):
self.bin = None # type: bool
self.conversation_id = None # type: str
self.data = None # type: str
self.from_client = None # type: str
self.from_ip = None # type: str
self.is_conversation = None # type: bool
self.is_room = None # type: bool
self.message_id = None # type: str
self.timestamp = None # type: float
self.to = None # type: str
@classmethod
def _find(cls, query_params): # type: (dict) -> Generator[Message, None, None]
content = client.get("/rtm/messages/history", params=query_params).json()
for data in content:
msg = cls()
msg._update_data(data)
yield msg
@classmethod
def find_by_conversation(
cls,
conversation_id,
limit=None,
reversed=None,
before_time=None,
before_message_id=None,
):
# type: (str, Optional[int], Optional[bool], Optional[Union[datetime, float]], Optional[str]) -> List[Message] # noqa: E501
"""获取某个对话中的聊天记录
:param conversation_id: 对话 id
:param limit: 返回条数限制，可选，服务端默认 100 条，最大 1000 条
:param reversed: 以默认排序(查找更老的历史消息)相反的方向返回结果（也即从某条消息记录开始查找更新的消息），服务端默认为 False
如果 reversed = True，
则 before_time/before_message_id 转变成最老的消息的时间戳和 message_id，
否则还是指最新的消息的时间戳和 message_id。
:param before_time: 查询起始的时间戳，返回小于这个时间(不包含)的记录，服务端默认是当前时间
:param before_message_id: 起始的消息 id，使用时必须加上对应消息的时间 before_time 参数，一起作为查询的起点
:return: 符合条件的聊天记录
"""
query_params = {} # type: Dict[str, Any]
query_params["convid"] = conversation_id
if limit is not None:
query_params["limit"] = limit
if reversed is not None:
query_params["reversed"] = reversed
if isinstance(before_time, datetime):
query_params["max_ts"] = round(before_time.timestamp() * 1000)
elif isinstance(before_time, six.integer_types) or isinstance(
before_time, float
):
query_params["max_ts"] = round(before_time * 1000)
if before_message_id is not None:
query_params["msgid"] = before_message_id
return list(cls._find(query_params))
@classmethod
def find_by_client(
cls, from_client, limit=None, before_time=None, before_message_id=None
):
# type: (str, Optional[int], Optional[Union[datetime, float]], Optional[str]) -> List[Message] # noqa: E501
"""获取某个 client 的聊天记录
:param from_client: 要获取聊天记录的 client id
:param limit: 返回条数限制，可选，服务端默认 100 条，最大 1000 条
:param before_time: 查询起始的时间戳，返回小于这个时间(不包含)的记录，服务端默认是当前时间
:param before_message_id: 起始的消息 id，使用时必须加上对应消息的时间 before_time 参数，一起作为查询的起点
:return: 符合条件的聊天记录
"""
query_params = {} # type: Dict[str, Any]
query_params["from"] = from_client
if limit is not None:
query_params["limit"] = limit
if isinstance(before_time, datetime):
query_params["max_ts"] = round(before_time.timestamp() * 1000)
elif isinstance(before_time, six.integer_types) or isinstance(
before_time, float
):
query_params["max_ts"] = round(before_time * 1000)
if before_message_id is not None:
query_params["msgid"] = before_message_id
return list(cls._find(query_params))
@classmethod
def find_all(cls, limit=None, before_time=None, before_message_id=None):
# type: (Optional[int], Optional[Union[datetime, float]], Optional[str]) -> List[Message] # noqa: E501
"""获取应用全部聊天记录
:param limit: 返回条数限制，可选，服务端默认 100 条，最大 1000 条
:param before_time: 查询起始的时间戳，返回小于这个时间(不包含)的记录，服务端默认是当前时间
:param before_message_id: 起始的消息 id，使用时必须加上对应消息的时间 before_time 参数，一起作为查询的起点
:return: 符合条件的聊天记录
"""
query_params = {} # type: Dict[str, Any]
if limit is not None:
query_params["limit"] = limit
if isinstance(before_time, datetime):
query_params["max_ts"] = round(before_time.timestamp() * 1000)
elif isinstance(before_time, six.integer_types) or isinstance(
before_time, float
):
query_params["max_ts"] = round(before_time * 1000)
if before_message_id is not None:
query_params["msgid"] = before_message_id
return list(cls._find(query_params))
def _update_data(self, server_data): # type: (dict) -> None
self.bin = server_data.get("bin")
self.conversation_id = server_data.get("conv-id")
self.data = server_data.get("data")
self.from_client = server_data.get("from")
self.from_ip = server_data.get("from-ip")
self.is_conversation = server_data.get("is-conv")
self.is_room = server_data.get("is-room")
self.message_id = server_data.get("msg-id")
self.timestamp = server_data.get("timestamp", 0) / 1000
self.to = server_data.get("to")