Add OpenId extensions

[cascardo/ipsilon.git] / ipsilon / providers / openid / extensions / sreg.py

diff --git a/ipsilon/providers/openid/extensions/sreg.py b/ipsilon/providers/openid/extensions/sreg.py
new file mode 100755 (executable)
index 0000000..a2b4db7
--- /dev/null
+++ b/ipsilon/providers/openid/extensions/sreg.py
@@ -0,0 +1,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):
+        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)