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
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Query and control an archive of messages stored on a server using
XEP-0313: Message Archive Management(MAM).
"""
from datetime import datetime, timezone
from poezio.theming import get_theme
from poezio.text_buffer import Message, TextBuffer
async def MAM(self, remote_jid, start, end):
self.remote_jid = remote_jid
self.start_date = start
self.end_date = end
text_buffer = self._text_buffer
results = self.core.xmpp.plugin['xep_0313'].retrieve(jid=self.remote_jid,
iterator=True, start=self.start_date, end=self.end_date)
async for rsm in results:
for msg in rsm['mam']['results']:
forwarded = msg['mam_result']['forwarded']
timestamp = forwarded['delay']['stamp']
message = forwarded['stanza']
text = str(message['body'])
time = datetime.strftime(timestamp, '%Y-%m-%d %H:%M:%S')
time = datetime.strptime(time, '%Y-%m-%d %H:%M:%S')
time = time.replace(tzinfo=timezone.utc).astimezone(tz=None)
nick = str(message['from']).split('/')[1]
color = get_theme().COLOR_OWN_NICK
text_buffer.add_message(
text,
time,
nick,
color,
True, # History
None, # User
False, # Highlight
None, # Identifier
None, # str_time
None, # Jid
)
self.core.refresh_window()
|