blob: 4d652116f7f13b47961dbbb1d4a077ffc6ee3b6c (
plain)
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
|
"""
Slixmpp: The Slick XMPP Library
Copyright (C) 2020 Mathieu Pasquet
This file is part of Slixmpp.
See the file LICENSE for copying permission.
"""
from typing import Set, Iterable
from slixmpp.xmlstream import ElementBase
try:
from emoji import UNICODE_EMOJI
if UNICODE_EMOJI.get('en'):
UNICODE_EMOJI = UNICODE_EMOJI['en']
except ImportError:
UNICODE_EMOJI = None
NS = 'urn:xmpp:reactions:0'
class Reactions(ElementBase):
name = 'reactions'
plugin_attrib = 'reactions'
namespace = NS
interfaces = {'id', 'values'}
def get_values(self, *, all_chars=False) -> Set[str]:
""""Get all reactions as str"""
reactions = set()
for reaction in self:
value = reaction['value']
if UNICODE_EMOJI and not all_chars:
if value in UNICODE_EMOJI:
reactions.add(reaction['value'])
else:
reactions.add(reaction['value'])
return reactions
def set_values(self, values: Iterable[str], *, all_chars=False):
""""Set all reactions as str"""
for element in self.xml.findall('reaction'):
self.xml.remove(element)
for reaction_txt in values:
reaction = Reaction()
reaction.set_value(reaction_txt, all_chars=all_chars)
self.append(reaction)
class Reaction(ElementBase):
name = 'reaction'
namespace = NS
interfaces = {'value'}
def get_value(self) -> str:
return self.xml.text
def set_value(self, value: str, *, all_chars=False):
if UNICODE_EMOJI and not all_chars:
if not value in UNICODE_EMOJI:
raise ValueError("%s is not a valid emoji" % value)
self.xml.text = value
|
