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
|
# Copyright (C) 2018 Philipp Hörist <philipp AT hoerist.com>
#
# This file is part of nbxmpp.
#
# SPDX-License-Identifier: GPL-3.0-or-later
from __future__ import annotations
from typing import TYPE_CHECKING
from nbxmpp.const import AvatarState
from nbxmpp.const import PresenceType
from nbxmpp.modules.base import BaseModule
from nbxmpp.namespaces import Namespace
from nbxmpp.protocol import Presence
from nbxmpp.structs import PresenceProperties
from nbxmpp.structs import StanzaHandler
if TYPE_CHECKING:
from nbxmpp.client import Client
class VCardAvatar(BaseModule):
def __init__(self, client: Client) -> None:
BaseModule.__init__(self, client)
self._client = client
self.handlers = [
StanzaHandler(
name="presence",
callback=self._process_avatar,
ns=Namespace.VCARD_UPDATE,
priority=15,
)
]
def _process_avatar(
self, _client: Client, stanza: Presence, properties: PresenceProperties
) -> None:
if properties.type != PresenceType.AVAILABLE:
return
update = stanza.getTag("x", namespace=Namespace.VCARD_UPDATE)
if update is None:
return
avatar_sha = update.getTagData("photo")
if avatar_sha is None:
properties.avatar_state = AvatarState.NOT_READY
self._log.info("%s is not ready to promote an avatar", stanza.getFrom())
# Empty update element, ignore
return
if avatar_sha == "":
properties.avatar_state = AvatarState.EMPTY
self._log.info("%s empty avatar advertised", stanza.getFrom())
return
properties.avatar_sha = avatar_sha
properties.avatar_state = AvatarState.ADVERTISED
self._log.info("%s advertises %s", stanza.getFrom(), avatar_sha)
|
