]>
Commit | Line | Data |
---|---|---|
606f938b | 1 | from django.contrib.auth.middleware import RemoteUserMiddleware |
34b7ba86 AD |
2 | from django.contrib.auth.backends import RemoteUserBackend |
3 | from django.contrib import auth | |
606f938b AD |
4 | |
5 | def zephyr(msg, clas='remit', instance='log', rcpt='adehnert',): | |
6 | import os | |
7 | os.system("zwrite -d -c '%s' -i '%s' '%s' -m '%s'" % (clas, instance, rcpt, msg, )) | |
8 | ||
9 | class ScriptsRemoteUserMiddleware(RemoteUserMiddleware): | |
10 | header = 'SSL_CLIENT_S_DN_Email' | |
34b7ba86 AD |
11 | |
12 | class ScriptsRemoteUserBackend(RemoteUserBackend): | |
13 | def clean_username(self, username, ): | |
606f938b AD |
14 | if '@' in username: |
15 | name, domain = username.split('@') | |
16 | assert domain.upper() == 'MIT.EDU' | |
17 | return name | |
18 | else: | |
34b7ba86 | 19 | return username |
9c504e48 AD |
20 | def configure_user(self, user, ): |
21 | username = user.username | |
22 | import ldap | |
23 | con = ldap.open('ldap.mit.edu') | |
24 | con.simple_bind_s("", "") | |
25 | dn = "dc=mit,dc=edu" | |
26 | fields = ['cn', 'sn', 'givenName', 'mail', ] | |
27 | result = con.search_s('dc=mit,dc=edu', ldap.SCOPE_SUBTREE, 'uid=%s'%username, fields) | |
28 | if len(result) == 1: | |
29 | user.first_name = result[0][1]['givenName'][0] | |
30 | user.last_name = result[0][1]['sn'][0] | |
31 | user.email = result[0][1]['mail'][0] | |
32 | user.save() | |
33 | return user |