summaryrefslogtreecommitdiffstats
path: root/ipsilon/providers/openid/extensions/sreg.py
blob: e1144fc936b7751751f205fa1103e8c11ac94a55 (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
#!/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 sreg


class OpenidExtension(OpenidExtensionBase):

    def __init__(self, *pargs):
        super(OpenidExtension, self).__init__('Simple Registration')
        self.type_uris = [
            sreg.ns_uri_1_1,
            sreg.ns_uri_1_0,
        ]

    def _resp(self, request, userdata):
        req = sreg.SRegRequest.fromOpenIDRequest(request)
        data = dict()
        for name in sreg.data_fields:
            if name in userdata:
                data[name] = userdata[name]
        return sreg.SRegResponse.extractResponse(req, data)

    def _display(self, request, userdata):
        resp = self._resp(request, userdata)
        return resp.data

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