1 files changed, 38 insertions, 0 deletions
diff --git a/ipsilon/providers/openid/extensions/teams.py b/ipsilon/providers/openid/extensions/teams.py
new file mode 100755
index 0000000..50c09af
--- /dev/null
+++ b/ipsilon/providers/openid/extensions/teams.py
@@ -0,0 +1,38 @@
+#!/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_teams import teams
+
+
+class Teams(OpenidExtensionBase):
+
+    def __init__(self, name):
+        super(Teams, self).__init__(name)
+        self.type_uris = [
+            teams.teams_uri,
+        ]
+
+    def _resp(self, request, userdata):
+        req = teams.TeamsRequest.fromOpenIDRequest(request)
+        if req is None:
+            return {}
+        data = userdata.get('groups', [])
+        return teams.TeamsResponse.extractResponse(req, data)
+
+    def _display(self, request, userdata):
+        resp = self._resp(request, userdata)
+        if resp.teams:
+            return {'Groups': resp.teams}
+        return {}
+
+    def _response(self, request, userdata):
+        return self._resp(request, userdata)
+
+
+class OpenidExtension(Teams):
+    def __init__(self):
+        super(OpenidExtension, self).__init__('Teams')