ipsilon/providers/openid/extensions/sreg.py

   1 # Copyright (C) 2014 Ipsilon project Contributors, for license see COPYING
   2
   3 from __future__ import absolute_import
   4
   5 from ipsilon.providers.openid.extensions.common import OpenidExtensionBase
   6 from openid.extensions import sreg
   7
   8
   9 class OpenidExtension(OpenidExtensionBase):
  10
  11     def __init__(self, *pargs):
  12         super(OpenidExtension, self).__init__('Simple Registration')
  13         self.type_uris = [
  14             sreg.ns_uri_1_1,
  15             sreg.ns_uri_1_0,
  16         ]
  17
  18     def _resp(self, request, userdata):
  19         req = sreg.SRegRequest.fromOpenIDRequest(request)
  20         data = dict()
  21         for name in sreg.data_fields:
  22             if name in userdata:
  23                 data[name] = userdata[name]
  24         return sreg.SRegResponse.extractResponse(req, data)
  25
  26     def _display(self, request, userdata):
  27         resp = self._resp(request, userdata)
  28         return resp.data
  29
  30     def _response(self, request, userdata):
  31         return self._resp(request, userdata)