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
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
|
# 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 = "2013-09-30"
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: None,
KEY_CTRY: "United States", # database.getCountriesById and database.getCitiesById
KEY_PHONE_HOME: None,
KEY_PHONE_MOBILE: None,
KEY_LOCALITY: "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 getLocationString(id, key, user):
"""
Get country or city name by id
Args:
id: the id to get string for
key: whether to get country or city name
user: the User object
Returns:
Country/city name,
"""
if user:
if key == KEY_CTRY:
method = "database.getCountriesById"
arg = "country_ids"
else:
method = "database.getCitiesById"
arg = "city_ids"
data = user.vk.method(method, {arg: id})
if data:
data = data[0]
return data["name"]
return id
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))
# todo: find a proper way to handle this
elif key == KEY_URL:
if user:
vcard.setTagData(key, fields[key] % data)
else:
vcard.setTagData(key, fields[key])
elif key in (KEY_CTRY, KEY_LOCALITY) and value:
adr = vcard.getTag(KEY_ADR) or vcard.setTag(KEY_ADR)
adr.setTagData(key, getLocationString(value, key, user))
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:
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)", source, destination)
if destination == TransportID:
template = VCARD_TEMPLATE.copy()
template[KEY_URL] = GITHUB_URL
template[KEY_BDAY] = BIRTHDAY
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", PhotoSize) # todo: a feature to show the user's site instead of their URL?
data = user.vk.getUserData(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]
|
