summaryrefslogtreecommitdiffstats
path: root/ipsilon/providers/openid/extensions/ax.py
blob: d00a4fcd901f39edb5588b061afe1472c9b16fb3 (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
63
64
#!/usr/bin/python
#
# Copyright (C) 2014  Ipsilon project Contributors, for licensee see COPYING

from __future__ import absolute_import

from ipsilon.providers.openid.extensions.common import OpenidExtensionBase
from openid.extensions import ax


AP_MAP = {
    'http://schema.openid.net/namePerson': 'fullname',
    'http://schema.openid.net/contact/email': 'email',
    'http://axschema.org/namePerson': 'fullname',
    'http://axschema.org/namePerson/first': 'firstname',
    'http://axschema.org/namePerson/last': 'lastname',
    'http://axschema.org/namePerson/friendly': 'nickname',
    'http://axschema.org/contact/email': 'email',
    'http://openid.net/schema/namePerson/first': 'firstname',
    'http://openid.net/schema/namePerson/last': 'lastname',
    'http://openid.net/schema/namePerson/friendly': 'nickname',
    'http://openid.net/schema/gender': 'gender',
    'http://openid.net/schema/language/pref': 'language',
    'http://fedoauth.org/openid/schema/GPG/keyid': 'gpg_keyid',
    'http://fedoauth.org/openid/schema/SSH/key': 'ssh_key',
}


class OpenidExtension(OpenidExtensionBase):

    def __init__(self, *pargs):
        super(OpenidExtension, self).__init__('Attribute Exchange')
        self.type_uris = [
            ax.AXMessage.ns_uri,
        ]

    def _resp(self, request, userdata):
        req = ax.FetchRequest.fromOpenIDRequest(request)
        if req is None:
            return {}
        resp = ax.FetchResponse(req)
        for name in req.requested_attributes:
            try:
                self.debug(name)
                if name in AP_MAP:
                    resp.addValue(name, userdata[AP_MAP[name]])
                else:
                    resp.addValue(name, userdata[name])
            except Exception:  # pylint: disable=broad-except
                pass
        return resp

    def _display(self, request, userdata):
        resp = self._resp(request, userdata)
        data = dict()
        for name, value in resp.data.items():
            key = name
            if name in AP_MAP:
                key = AP_MAP[name]
            data[key] = ', '.join(value if value else [])
        return data

    def _response(self, request, userdata):
        return self._resp(request, userdata)