浏览代码

autologin on registration

sferot 11 年前
父节点
当前提交
48f7858c88
共有 1 个文件被更改,包括 9 次插入1 次删除
  1. 9 1
      pboard/pboard/controllers/api.py

+ 9 - 1
pboard/pboard/controllers/api.py 查看文件

14
 from PIL import Image as pil
14
 from PIL import Image as pil
15
 
15
 
16
 import tg
16
 import tg
17
-from tg import expose, flash, require, url, lurl, request, redirect, tmpl_context
17
+from tg import expose, flash, require, url, lurl, request, response, redirect, tmpl_context
18
 from tg.i18n import ugettext as _, lazy_ugettext as l_
18
 from tg.i18n import ugettext as _, lazy_ugettext as l_
19
 from tg import predicates as tgp
19
 from tg import predicates as tgp
20
 
20
 
26
 from pboard.model import auth as pma
26
 from pboard.model import auth as pma
27
 from pboard import model as pm
27
 from pboard import model as pm
28
 from pboard.lib.auth import can_read, can_write
28
 from pboard.lib.auth import can_read, can_write
29
+import repoze.who.api
29
 
30
 
30
 __all__ = ['PODPublicApiController', 'PODApiController']
31
 __all__ = ['PODPublicApiController', 'PODApiController']
31
 
32
 
68
 
69
 
69
         pm.DBSession.flush()
70
         pm.DBSession.flush()
70
 
71
 
72
+        who_api = repoze.who.api.get_api(request.environ)
73
+        creds = {}
74
+        creds['login'] = email
75
+        creds['password'] = password
76
+        authenticated, headers = who_api.login(creds)
77
+        response.headers = headers
78
+
71
         flash(_('Account successfully created: %s') % (email), 'info')
79
         flash(_('Account successfully created: %s') % (email), 'info')
72
         redirect(lurl('/'))
80
         redirect(lurl('/'))
73
 
81