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
|
# coding: utf-8
# This file is a part of VK4XMPP transport
# © simpleApps, 2014 — 2016.
from __main__ import *
from __main__ import _
VCARD_SEMAPHORE = threading.Semaphore()
DESCRIPTION = "VK4XMPP Transport\n© simpleApps, 2013 — 2016."
GITHUB_URL = "https://github.com/mrDoctorWho/vk4xmpp"
BIRTHDAY = "30.08.2013"
KEY_NICKNAME = "NICKNAME"
KEY_NAME = "FN"
KEY_DESC = "DESC"
KEY_PHOTO = "PHOTO"
KEY_BINVAL = "BINVAL"
KEY_URL = "URL"
KEY_ADR = "ADR"
KEY_HOME = "HOME"
KEY_BDAY = "BDAY"
KEY_CTRY = "CTRY"
KEY_PHONE_HOME = "HOME"
KEY_PHONE_MOBILE = "MOBILE"
KEY_TEL = "TEL"
KEY_NUMBER = "NUMBER"
KEY_VOICE = "VOICE"
KEY_LOCALITY = "LOCALITY"
if AdditionalAbout:
DESCRIPTION = "%s\n%s" % (DESCRIPTION, AdditionalAbout)
# Vcard defaults
VCARD_TEMPLATE = {KEY_NICKNAME: IDENTIFIER["short"],
KEY_NAME: IDENTIFIER["name"],
KEY_DESC: DESCRIPTION,
KEY_PHOTO: URL_VCARD_NO_IMAGE,
KEY_URL: GITHUB_URL,
KEY_BDAY: BIRTHDAY,
KEY_CTRY: "United States", # database.getCountriesById and database.getCitiesById
KEY_PHONE_HOME: None,
KEY_PHONE_MOBILE: None,
KEY_LOCALITY: {"title": "Los Angeles"} # you'd love it here (yeah, here...)
}
VCARD_FIELDS = {KEY_NICKNAME: "screen_name",
KEY_NAME: "name",
KEY_URL: "https://vk.com/id%(id)s",
KEY_BDAY: "bdate",
KEY_CTRY: "country",
KEY_LOCALITY: "city",
KEY_PHONE_HOME: "home_phone",
KEY_PHONE_MOBILE: "mobile_phone",
KEY_URL: "site",
KEY_PHOTO: PhotoSize,
KEY_DESC: None,
}
def buildVcard(data, template=VCARD_TEMPLATE, fields=VCARD_FIELDS, user=None):
"""
Builds a vcard.
Uses VCARD_TEMPLATE as the base, then adds values from data.
Values from data are get with the help of the VCARD_FIELDS dict.
Args:
data: users.get result
user: the user object
Returns:
The user's VCARD.
"""
vcard = xmpp.Node("vCard", {"xmlns": xmpp.NS_VCARD})
for key, value in template.iteritems():
value = data.get(fields[key], value)
if key == KEY_PHOTO:
photo = vcard.setTag(KEY_PHOTO)
photo.setTagData(KEY_BINVAL, utils.getLinkData(value))
elif key in (KEY_CTRY, KEY_LOCALITY) and value:
adr = vcard.getTag(KEY_ADR) or vcard.setTag(KEY_ADR)
adr.setTagData(key, value.get("title"))
elif key == KEY_PHONE_MOBILE and value:
tel = vcard.getTag(KEY_TEL) or vcard.setTag(KEY_TEL)
tel.setTagData(KEY_NUMBER, value)
elif key == KEY_PHONE_HOME and value:
tel = vcard.getTag(KEY_TEL) or vcard.setTag(KEY_TEL)
tel.setTagData(KEY_PHONE_HOME, value)
elif key == KEY_BDAY and value:
if value.count(".") == 1:
value += time.strftime(".%Y")
value = time.strftime("%Y-%m-%d", time.strptime(value, "%d.%m.%Y"))
vcard.setTagData(key, value)
elif value and value != "None":
vcard.setTagData(key, value)
return vcard
@utils.threaded
def vcard_handler(cl, iq):
# Vcard feature makes transport hang (especially the photo part)
# Many clients love to query vcards so much, so the solution was in adding a semaphore here and sleep() at the bottom
# This is probably not a good idea, but for now this is the best one
with VCARD_SEMAPHORE:
jidFrom = iq.getFrom()
jidTo = iq.getTo()
source = jidFrom.getStripped()
destination = jidTo.getStripped()
result = iq.buildReply("result")
logger.debug("got vcard request to %s (jid: %s)", destination, source)
if destination == TransportID:
template = VCARD_TEMPLATE.copy()
vcard = buildVcard(template, template, template)
result.setPayload([vcard])
elif source in Users:
user = Users[source]
if user.friends:
id = vk2xmpp(destination)
args = ("screen_name", "bdate", "city", "country", "contacts", "home_town", "site", PhotoSize) # todo: a feature to show the user's site instead of their URL?
data = user.vk.getData(id, args)
data["id"] = id
if not user.settings.use_nicknames:
data["screen_name"] = data["name"]
vCard = buildVcard(data, VCARD_TEMPLATE, VCARD_FIELDS, user)
result.setPayload([vCard])
else:
result = utils.buildIQError(iq, xmpp.ERR_BAD_REQUEST, _("Your friend-list is empty."))
else:
result = utils.buildIQError(iq, xmpp.ERR_REGISTRATION_REQUIRED, _("You're not registered for this action."))
sender(cl, result)
MOD_TYPE = "iq"
MOD_HANDLERS = ((vcard_handler, "get", xmpp.NS_VCARD, False),)
MOD_FEATURES = [xmpp.NS_VCARD]
|
